python写的用WMI检测windows系统信息的脚本

 闲来无聊，用python写了一个检测windows系统硬件信息的脚本，主要就是用WMI模块来完成，分享给大家，希望对大家有所帮助。linux的系统信息直接用shell即可获取，这里不做介绍。获取主要检测内容为：系统平台，内存，硬盘硬件信息（详细的之前写了一个，大家参考），CPU信息，网卡信息等。

1. #!/usr/bin/env python 
2. # -*- coding: utf-8 -*- 
3. import wmi 
4. import sys,time,platform 
5.  
6. def get_system_info(os): 
7.     """  
8.     获取操作系统版本。  
9.     """  
10.     print 
11.     print "Operating system:" 
12.     if os == "Windows": 
13.         c = wmi.WMI () 
14.         for sys in c.Win32_OperatingSystem(): 
15.             print '\t' + "Version :\t%s" % sys.Caption.encode("GBK") 
16.             print '\t' + "Vernum :\t%s" % sys.BuildNumber 
17.  
18. def get_memory_info(os): 
19.     """  
20.     获取物理内存和虚拟内存。  
21.     """  
22.     print 
23.     print "memory_info:" 
24.     if os == "Windows": 
25.         c = wmi.WMI () 
26.         cs = c.Win32_ComputerSystem()  
27.         pfu = c.Win32_PageFileUsage()  
28.         MemTotal = int(cs[0].TotalPhysicalMemory)/1024/1024 
29.         print '\t' + "TotalPhysicalMemory :" + '\t' + str(MemTotal) + "M" 
30.         #tmpdict["MemFree"] = int(os[0].FreePhysicalMemory)/1024  
31.         SwapTotal = int(pfu[0].AllocatedBaseSize) 
32.         print '\t' + "SwapTotal :" + '\t' + str(SwapTotal) + "M" 
33.         #tmpdict["SwapFree"] = int(pfu[0].AllocatedBaseSize - pfu[0].CurrentUsage) 
34.  
35. def get_disk_info(os):  
36.     """  
37.     获取物理磁盘信息。  
38.     """  
39.     print 
40.     print "disk_info:" 
41.     if os == "Windows": 
42.         tmplist = []  
43.         c = wmi.WMI () 
44.         for physical_disk in c.Win32_DiskDrive(): 
45.             if physical_disk.Size: 
46.                 print '\t' + str(physical_disk.Caption) + ' :\t' + str(long(physical_disk.Size)/1024/1024/1024) + "G" 
47.  
48. def get_cpu_info(os):  
49.     """  
50.     获取CPU信息。  
51.     """  
52.     print 
53.     print "cpu_info:" 
54.     if os == "Windows": 
55.         tmpdict = {}  
56.         tmpdict["CpuCores"] = 0  
57.         c = wmi.WMI ()  
58.         for cpu in c.Win32_Processor():             
59.             tmpdict["CpuType"] = cpu.Name  
60.         try:  
61.             tmpdict["CpuCores"] = cpu.NumberOfCores  
62.         except:  
63.             tmpdict["CpuCores"] += 1  
64.             tmpdict["CpuClock"] = cpu.MaxClockSpeed     
65.         print '\t' + 'CpuType :\t' + str(tmpdict["CpuType"]) 
66.         print '\t' + 'CpuCores :\t' + str(tmpdict["CpuCores"]) 
67.  
68.  
69. def get_network_info(os):  
70.     """  
71.     获取网卡信息和当前TCP连接数。  
72.     """  
73.     print 
74.     print "network_info:" 
75.     if os == "Windows": 
76.         tmplist = []  
77.         c = wmi.WMI ()  
78.         for interface in c.Win32_NetworkAdapterConfiguration (IPEnabled=1):  
79.                 tmpdict = {}  
80.                 tmpdict["Description"] = interface.Description  
81.                 tmpdict["IPAddress"] = interface.IPAddress[0]  
82.                 tmpdict["IPSubnet"] = interface.IPSubnet[0]  
83.                 tmpdict["MAC"] = interface.MACAddress 
84.                 tmplist.append(tmpdict)  
85.         for i in tmplist: 
86.             print '\t' + i["Description"] 
87.             print '\t' + '\t' + "MAC :" + '\t' + i["MAC"] 
88.             print '\t' + '\t' + "IPAddress :" + '\t' + i["IPAddress"] 
89.             print '\t' + '\t' + "IPSubnet :" + '\t' + i["IPSubnet"] 
90.         for interfacePerfTCP in c.Win32_PerfRawData_Tcpip_TCPv4():  
91.                 print '\t' + 'TCP Connect :\t' + str(interfacePerfTCP.ConnectionsEstablished)  
92.  
93. if __name__ == "__main__": 
94.     os = platform.system() 
95.     get_system_info(os) 
96.     get_memory_info(os) 
97.     get_disk_info(os) 
98.     get_cpu_info(os) 
99.     get_network_info(os) 

运行结果如下：

本文转自 lover00751CTO博客，原文链接：http://blog.51cto.com/wangwei007/1033760，如需转载请自行联系原作者