文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

python 收集主机信息

2023-01-31 07:02

关注

这篇文章是之前几篇的一个小综合。也是通过收集主机的一些参数信息,熟悉python里的文件读取,字符切割,字典存储等知识。


文章结构:

(1) 文章目的,需要收集的信息,并逐个分析如何获取信息。

(2) 给出实现的python 脚本。

(3) 总结。


1. 目的

  用python 脚本实现收集主机信息:

  (1)主机名(hostname).

  当然了,在命令行下,直接使用"hostname" 命令就可以直接输出本机的hostname了。但是我们现在用python 一般是用读取文件的方式获取信息。主机的hostname 在"/etc/sysconfig/network" 文件里。

  wKiom1gTUzuzSRbjAAAuRikFJYw284.png

  (2)ip地址.

  要求收集第一块网卡的信息,网名,ip地址。在"ifconfig"命令的输出里,获取信息。前面有一篇博客已经介绍过具体的方法。

  (3)操作系统版本,osver.

  在"/etc/issue" 文件里有系统操作版本号的信息:

  wKiom1gTVBXRnU4sAAAaDxhBJ_s009.png

  我们只需要将第一行的信息取出来就可以了。


  (4)服务器厂商(vender)、服务器型号(product)和服务器序列号(sn).前面有一篇介绍收集dmidecode 信息的博客园,已经具体分析过。


  (5)cpu型号(cpu_model)和cpu核数(cpu_num).

  在"/proc/cpuinfo" 文件里可以获取,一般是一个"process" 就是一个cpu 核,所以通过统计"process" 字段就可以统计cpu 核数了。cpu型号,就是"cpu model" 了。如下:

  wKioL1gTVaqxvvKVAABsrLzykh8825.png

  做实验的机器,只有一个cpu 核数,所以"process" 字段就只有一个字段。通常cpu型号,"model name",只需统计型号和主频。


  (6)内存大小.

  在"/proc/meminfo" 文件的,"MemTotal" 就是内存的大小,统计的时候按实际将单位转换一下就可以了。

  wKiom1gTVp2TmbHtAAAtl93vKFQ452.png


2. 实现方法

  先对ip 信息和dmidecode 信息进行处理,因为这两个信息在前面的博客都已经实现过了,比较容易。首先,就是读取"ifconfig" 和 "dmidecode" 命令的输出,然后分割保存成一个列表,再对列表进行分析。它们俩的分析过程基本一样,就一起给出:

 

#!/usr/bin/env  python 

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

from  subprocess  import  Popen, PIPE

def  getIfconfig():
    p = Popen(['ifconfig'], stdout=PIPE, stderr=PIPE)
    data = p.stdout.read()
    return data
    
def  getDmi():
    p = Popen(['dmidecode'], stdout=PIPE, stderr=PIPE)
    data = p.stdout.read()
    return data
   
    
"""
从getIfconfig() 和getDmi() 函数返回的都是一个字符串。下面再定义一个
parseData(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():  #当遇到顶格的行,就把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 [i for i in parsed_data if i]  #去掉空行的元素
   
   
    
"""
parseData(data) 函数返回的就是一个处理过的列表,将收集到的ip 字符串信息和 dmidecode 字符串信息,交给
下面定义的parseIfconfig() 和parseDmi() 函数分别处理,返回ip 信息的字典,和dmidecode 信息的字典。
"""
   
def  parseIfconfig(parsed_data):
    parsed_data = [i for parsed_data if i and  not i.startswith('lo')]  #将"lo" 网卡的信息去掉
    dic = {}
    for lines in parsed_data:
        devname = lines.split('\n')[0].split()[0]
        macaddr = lines.split('\n')[0].split()[-1]
        ipaddr  = lines.split('\n')[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 ]
    parsed_data = [i.strip().split(':') for i in parsed_data if i]
    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('=')[1].strip()
                break
    return {'HostName': HostName}
    
    
"""
getOSver: 函数
fn : 文件名参数
功能: 打开fn 文件,读取操作系统版本信息
"""
def  getOSver(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(':')[1]
                 model_name = model_name.split()[0] + ' ' + model_name.split()[-1]
    dic['cpu_num'] = num
    dic['cpu_model'] = model_name
    return dic
    

"""
getMemory: 函数
fn : 文件名参数
功能: 打开fn 文件,读取系统MemTotal 数值
"""
def  getMemory(fn):
    with open(fn) 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__':
    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

测试一下结果:

wKioL1gTfQrikAyxAABGW4gMleI406.png在字典里保存了,文章开头列出所需的9个主机信息。


3. 总结:

还是熟悉读取文件的方法: 对字符切割,对列表的操作等等。如果需要读取其他的主机信息,再增加函数就可以了。




阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯