python 采集系统_Python：收集系统信息

#!/usr/bin/env python

from subprocess import Popen, PIPE

def getIfconfig():

p = Popen([‘ifconfig‘], stdout=PIPE)

data = p.stdout.read()

return data

def getDmi():

p = Popen([‘dmidecode‘], stdout=PIPE)

data = p.stdout.read()

return 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].split():

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 and not i.startswith(‘lo‘)]

def parseIfconfig(parsed_data):

dic = {}

for lines in parsed_data:

line_list = lines.split(‘\n‘)

devname = line_list[0].split()[0]

macaddr = line_list[0].split()[-1]

ipaddr = line_list[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]

dmi_dic = dict([i.strip().split(‘:‘) for i in parsed_data])

dic[‘vendor‘] = dmi_dic[‘Manufacturer‘].strip()

dic[‘product‘] = dmi_dic[‘Product Name‘].strip()

dic[‘sn‘] = dmi_dic[‘Serial Number‘].strip()[:15]

return dic

def getHostname(f):

with open(f) as fd:

for line in fd:

if line.startswith(‘HOSTNAME‘):

hostname = line.split(‘=‘)[1].strip()

break

return {‘hostname‘:hostname}

def getOsver(f):

with open(f) as fd:

for line in fd:

osver = line.strip()

break

return {‘osver‘:osver}

def getCpu(f):

num = 0

with open(f) as fd:

for line in fd:

if line.startswith(‘processor‘):

num += 1

if line.startswith(‘model name‘):

cpu_model = line.split(‘:‘)[1].split()

cpu_model = cpu_model[0]+‘ ‘+cpu_model[-1]

return {‘cpu_num‘:num, ‘cpu_model‘:cpu_model}

def getMemory(f):

with open(f) 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__‘:

dic = {}

data_ip = getIfconfig()

parsed_data_ip = parseData(data_ip)

ip = parseIfconfig(parsed_data_ip)

data_dmi = getDmi()

parsed_data_dmi = parseData(data_dmi)

dmi = parseDmi(parsed_data_dmi)

hostname = getHostname(‘/etc/sysconfig/network‘)

osver = getOsver(‘/etc/issue‘)

cpu = getCpu(‘/proc/cpuinfo‘)

mem = getMemory(‘/proc/meminfo‘)

dic.update(ip)

dic.update(dmi)

dic.update(hostname)

dic.update(osver)

dic.update(cpu)

dic.update(mem)

print dic