sysinfo_windows.py

#!/usr/bin/env python3
"""
sysinfo_windows.py — Windows system information collector.
Produces the same output structure as task4_1.sh.

Requires: psutil  (auto-installed if missing)
Output:   sysinfo_windows.out (same directory as this script)
"""

import os
import sys
import socket
import platform
import ipaddress
import subprocess
import time
from datetime import datetime, timezone, timedelta

# ── Auto-install psutil ───────────────────────────────────────────────────────

try:
    import psutil
except ImportError:
    print("psutil not found — installing...", flush=True)
    subprocess.check_call([sys.executable, "-m", "pip", "install", "psutil", "-q"])
    import psutil

# ── Helpers ───────────────────────────────────────────────────────────────────

_NO_DATA = {"", "not available", "not specified", "not present",
            "unknown", "to be filled by o.e.m.", "default string"}

def _clean(val: str) -> str:
    """Return val stripped, or empty string if it is a DMI non-value."""
    v = (val or "").strip()
    return "" if v.lower() in _NO_DATA else v


def _ps(command: str) -> str:
    """Run a PowerShell command and return stripped stdout, or '' on failure."""
    try:
        result = subprocess.run(
            ["powershell", "-NoProfile", "-NonInteractive", "-Command", command],
            capture_output=True, text=True, timeout=15,
        )
        return result.stdout.strip()
    except Exception:
        return ""


def _netmask_to_prefix(netmask: str) -> int:
    try:
        return ipaddress.IPv4Network(f"0.0.0.0/{netmask}").prefixlen
    except ValueError:
        return 0

# ── Hardware ──────────────────────────────────────────────────────────────────

def get_cpu() -> str:
    val = _ps("(Get-CimInstance Win32_Processor | Select-Object -First 1).Name")
    return _clean(val) or platform.processor() or "Unknown"


def get_ram() -> str:
    mem = psutil.virtual_memory()
    return f"{mem.total // (1024 ** 2)} MB"


def get_motherboard() -> str:
    mfr  = _clean(_ps("(Get-CimInstance Win32_BaseBoard | Select-Object -First 1).Manufacturer"))
    prod = _clean(_ps("(Get-CimInstance Win32_BaseBoard | Select-Object -First 1).Product"))
    if not prod:
        prod = "Unknown"
    return f"{mfr} {prod}".strip() if mfr else prod


def get_serial() -> str:
    val = _clean(_ps("(Get-CimInstance Win32_BIOS | Select-Object -First 1).SerialNumber"))
    return val or "Unknown"

# ── System ────────────────────────────────────────────────────────────────────

def get_os_distro() -> str:
    val = _clean(_ps("(Get-CimInstance Win32_OperatingSystem | Select-Object -First 1).Caption"))
    return val or f"{platform.system()} {platform.release()}"


def get_kernel_version() -> str:
    return platform.version() or "Unknown"


def get_install_date() -> str:
    # Get ISO-8601 string from PowerShell to avoid quote-escaping issues
    iso = _ps(
        "(Get-CimInstance Win32_OperatingSystem).InstallDate"
        ".ToUniversalTime().ToString('s')"
    )
    if iso:
        try:
            dt = datetime.strptime(iso, "%Y-%m-%dT%H:%M:%S")
            return dt.strftime("%a %b %d %H:%M:%S UTC %Y")
        except ValueError:
            pass
    return "Unknown"


def get_hostname() -> str:
    return socket.gethostname()


def get_uptime() -> str:
    elapsed_secs = int(time.time() - psutil.boot_time())
    total_mins   = elapsed_secs // 60
    weeks  = total_mins // 10080
    days   = (total_mins % 10080) // 1440
    hours  = (total_mins % 1440) // 60
    mins   = total_mins % 60

    parts = []
    if weeks:  parts.append(f"{weeks} weeks")
    if days:   parts.append(f"{days} days")
    if hours:  parts.append(f"{hours} hours")
    if mins:   parts.append(f"{mins} minutes")
    return ", ".join(parts) if parts else "0 minutes"


def get_processes() -> int:
    return len(psutil.pids())


def get_users() -> int:
    return len(psutil.users())

# ── Network ───────────────────────────────────────────────────────────────────

def get_network() -> list[str]:
    addrs = psutil.net_if_addrs()
    # Preserve interface order from net_if_stats (which reflects system order)
    stats = psutil.net_if_stats()
    ifaces = list(stats.keys())
    # Add any interface present in addrs but not in stats
    for name in addrs:
        if name not in ifaces:
            ifaces.append(name)

    lines = []
    for iface in ifaces:
        ipv4_addrs = [
            a for a in addrs.get(iface, [])
            if a.family == socket.AF_INET
        ]
        if ipv4_addrs:
            entries = []
            for a in ipv4_addrs:
                prefix = _netmask_to_prefix(a.netmask) if a.netmask else 0
                entries.append(f"{a.address}/{prefix}")
            lines.append(f"{iface}: {', '.join(entries)}")
        else:
            lines.append(f"{iface}: -")
    return lines

# ── Main ──────────────────────────────────────────────────────────────────────

def main():
    output_path = os.path.join(os.path.dirname(os.path.abspath(__file__)),
                               "sysinfo_windows.out")
    sections = [
        "--- Hardware ---",
        f"CPU: {get_cpu()}",
        f"RAM: {get_ram()}",
        f"Motherboard: {get_motherboard()}",
        f"System Serial Number: {get_serial()}",
        "--- System ---",
        f"OS Distribution: {get_os_distro()}",
        f"Kernel version: {get_kernel_version()}",
        f"Installation date: {get_install_date()}",
        f"Hostname: {get_hostname()}",
        f"Uptime: {get_uptime()}",
        f"Processes running: {get_processes()}",
        f"Users logged in: {get_users()}",
        "--- Network ---",
        *get_network(),
    ]

    with open(output_path, "w", encoding="utf-8") as f:
        f.write("\n".join(sections) + "\n")

    print(f"Written: {output_path}")


if __name__ == "__main__":
    main()