python 提取linux的硬件信息

1、代码实现目的

注：代码可以直接粘贴使用

为了实现对主机的整体有明确的了解，实现了下面的代码功能

代码主要功能是：

• 提取到主机的内存、硬盘、CPU、型号等信息
• 插入到mysql的数据库中

2、代码具体实现步骤

2.1首先要先在mysql中新建数据库和数据表

新建数据库：

?

1	create database host;

新建数据表：

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

CREATE TABLE `host_data` (   `id` int(11) NOT NULL AUTO_INCREMENT,   `product` varchar(255) DEFAULT NULL,   `cpu_num` varchar(255) DEFAULT NULL,   `hostname` varchar(255) DEFAULT NULL,   `ip_out` varchar(255) NOT NULL DEFAULT '',   `cpu_model` varchar(255) DEFAULT NULL,   `ip_in` varchar(255) DEFAULT NULL,   `osver` varchar(255) DEFAULT NULL,   `sn` varchar(255) DEFAULT NULL,   `Memory` varchar(255) DEFAULT NULL,   `vender` varchar(255) DEFAULT NULL,   `Disk` varchar(255) DEFAULT NULL,   PRIMARY KEY (`id`,`ip_out`),   UNIQUE KEY `ip_out` (`ip_out`) ) ENGINE=InnoDB AUTO_INCREMENT=33 DEFAULT CHARSET=utf8;

　

2.2 下面就是具体获取硬件信息的代码：

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288

#!/usr/bin/env python # encoding: utf-8   ''' 收集主机的信息： 主机名称、IP、系统版本、服务器厂商、型号、序列号、CPU信息、内存信息 '''   from subprocess import Popen, PIPE import os, sys import socket, fcntl, struct from collections import namedtuple import pymysql import re   '''获取所以磁盘信息'''   disk_ntuple = namedtuple('partition', 'device mountpoint fstype') usage_ntuple = namedtuple('usage', 'total used free percent')     ''' 获取 ifconfig 命令的输出 '''   def getIfconfig():     p = Popen(['ifconfig'], stdout=PIPE)     data = p.stdout.read()     return data     ''' 获取 dmidecode 命令的输出 '''   def getDmi():     p = Popen(['dmidecode'], stdout=PIPE)     data = p.stdout.read()     return data     '''获取磁盘总大小  格式sda[300GB]+sdb[300GB]'''   def humanize_bytes(bytesize, precision=0):     abbrevs = (         (10**15, 'PB'),         (10**12, 'TB'),         (10**9, 'GB'),         (10**6, 'MB'),         (10**3, 'kB'),         (1, 'bytes')     )     if bytesize == 1:         return '1 byte'     for factor, suffix in abbrevs:         if bytesize >= factor:             break     return '%.*f%s' % (precision, round(float(bytesize) / factor), suffix)   def dev_phy_size():   with open('/proc/partitions','r') as dp:     res = ''     for disk in dp.readlines():       if re.search(r'[s,h,v]d[a-z]\n',disk):         blknum = disk.strip().split(' ')[-2]         dev = disk.strip().split(' ')[-1]         size = int(blknum)*1024         consist = dev+'['+humanize_bytes(size).strip()+']'         res += consist + '+'     return {'Disk':res[:-1]}     # '''获取分区路径和挂载、格式信息''' # def disk_partitions(all=False): #     """Return all mountd partitions as a nameduple. #     If all == False return phyisical partitions only. #     """ #     phydevs = [] #     f = open("/proc/filesystems", "r") #     for line in f: #         if not line.startswith("nodev"): #             phydevs.append(line.strip()) # #     retlist = [] #     f = open('/etc/mtab', "r") #     for line in f: #         if not all and line.startswith('none'): #             continue #         fields = line.split() #         device = fields[0] #         mountpoint = fields[1] #         fstype = fields[2] #         if not all and fstype not in phydevs: #             continue #         if device == 'none': #             device = '' #         ntuple = disk_ntuple(device, mountpoint, fstype) #         retlist.append(ntuple) #     return retlist # # # # '''获取磁盘的使用情况（总、已用、可用、百分比）''' # def disk_usage(path): #     """Return disk usage associated with path.""" #     hd = {} #     st = os.statvfs(path) #     free = (st.f_bavail * st.f_frsize) #     total = (st.f_blocks * st.f_frsize) #     used = (st.f_blocks - st.f_bfree) * st.f_frsize #     try: #         percent = ret = (float(used) / total) * 100 #     except ZeroDivisionError: #         percent = 0 #     hd['total'] = total #     hd['used'] = used #     hd['free'] = free #     hd['percent'] = round(percent, 1) #     return hd # # # def access_disk(): #     hd = {} #     disks = disk_partitions() #     for i in disks: #         d = list(i)[1] #         a_disks = disk_usage(d) #         hd["cipan"] = a_disks #     return hd   ''' 根据空行分段落 返回段落列表'''     def parseData(data):     parsed_data = []     new_line = ''     data = [i for i in data.split('\n') if i]     for line in data:         if line[0].strip():             parsed_data.append(new_line)             new_line = line + '\n'         else:             new_line += line + '\n'     parsed_data.append(new_line)     return [i for i in parsed_data if i]     ''' 根据输入的段落数据分析出ifconfig的每个网卡ip信息 '''   '''外网IP''' def parseIfconfig_out(ifname):     dic = {}     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)     ipaddr = socket.inet_ntoa(fcntl.ioctl(s.fileno(), 0x8915, struct.pack('256s', ifname[:15]))[20:24])     dic['ip_out'] = ipaddr     return dic '''内网IP''' def parseIfconfig_in(ifname):     dic = {}     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)     ipaddr = socket.inet_ntoa(fcntl.ioctl(s.fileno(), 0x8915, struct.pack('256s', ifname[:15]))[20:24])     dic['ip_in'] = ipaddr     return dic     ''' 根据输入的dmi段落数据 分析出指定参数 '''     def parseDmi(parsed_data):     dic = {}     parsed_data = [i for i in parsed_data if i.startswith('System Information')]     parsed_data = [i for i in parsed_data[0].split('\n')[1:] if i]     dmi_dic = dict([i.strip().split(':') for i in parsed_data])     dic['vender'] = dmi_dic['Manufacturer'].strip()     dic['product'] = dmi_dic['Product Name'].strip()     dic['sn'] = dmi_dic['Serial Number'].strip()     return dic     ''' 获取Linux系统主机名称 '''     def getHostname():     p = Popen(['hostname'], stdout=PIPE)     data = p.stdout.read().split('\n')[0]     return {'hostname': data}     ''' 获取Linux系统的版本信息 '''   def getOsVersion():     with open('/etc/issue') as fd:         for line in fd:             osver = line.strip()             break     return {'osver': osver}     ''' 获取CPU的型号和CPU的核心数 '''     def getCpu():     num = 0     with open('/proc/cpuinfo') as fd:         for line in fd:             if line.startswith('processor'):                 num += 1             if line.startswith('model name'):                 cpu_model = line.split(':')[1].strip().split()                 cpu_model = cpu_model[0] + ' ' + cpu_model[2] + ' ' + cpu_model[-1]     return {'cpu_num': num, 'cpu_model': cpu_model}     ''' 获取Linux系统的总物理内存 '''     def getMemory():     with open('/proc/meminfo') as fd:         for line in fd:             if line.startswith('MemTotal'):                 mem = int(line.split()[1].strip())                 break     mem = '%.f' % (mem / 1024000.0) + ' GB'     return {'Memory': mem}         def main():     dic = {}     data_ip = getIfconfig()     parsed_data_ip = parseData(data_ip)     ip_out = parseIfconfig_out(sys.argv[1])     ip_in = parseIfconfig_in(sys.argv[2])       data_dmi = getDmi()     parsed_data_dmi = parseData(data_dmi)     dmi = parseDmi(parsed_data_dmi)       acc_disks = dev_phy_size()     # print acc_disks       hostname = getHostname()     osver = getOsVersion()     cpu = getCpu()     mem = getMemory()       dic.update(ip_in)     dic.update(ip_out)     dic.update(dmi)     dic.update(hostname)     dic.update(osver)     dic.update(cpu)     dic.update(mem)     dic.update(acc_disks)     return dic     def db_connect():     conn = pymysql.connect(host="x.x.x.x",                            db='xxxxx',                            user="root",                            passwd="*******",                            charset="utf8")     cur = conn.cursor()     return conn,cur   def save_mysql(conn, cur, dic):     k = dic.keys()     v = dic.values()     key = ",".join(k)     vs = list()     for i in v:         vs.append("'%s'" % i)     values = ",".join(vs)     # print values, k     try:         # sql = "REPLACE INTO host_data(%s) VALUES(%s)" % (key, values)         sql = "INSERT INTO host_data(%s) VALUES(%s)" % (key, values)         # print sql         cur.execute(sql)         conn.commit()     except Exception as e:         # print Exception, ": ", e         conn.rollback()       if __name__ == '__main__':     conn, cur = db_connect()     dic_data = main()     save_mysql(conn, cur, dic_data)

注：

• 脚本需要修改  db_connect() 函数的数据库信息  （db参数是数据库的库名）
• 运行方式：python  脚本名称  外网网卡名称   内网网卡名称（因为每台机器的网卡名称不一样，所以在执行脚本的时候最好的把内外网的网卡名称以传值的形式给到脚本）
• 比如: python host.py  eth1  eth0