zoukankan      html  css  js  c++  java
  • 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(' ') if i]
        for line in data:
            if line[0].strip():
                 parsed_data.append(new_line)
                 new_line = line+' '
            else:
                new_line += line + ' '
        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(' ')
            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(' ')[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'}
  • 相关阅读:
    C/C++一些库函数的实现
    约瑟夫环问题(Josephus)
    union关键字及大小端模式
    指针数组和数组指针
    巧用位运算
    C/C++生成可执行文件过程
    C语言+Modbus+NXP整体规划
    bug和待完善
    python学习第二课
    python学习第一课
  • 原文地址:https://www.cnblogs.com/weifeng1463/p/7521315.html
Copyright © 2011-2022 走看看