15

2025-10-31 18:38:56 +08:00
parent 8d7d234dbe
commit a3c3d670e7
8 changed files with 339 additions and 0 deletions
@@ -0,0 +1,87 @@
+import subprocess
+import re
+
+class NetworkManager:
+    """获取并设置本机网络适配器信息（Windows）"""
+
+    def get_all_adapters(self):
+        """返回本机所有网卡信息列表，每个网卡是字典"""
+        result = subprocess.run("ipconfig /all", capture_output=True, text=True, shell=True)
+        lines = result.stdout.splitlines()
+        
+        adapters = []
+        adapter = None
+        
+        for line in lines:
+            line = line.strip()
+            if not line:
+                continue
+
+            # 匹配网卡名称
+            m = re.match(r'(?:以太网适配器|无线局域网适配器|未知适配器|Ethernet|Wi-Fi|Adapter)\s*(.+):', line)
+            if m:
+                if adapter:
+                    adapters.append(adapter)
+                adapter = {
+                    'name': m.group(1),
+                    'ip': '',
+                    'mask': '',
+                    'gateway': '',
+                    'dns1': '',
+                    'dns2': '',
+                    'dhcp': '',
+                    'media': '',
+                    'desc': ''
+                }
+                continue
+
+            if adapter is None:
+                continue
+
+            # 解析字段
+            if "IPv4 地址" in line or "IPv4 Address" in line:
+                adapter['ip'] = line.split(':')[-1].strip().split('(')[0]
+            elif "子网掩码" in line or "Subnet Mask" in line:
+                adapter['mask'] = line.split(':')[-1].strip()
+            elif "默认网关" in line or "Default Gateway" in line:
+                if not adapter['gateway']:
+                    adapter['gateway'] = line.split(':')[-1].strip()
+            elif "DHCP 已启用" in line or "DHCP Enabled" in line:
+                adapter['dhcp'] = line.split(':')[-1].strip()
+            elif "媒体状态" in line or "Media State" in line:
+                adapter['media'] = line.split(':')[-1].strip()
+            elif "描述" in line or "Description" in line:
+                adapter['desc'] = line.split(':')[-1].strip()
+            elif "DNS 服务器" in line or "DNS Servers" in line:
+                dns = line.split(':')[-1].strip()
+                if not adapter['dns1']:
+                    adapter['dns1'] = dns
+                elif not adapter['dns2']:
+                    adapter['dns2'] = dns
+
+        if adapter:
+            adapters.append(adapter)
+
+        return adapters
+
+    def set_static_ip(self, name, ip, mask, gateway):
+        """设置静态 IP"""
+        cmd = f'netsh interface ip set address name="{name}" static {ip} {mask} {gateway} 1'
+        return subprocess.run(cmd, shell=True, capture_output=True, text=True, encoding="gbk").stdout
+
+    def set_dhcp(self, name):
+        """切换 DHCP"""
+        cmd1 = f'netsh interface ip set address name="{name}" dhcp'
+        cmd2 = f'netsh interface ip set dns name="{name}" dhcp'
+        out1 = subprocess.run(cmd1, shell=True, capture_output=True, text=True, encoding="gbk").stdout
+        out2 = subprocess.run(cmd2, shell=True, capture_output=True, text=True, encoding="gbk").stdout
+        return out1 + "\n" + out2
+
+    def set_dns(self, name, dns1, dns2=None):
+        """设置 DNS"""
+        cmd1 = f'netsh interface ip set dns name="{name}" static {dns1} primary'
+        out = subprocess.run(cmd1, shell=True, capture_output=True, text=True, encoding="gbk").stdout
+        if dns2:
+            cmd2 = f'netsh interface ip add dns name="{name}" {dns2} index=2'
+            out += subprocess.run(cmd2, shell=True, capture_output=True, text=True, encoding="gbk").stdout
+        return out