python 收集主机信息
这篇文章是之前几篇的一个小综合。也是通过收集主机的一些参数信息，熟悉python里的文件读取，字符切割，字典存储等知识。
文章结构：
(1) 文章目的，需要收集的信息，并逐个分析如何获取信息。
(2) 给出实现的python 脚本。
(3) 总结。
1. 目的
用python 脚本实现收集主机信息：
(1)主机名(hostname).
当然了，在命令行下，直接使用"hostname" 命令就可以直接输出本机的hostname了。但是我们现在用python 一般是用读取文件的方式获取信息。主机的hostname 在"/etc/sysconfig/network" 文件里。
(2)ip地址.
要求收集第一块网卡的信息，网名，ip地址。在"ifconfig"命令的输出里，获取信息。前面有一篇博客已经介绍过具体的方法。
(3)操作系统版本，osver.
在"/etc/issue" 文件里有系统操作版本号的信息:
我们只需要将第一行的信息取出来就可以了。
(4)服务器厂商(vender)、服务器型号(product)和服务器序列号(sn).前面有一篇介绍收集dmidecode 信息的博客园，已经具体分析过。
(5)cpu型号(cpu_model)和cpu核数(cpu_num).
在"/proc/cpuinfo" 文件里可以获取，一般是一个"process" 就是一个cpu 核，所以通过统计"process" 字段就可以统计cpu 核数了。cpu型号，就是"cpu model" 了。如下：
做实验的机器，只有一个cpu 核数，所以"process" 字段就只有一个字段。通常cpu型号，"model name"，只需统计型号和主频。
(6)内存大小.
在"/proc/meminfo" 文件的，"MemTotal" 就是内存的大小，统计的时候按实际将单位转换一下就可以了。
2. 实现方法
先对ip 信息和dmidecode 信息进行处理，因为这两个信息在前面的博客都已经实现过了，比较容易。首先，就是读取"ifconfig" 和 "dmidecode" 命令的输出，然后分割保存成一个列表，再对列表进行分析。它们俩的分析过程基本一样，就一起给出：
#!/usr/bin/env python
"""
file name: collect_info_a.py
"""
from subprocess import Popen, PIPE
def getIfconfig():
p = Popen(['ifconfig'], stdout=PIPE, stderr=PIPE)
data = p.stdout.read()
return data
def getDmi():
p = Popen(['dmidecode'], stdout=PIPE, stderr=PIPE)
data = p.stdout.read()
return data
"""
从getIfconfig() 和getDmi() 函数返回的都是一个字符串。下面再定义一个
parseData(data) 的函数，将字符串分割成一个列表，每遇到顶格的行，就是
新的一段信息.
"""
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(): #当遇到顶格的行，就把new_line 保存的上一段信息，append 到parsed_line
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] #去掉空行的元素
"""
parseData(data) 函数返回的就是一个处理过的列表，将收集到的ip 字符串信息和 dmidecode 字符串信息，交给
下面定义的parseIfconfig() 和parseDmi() 函数分别处理，返回ip 信息的字典，和dmidecode 信息的字典。
"""
def parseIfconfig(parsed_data):
parsed_data = [i for parsed_data if i and not i.startswith('lo')] #将"lo" 网卡的信息去掉
dic = {}
for lines in parsed_data:
devname = lines.split('\n')[0].split()[0]
macaddr = lines.split('\n')[0].split()[-1]
ipaddr = lines.split('\n')[1].split()[1].split(':')[1]
break #由于只需要第一张网卡的信息，所以这里就可以退出循环了
dic['ip'] = ipaddr
return dic
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 ]
parsed_data = [i.strip().split(':') for i in parsed_data if i]
dmi_dic = dict(parsed_data)
dic = {}
dic['vender'] = dmi_dic['Manufacturer'].strip()
dic['product'] = dmi_dic['Product Name'].strip()
dic['sn'] = dmi_dic['Serial Number'].strip()
return dic
"""
getHostName: 函数
fn : 文件名参数
功能： 通过fn 传入文件名，读取HOSTNAME 信息
"""
def getHostName(fn):
with open(fn) as fd:
for line in fd:
if line.startswith('HOSTNAME'):
HostName = line.split('=')[1].strip()
break
return {'HostName': HostName}
"""
getOSver: 函数
fn : 文件名参数
功能： 打开fn 文件，读取操作系统版本信息
"""
def getOSver(fn):
with open(fn) as fd:
for line in fd:
osver = line.strip()
break
return {'osver': osver}
"""
getCpu: 函数
fn : 文件名参数
功能： 读取fn 文件信息，读取cpu 核数和cpu 型号
"""
def getCpu(fn):
num = 0
dic = {}
with open(fn) as fd:
for line in fd:
if line.startswith('processor'):
num += 1
if line.startswith('model name'):
model_name = line.split(':')[1]
model_name = model_name.split()[0] + ' ' + model_name.split()[-1]
dic['cpu_num'] = num
dic['cpu_model'] = model_name
return dic
"""
getMemory: 函数
fn : 文件名参数
功能： 打开fn 文件，读取系统MemTotal 数值
"""
def getMemory(fn):
with open(fn) as fd:
for line in fd:
if line.startswith('MemTotal'):
mem = int(line.split()[1].strip())
break
mem = '%s' % int((mem/1024.0)) +'M'
return {'Memory': mem}
if __name__ == '__main__':
data_ip = getIfconfig()
parsed_data_ip = parseData(data_ip)
ip = parseIfconfig(parsed_data_ip)
data_dmi = getDmi()
parsed_data_dmi = parseData(data_dmi)
dmi = prseDmi(parsed_data_dmi)
HostName = getHostName('/etc/sysconfig/network')
osver = getOSver('/etc/issue')
cpu = getCpu('/proc/cpuinfo')
mem = getMemory('/proc/meminfo')
dic = {} #定义空字典，上面收集到的主机信息都是字典形式的，就是为了现在能将它们update 在一个字典
dic.update(ip)
dic.update(dmi)
dic.update(HostName)
dic.update(osver)
dic.update(cpu)
dic.update(mem)
print dic测试一下结果：
在字典里保存了，文章开头列出所需的9个主机信息。
3. 总结：
还是熟悉读取文件的方法： 对字符切割，对列表的操作等等。如果需要读取其他的主机信息，再增加函数就可以了。