当前位置:

使用 python 收集获取 Linux 系统主机信息

2019-11-24  作者:炉石    阅读:2068

#!/usr/bin/env python

#encoding: utf-8


'''

收集主机的信息:

主机名称、IP、系统版本、服务器厂商、型号、序列号、CPU信息、内存信息

'''


from subprocess import Popen, PIPE

import os,sys


''' 获取 ifconfig 命令的输出 '''

def getIfconfig():

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

    data = p.stdout.read()

    return data


''' 获取 dmidecode 命令的输出 '''

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]


''' 根据输入的段落数据分析出ifconfig的每个网卡ip信息 '''

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


''' 根据输入的dmi段落数据 分析出指定参数 '''

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['vender'] = dmi_dic['Manufacturer'].strip()

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

    dic['sn'] = dmi_dic['Serial Number'].strip()

    return dic


''' 获取Linux系统主机名称 '''

def getHostname():

    with open('/etc/sysconfig/network') as fd:

        for line in fd:

            if line.startswith('HOSTNAME'):

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

                break

    return {'hostname':hostname}


''' 获取Linux系统的版本信息 '''

def getOsVersion():

    with open('/etc/issue') as fd:

        for line in fd:

            osver = line.strip()

            break

    return {'osver':osver}


''' 获取CPU的型号和CPU的核心数 '''

def getCpu():

    num = 0

    with open('/proc/cpuinfo') as fd:

        for line in fd:

            if line.startswith('processor'):

                num += 1

            if line.startswith('model name'):

                cpu_model = line.split(':')[1].strip().split()

                cpu_model = cpu_model[0] + ' ' + cpu_model[2]  + ' ' + cpu_model[-1]

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


''' 获取Linux系统的总物理内存 '''

def getMemory():

    with open('/proc/meminfo') as fd:

        for line in fd:

            if line.startswith('MemTotal'):

                mem = int(line.split()[1].strip())

                break

    mem = '%.f' % (mem / 1024.0) + ' MB'

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

    osver = getOsVersion()

    cpu = getCpu()

    mem = getMemory()

    

    dic.update(ip)

    dic.update(dmi)

    dic.update(hostname)

    dic.update(osver)

    dic.update(cpu)

    dic.update(mem)


    ''' 将获取到的所有数据信息并按简单格式对齐显示 '''

    for k,v in dic.items():

        print '%-10s:%s' % (k, v)



实验测试结果:

product   :VMware Virtual Platform

osver     :CentOS release 6.4 (Final)

sn        :VMware-56 4d b4 6c 05 e5 20 dc-c6 49 0c e1 e0 18 1c 75

Memory    :1870 MB

cpu_num   :2

ip        :192.168.0.8

vender    :VMware, Inc.

hostname  :vip

cpu_model :Intel(R) i7-4710MQ 2.50GHz