#!/usr/bin/env python3 """系统信息工具 - 获取系统运行时间、CPU、内存、磁盘、网络等状态信息。""" import sys import argparse import platform import datetime import os try: import psutil except ImportError: print("❌ 缺少依赖库: psutil。请运行: pip install psutil") sys.exit(1) def get_system_info(): """获取系统基本信息""" return { "系统": platform.system(), "主机名": platform.node(), "版本": platform.version(), "架构": platform.machine(), "处理器": platform.processor() or "未知", "Python版本": platform.python_version() } def get_uptime(): """获取系统运行时间""" boot_time = datetime.datetime.fromtimestamp(psutil.boot_time()) now = datetime.datetime.now() uptime = now - boot_time days = uptime.days hours = uptime.seconds // 3600 minutes = (uptime.seconds % 3600) // 60 seconds = uptime.seconds % 60 return { "启动时间": boot_time.strftime("%Y-%m-%d %H:%M:%S"), "运行时间": f"{days}天 {hours}小时 {minutes}分钟 {seconds}秒" } def get_cpu_info(): """获取CPU信息""" return { "物理核心数": psutil.cpu_count(logical=False), "逻辑核心数": psutil.cpu_count(logical=True), "使用率": f"{psutil.cpu_percent(interval=1)}%", "频率": f"{psutil.cpu_freq().current:.2f} MHz" if psutil.cpu_freq() else "未知" } def get_memory_info(): """获取内存信息""" mem = psutil.virtual_memory() swap = psutil.swap_memory() return { "总内存": f"{mem.total / (1024**3):.2f} GB", "可用内存": f"{mem.available / (1024**3):.2f} GB", "已用内存": f"{mem.used / (1024**3):.2f} GB", "内存使用率": f"{mem.percent}%", "交换分区总大小": f"{swap.total / (1024**3):.2f} GB" if swap.total > 0 else "无", "交换分区使用率": f"{swap.percent}%" if swap.total > 0 else "无" } def get_disk_info(path="/"): """获取磁盘信息""" try: disk = psutil.disk_usage(path) return { "路径": path, "总空间": f"{disk.total / (1024**3):.2f} GB", "已用空间": f"{disk.used / (1024**3):.2f} GB", "可用空间": f"{disk.free / (1024**3):.2f} GB", "使用率": f"{disk.percent}%" } except Exception as e: return {"错误": f"无法获取磁盘信息: {str(e)}"} def get_all_disks(): """获取所有磁盘分区信息""" partitions = [] for part in psutil.disk_partitions(): try: usage = psutil.disk_usage(part.mountpoint) partitions.append({ "挂载点": part.mountpoint, "文件系统": part.fstype, "总空间": f"{usage.total / (1024**3):.2f} GB", "已用空间": f"{usage.used / (1024**3):.2f} GB", "可用空间": f"{usage.free / (1024**3):.2f} GB", "使用率": f"{usage.percent}%" }) except: continue return partitions def get_process_info(): """获取进程信息""" return { "进程总数": len(psutil.pids()), "运行中进程": sum(1 for p in psutil.process_iter(['status']) if p.info['status'] == 'running'), "睡眠进程": sum(1 for p in psutil.process_iter(['status']) if p.info['status'] == 'sleeping') } def get_network_info(): """获取网络信息""" net = psutil.net_io_counters() return { "发送数据": f"{net.bytes_sent / (1024**2):.2f} MB", "接收数据": f"{net.bytes_recv / (1024**2):.2f} MB", "发送包数": net.packets_sent, "接收包数": net.packets_recv } def get_load_average(): """获取系统负载（仅Unix系统）""" if hasattr(os, 'getloadavg'): load1, load5, load15 = os.getloadavg() cpu_count = psutil.cpu_count() return { "1分钟负载": f"{load1:.2f} (每核心: {load1/cpu_count:.2f})", "5分钟负载": f"{load5:.2f} (每核心: {load5/cpu_count:.2f})", "15分钟负载": f"{load15:.2f} (每核心: {load15/cpu_count:.2f})" } else: return {"信息": "系统负载信息在Windows上不可用"} def get_temperature_info(): """获取温度信息（仅部分系统支持）""" try: temps = psutil.sensors_temperatures() if not temps: return {"信息": "温度传感器信息不可用"} result = {} for name, entries in temps.items(): for entry in entries: result[f"{name}"] = f"{entry.current}°C" if hasattr(entry, 'high') and entry.high: result[f"{name}_警告阈值"] = f"{entry.high}°C" break return result except: return {"信息": "温度传感器信息不可用"} def get_battery_info(): """获取电池信息（仅笔记本）""" try: battery = psutil.sensors_battery() if not battery: return {"信息": "电池信息不可用（可能是台式机）"} return { "电量百分比": f"{battery.percent}%", "充电中": "是" if battery.power_plugged else "否", "剩余时间": str(datetime.timedelta(seconds=battery.secsleft)) if battery.secsleft != -1 else "未知" } except: return {"信息": "电池信息不可用"} def format_output(data, output_format="text"): """格式化输出""" if output_format == "json": import json return json.dumps(data, ensure_ascii=False, indent=2) else: lines = [] for key, value in data.items(): if isinstance(value, dict): lines.append(f"\n📁 {key}:") for k, v in value.items(): lines.append(f" {k}: {v}") else: lines.append(f"{key}: {value}") return "\n".join(lines) def main(): parser = argparse.ArgumentParser( description="系统信息工具 - 获取系统运行时间、CPU、内存、磁盘、网络等状态信息。", epilog=""" 使用示例: # 获取系统基本信息 sysinfo --info system # 获取CPU信息 sysinfo --info cpu # 获取内存信息 sysinfo --info memory # 获取指定磁盘信息 sysinfo --info disk --disk-path /home # 获取所有磁盘分区 sysinfo --info disks # 获取所有信息（文本格式） sysinfo --info all # 获取所有信息（JSON格式） sysinfo --info all --format json # 获取系统负载 sysinfo --info load # 获取温度信息 sysinfo --info temperature # 获取电池信息 sysinfo --info battery # 组合多个信息 sysinfo --info cpu --info memory --info disk """ ) # 支持多个info参数 parser.add_argument("--info", "-i", action="append", choices=["system", "uptime", "cpu", "memory", "disk", "disks", "process", "network", "load", "temperature", "battery", "all"], help="要获取的信息类型（可多次使用）") parser.add_argument("--disk-path", "-d", type=str, default="/", help="磁盘检查路径（用于disk信息）") parser.add_argument("--format", "-f", choices=["text", "json"], default="text", help="输出格式（默认: text）") args = parser.parse_args() # 如果没有指定任何info，显示帮助 if not args.info: parser.print_help() return # 收集信息 result = {} info_types = args.info if "all" in info_types: info_types = ["system", "uptime", "cpu", "memory", "disk", "disks", "process", "network", "load", "temperature", "battery"] for info_type in info_types: if info_type == "system": result["系统信息"] = get_system_info() elif info_type == "uptime": result["运行时间"] = get_uptime() elif info_type == "cpu": result["CPU信息"] = get_cpu_info() elif info_type == "memory": result["内存信息"] = get_memory_info() elif info_type == "disk": result[f"磁盘信息({args.disk_path})"] = get_disk_info(args.disk_path) elif info_type == "disks": result["所有磁盘分区"] = get_all_disks() elif info_type == "process": result["进程信息"] = get_process_info() elif info_type == "network": result["网络信息"] = get_network_info() elif info_type == "load": result["系统负载"] = get_load_average() elif info_type == "temperature": result["温度信息"] = get_temperature_info() elif info_type == "battery": result["电池信息"] = get_battery_info() # 输出结果 print(format_output(result, args.format)) if __name__ == "__main__": main()