Update api.py

api.py

@@ -23,7 +23,87 @@ PORT_RANGE = (9300, 9700)
 CHROME_PATH = "google-chrome"  # Linux下可用命令，或自定义绝对路径
 PROFILE_BASE = "/tmp/profiles"
 
+import re
+import platform
+import psutil
+import datetime
 
+def get_system_info():
+    # CPU型号
+    try:
+        with open("/proc/cpuinfo") as f:
+            cpuinfo = f.read()
+        model_match = re.search(r"model name\s*:\s*(.+)", cpuinfo)
+        cpu_model = model_match.group(1) if model_match else "Unknown"
+    except Exception:
+        cpu_model = "Unknown"
+
+    # GPU型号
+    try:
+        gpu_info = subprocess.check_output("nvidia-smi --query-gpu=name --format=csv,noheader", shell=True).decode().strip()
+        gpu_model = gpu_info.split('\n')[0] if gpu_info else "Unknown"
+    except Exception:
+        try:
+            lspci_info = subprocess.check_output("lspci | grep VGA", shell=True).decode().strip()
+            gpu_model = lspci_info.split(":")[-1].strip() if lspci_info else "Unknown"
+        except Exception:
+            gpu_model = "Unknown"
+
+    # 操作系统信息
+    os_name = platform.system()
+    os_version = platform.version()
+    os_release = platform.release()
+
+    # 主机名
+    hostname = socket.gethostname()
+
+    # 启动时间
+    boot_time = datetime.datetime.fromtimestamp(psutil.boot_time()).strftime("%Y-%m-%d %H:%M:%S")
+
+    # 运行时长
+    uptime_seconds = int(time.time() - psutil.boot_time())
+    uptime = str(datetime.timedelta(seconds=uptime_seconds))
+
+    # 物理内存
+    mem = psutil.virtual_memory()
+    mem_total = f"{int(mem.total / 1024 / 1024)}MB"
+
+    # CPU核心数
+    cpu_count_logical = psutil.cpu_count()
+    cpu_count_physical = psutil.cpu_count(logical=False)
+
+    # 磁盘
+    disk = psutil.disk_usage('/')
+    disk_total = f"{int(disk.total / 1024 / 1024 / 1024)}GB"
+    disk_used = f"{int(disk.used / 1024 / 1024 / 1024)}GB"
+
+    # 当前用户
+    try:
+        user = os.getlogin()
+    except Exception:
+        user = "Unknown"
+
+    # IP地址
+    try:
+        ip = socket.gethostbyname(hostname)
+    except Exception:
+        ip = "Unknown"
+
+    return {
+        "cpu_model": cpu_model,
+        "gpu_model": gpu_model,
+        "os": f"{os_name} {os_release} ({os_version})",
+        "hostname": hostname,
+        "boot_time": boot_time,
+        "uptime": uptime,
+        "mem_total": mem_total,
+        "cpu_count_logical": cpu_count_logical,
+        "cpu_count_physical": cpu_count_physical,
+        "disk_total": disk_total,
+        "disk_used": disk_used,
+        "user": user,
+        "ip": ip
+    }
 def get_system_usage():
     cpu = subprocess.check_output("top -bn1 | grep 'Cpu(s)'", shell=True).decode().strip()
     mem = subprocess.check_output("free -m | grep Mem", shell=True).decode().strip()
@@ -56,8 +136,9 @@ def get_system_usage():
         "Free": f"{mem_free}MB",
         "Available": f"{mem_available}MB"
     }
+    sys_info = get_system_info()
     logger.info(f"CPU: {cpu_info}, Memory: {mem_info}")
-    info = {"CPU": cpu_info, "Memory": mem_info}
+    info = {"CPU": cpu_info, "Memory": mem_info, "System": sys_info}
     return info