zhoujun.g 发表于 2018-8-6 11:40:59

python 收集主机信息

#!/usr/bin/envpython  

  
"""
  
file name: collect_info_a.py
  
"""
  

  
fromsubprocessimportPopen, PIPE
  

  
defgetIfconfig():
  
    p = Popen(['ifconfig'], stdout=PIPE, stderr=PIPE)
  
    data = p.stdout.read()
  
    return data
  

  
defgetDmi():
  
    p = Popen(['dmidecode'], stdout=PIPE, stderr=PIPE)
  
    data = p.stdout.read()
  
    return data
  

  

  
"""
  
从getIfconfig() 和getDmi() 函数返回的都是一个字符串。下面再定义一个
  
parseData(data) 的函数,将字符串分割成一个列表,每遇到顶格的行,就是
  
新的一段信息.
  
"""
  
defparseData(data):
  
    parsed_data = []
  
    new_line = ''
  
    data = #将字符串分割,去掉空行的元素
  
    forlinein data:
  
      ifline.strip():#当遇到顶格的行,就把new_line 保存的上一段信息,append 到parsed_line
  
            parsed_data.append(new_line)
  
            new_line = line+'\n'#重新保存新的一段的信息
  
      else:
  
            new_line += line+'\n'
  
    parsed_data.append(new_line)
  
    return #去掉空行的元素
  

  

  

  
"""
  
parseData(data) 函数返回的就是一个处理过的列表,将收集到的ip 字符串信息和 dmidecode 字符串信息,交给
  
下面定义的parseIfconfig() 和parseDmi() 函数分别处理,返回ip 信息的字典,和dmidecode 信息的字典。
  
"""
  

  
defparseIfconfig(parsed_data):
  
    parsed_data = #将"lo" 网卡的信息去掉
  
    dic = {}
  
    for lines in parsed_data:
  
      devname = lines.split('\n').split()
  
      macaddr = lines.split('\n').split()[-1]
  
      ipaddr= lines.split('\n').split().split(':')
  
      break   #由于只需要第一张网卡的信息,所以这里就可以退出循环了
  
    dic['ip'] = ipaddr
  
    return dic
  

  
defparseDmi(parsed_data):
  
    dic = {}
  
    parsed_data = #把这段信息直接整出来
  
    parsed_data = .split('\n') if i ]
  
    parsed_data =
  
    dmi_dic = dict(parsed_data)
  
    dic = {}
  
    dic['vender'] = dmi_dic['Manufacturer'].strip()
  
    dic['product'] = dmi_dic['Product Name'].strip()
  
    dic['sn']= dmi_dic['Serial Number'].strip()
  
    return dic
  

  

  
"""
  
getHostName: 函数
  
fn : 文件名参数
  
功能: 通过fn 传入文件名,读取HOSTNAME 信息
  
"""
  
def getHostName(fn):
  
    with open(fn) as fd:
  
      for line in fd:
  
            if line.startswith('HOSTNAME'):
  
                HostName = line.split('=').strip()
  
                break
  
    return {'HostName': HostName}
  

  

  
"""
  
getOSver: 函数
  
fn : 文件名参数
  
功能: 打开fn 文件,读取操作系统版本信息
  
"""
  
defgetOSver(fn):
  
    with open(fn) as fd:
  
      for line in fd:
  
            osver = line.strip()
  
            break
  
    return {'osver': osver}
  

  

  
"""
  
getCpu: 函数
  
fn : 文件名参数
  
功能: 读取fn 文件信息,读取cpu 核数和cpu 型号
  
"""
  
def getCpu(fn):
  
    num = 0
  
    dic = {}
  
    with open(fn) as fd:
  
         for line in fd:
  
             if line.startswith('processor'):
  
               num += 1
  
             if line.startswith('model name'):
  
               model_name = line.split(':')
  
               model_name = model_name.split() + ' ' + model_name.split()[-1]
  
    dic['cpu_num'] = num
  
    dic['cpu_model'] = model_name
  
    return dic
  

  

  
"""
  
getMemory: 函数
  
fn : 文件名参数
  
功能: 打开fn 文件,读取系统MemTotal 数值
  
"""
  
defgetMemory(fn):
  
    with open(fn) as fd:
  
      for line in fd:
  
            if line.startswith('MemTotal'):
  
                mem = int(line.split().strip())
  
                break
  
    mem = '%s' % int((mem/1024.0)) +'M'
  
    return {'Memory': mem}
  

  

  
if__name__ == '__main__':
  
    data_ip = getIfconfig()
  
    parsed_data_ip = parseData(data_ip)
  
    ip = parseIfconfig(parsed_data_ip)
  

  
    data_dmi = getDmi()
  
    parsed_data_dmi = parseData(data_dmi)
  
    dmi = prseDmi(parsed_data_dmi)
  
    HostName = getHostName('/etc/sysconfig/network')
  
    osver = getOSver('/etc/issue')
  
    cpu = getCpu('/proc/cpuinfo')
  
    mem = getMemory('/proc/meminfo')
  
    dic = {} #定义空字典,上面收集到的主机信息都是字典形式的,就是为了现在能将它们update 在一个字典
  
    dic.update(ip)
  
    dic.update(dmi)
  
    dic.update(HostName)
  
    dic.update(osver)
  
    dic.update(cpu)
  
    dic.update(mem)
  
    print dic
页: [1]
查看完整版本: python 收集主机信息