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].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]

 

def parseIfconfig(parsed_data):

    dic = {}

    parsed_data = [i for i in parsed_data if not i.startswith('lo')]

    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')

    dic.update(ip)

    dic.update(dmi)

    dic.update(hostname)

    dic.update(osver)

    dic.update(cpu)

    dic.update(mem)

    print dic

 

 

[root@web10 day03]# python collect_info.py

{'osver': 'CentOS release 6.7 (Final)', 'product': 'VMware Virtual Platform', 'vendor': 'VMware, Inc.', 'sn': 'VMware-56 4d 59', 'memory': '2175M', 'cpu_num': 2, 'ip': '192.168.55.200', 'hostname': 'web10.aming.com', 'cpu_model': 'Intel(R) 2.50GHz'}