python 收集主机信息
#!/usr/bin/envpython"""
file name: collect_info_a.py
"""
fromsubprocessimportPopen, PIPE
defgetIfconfig():
p = Popen(['ifconfig'], stdout=PIPE, stderr=PIPE)
data = p.stdout.read()
return data
defgetDmi():
p = Popen(['dmidecode'], stdout=PIPE, stderr=PIPE)
data = p.stdout.read()
return data
"""
从getIfconfig() 和getDmi() 函数返回的都是一个字符串。下面再定义一个
parseData(data) 的函数,将字符串分割成一个列表,每遇到顶格的行,就是
新的一段信息.
"""
defparseData(data):
parsed_data = []
new_line = ''
data = #将字符串分割,去掉空行的元素
forlinein data:
ifline.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 #去掉空行的元素
"""
parseData(data) 函数返回的就是一个处理过的列表,将收集到的ip 字符串信息和 dmidecode 字符串信息,交给
下面定义的parseIfconfig() 和parseDmi() 函数分别处理,返回ip 信息的字典,和dmidecode 信息的字典。
"""
defparseIfconfig(parsed_data):
parsed_data = #将"lo" 网卡的信息去掉
dic = {}
for lines in parsed_data:
devname = lines.split('\n').split()
macaddr = lines.split('\n').split()[-1]
ipaddr= lines.split('\n').split().split(':')
break #由于只需要第一张网卡的信息,所以这里就可以退出循环了
dic['ip'] = ipaddr
return dic
defparseDmi(parsed_data):
dic = {}
parsed_data = #把这段信息直接整出来
parsed_data = .split('\n') if i ]
parsed_data =
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('=').strip()
break
return {'HostName': HostName}
"""
getOSver: 函数
fn : 文件名参数
功能: 打开fn 文件,读取操作系统版本信息
"""
defgetOSver(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(':')
model_name = model_name.split() + ' ' + model_name.split()[-1]
dic['cpu_num'] = num
dic['cpu_model'] = model_name
return dic
"""
getMemory: 函数
fn : 文件名参数
功能: 打开fn 文件,读取系统MemTotal 数值
"""
defgetMemory(fn):
with open(fn) as fd:
for line in fd:
if line.startswith('MemTotal'):
mem = int(line.split().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
页:
[1]