python獲取windows的硬件信息

# -*- coding: UTF-8 -*-
'''
=================================================
@Project -> File   ：CMDB -> collect_windows_info.py
@IDE    ：PyCharm
@Author ：XuMou
@Date   ：2020/7/30 12:57
==================================================
'''

import platform
import win32com
import wmi

"""
本模塊基於windows操作系統，依賴wmi和win32com庫，需要提前使用pip進行安裝，
pip install wmi
pip install pypiwin32
或者下載安裝包手動安裝。
"""


class Win32Info(object):

    def __init__(self):
        # 固定用法，更多內容請參考模塊說明
        self.wmi_obj = wmi.WMI()
        self.wmi_service_obj = win32com.client.Dispatch("WbemScripting.SWbemLocator")
        self.wmi_service_connector = self.wmi_service_obj.ConnectServer(".", "root\cimv2")

    def collect(self):
        data = {
            'os_type': platform.system(),
            'os_release': "%s %s  %s " % (platform.release(), platform.architecture()[0], platform.version()),
            'os_distribution': 'Microsoft',
            'asset_type': 'server'
        }

        # 分別獲取各種硬件信息
        data.update(self.get_cpu_info())
        data.update(self.get_ram_info())
        data.update(self.get_motherboard_info())
        data.update(self.get_disk_info())
        data.update(self.get_nic_info())
        # 最后返回一個數據字典
        return data

    def get_cpu_info(self):
        """
        獲取CPU的相關數據，這里只采集了三個數據，實際有更多，請自行選擇需要的數據
        :return:
        """
        data = {}
        cpu_lists = self.wmi_obj.Win32_Processor()
        cpu_core_count = 0
        for cpu in cpu_lists:
            cpu_core_count += cpu.NumberOfCores

        cpu_model = cpu_lists[0].Name   # CPU型號（所有的CPU型號都是一樣的）
        data["cpu_count"] = len(cpu_lists)      # CPU個數
        data["cpu_model"] = cpu_model
        data["cpu_core_count"] = cpu_core_count  # CPU總的核數

        return data

    def get_ram_info(self):
        """
        收集內存信息
        :return:
        """
        data = []
        # 這個模塊用SQL語言獲取數據
        ram_collections = self.wmi_service_connector.ExecQuery("Select * from Win32_PhysicalMemory")
        for ram in ram_collections:    # 主機中存在很多根內存，要循環所有的內存數據
            ram_size = int(int(ram.Capacity) / (1024**3))  # 轉換內存單位為GB
            item_data = {
                "slot": ram.DeviceLocator.strip(),
                "capacity": ram_size,
                "model": ram.Caption,
                "manufacturer": ram.Manufacturer,
                "sn": ram. SerialNumber,
            }
            data.append(item_data)  # 將每條內存的信息，添加到一個列表里

        return {"ram": data}    # 再對data列表封裝一層，返回一個字典，方便上級方法的調用

    def get_motherboard_info(self):
        """
        獲取主板信息
        :return:
        """
        computer_info = self.wmi_obj.Win32_ComputerSystem()[0]
        system_info = self.wmi_obj.Win32_OperatingSystem()[0]
        data = {}
        data['manufacturer'] = computer_info.Manufacturer
        data['model'] = computer_info.Model
        data['wake_up_type'] = computer_info.WakeUpType
        data['sn'] = system_info.SerialNumber
        return data

    def get_disk_info(self):
        """
        硬盤信息
        :return:
        """
        data = []
        for disk in self.wmi_obj.Win32_DiskDrive():     # 每塊硬盤都要獲取相應信息
            disk_data = {}
            interface_choices = ["SAS", "SCSI", "SATA", "SSD"]
            for interface in interface_choices:
                if interface in disk.Model:
                    disk_data['interface_type'] = interface
                    break
            else:
                disk_data['interface_type'] = 'unknown'

            disk_data['slot'] = disk.Index
            disk_data['sn'] = disk.SerialNumber
            disk_data['model'] = disk.Model
            disk_data['manufacturer'] = disk.Manufacturer
            disk_data['capacity'] = int(int(disk.Size) / (1024**3))
            data.append(disk_data)

        return {'physical_disk_driver': data}

    def get_nic_info(self):
        """
        網卡信息
        :return:
        """
        data = []
        for nic in self.wmi_obj.Win32_NetworkAdapterConfiguration():
            if nic.MACAddress is not None:
                nic_data = {}
                nic_data['mac'] = nic.MACAddress
                nic_data['model'] = nic.Caption
                nic_data['name'] = nic.Index
                if nic.IPAddress is not None:
                    nic_data['ip_address'] = nic.IPAddress[0]
                    nic_data['net_mask'] = nic.IPSubnet
                else:
                    nic_data['ip_address'] = ''
                    nic_data['net_mask'] = ''
                data.append(nic_data)

        return {'nic': data}


if __name__ == "__main__":
    # 測試代碼
    data = Win32Info().collect()
    for key in data:
        print(key, ":", data[key])