Python：收集系统信息

发布时间：2019-09-19 08:04:20编辑：auto阅读（1650）


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

    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


    验证结果如下：

    wKioL1dcFxyh96H4AAAwz1GaGkY397.png




关键字：

上一篇： django安装了pymysql还报My

下一篇： Python动态类型简介