收集主机的以下信息,并以字典形式输出。

1、主机名:hostname

2、IP地址:ip

3、操作系统版本:osver

4、服务器厂商:vendor

5、服务器型号:product

6、服务器序列号:sn

7、cpu型号:cpu_model

8、cpu核数:cpu_num

9、内存大小:memory



代码如下:


vim collect_info.py

#!/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


验证结果如下:

Python:收集系统信息_第1张图片