歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
您现在的位置: Linux教程網 >> UnixLinux >  >> Linux編程 >> Linux編程

Python腳本獲取Linux系統信息

其實獲取Linux系統信息,如果使用python的psutil的模塊,更加的簡潔好用。自己寫的這個腳本算是一個python練習。

#!/usr/bin/env python2.7
#www.linuxidc.com
# -*- coding:utf-8 -*-
import os
import subprocess
import re
import hashlib
#對字典取子集
def sub_dict(form_dict, sub_keys, default=None):
    return dict([(k, form_dict.get(k.strip(), default)) for k in sub_keys.split(',')])
#讀取cpuinfo信息
# dmidecode -t 4
def read_cpuinfo():
    cpu_stat = []
    with open('/proc/cpuinfo', 'r') as f:
        data = f.read()
        for line in data.split('\n\n'):
            cpu_stat.append(line)
    return cpu_stat[-2]
#讀取fdisk信息
def read_fdisk():
    p = subprocess.Popen('fdisk -l', stdout=subprocess.PIPE, shell=True)
    out = p.communicate()[0]
    info = []
    for i in out.split('\n\n'):
        for x in i.splitlines():
            if x:
                info.append(x)
    return info
#讀取dmidecode信息
def read_dmidecode():
    p = subprocess.Popen('dmidecode -t 1', stdout=subprocess.PIPE, shell=True)
    return p.communicate()[0]
#讀取ifconfig信息
def read_ifconfig():
    p = subprocess.Popen('ifconfig', stdout=subprocess.PIPE, shell=True)
    return p.communicate()[0]
#返回cpu信息:CPU型號、顆數、核數
def get_cpuinfo(data):
    cpu_info = {}
    for i in data.splitlines():
        k, v = [x.strip() for x in i.split(':')]
        cpu_info[k] = v
       
    cpu_info['physical id'] = str(int(cpu_info.get('physical id')) + 1)
    return sub_dict(cpu_info, 'model name,physical id,cpu cores')
#返回每塊硬盤大小
def get_diskinfo(data):
    disk_info = {}
    m_disk = re.compile(r'^Disk\s/dev')
               
    for i in data:
        if m_disk.match(i):
            i = i.split(',')[0]
            k, v = [x for x in i.split(':')]
            disk_info[k] = v
    return disk_info
#返回硬件信息:品牌、型號
def get_dmiinfo(data):
    dmi_info = {}
    line_in = False
    for line in data.splitlines():
        if line.startswith('System Information'):
            line_in = True
            continue
        if line.startswith('\t') and line_in:
            k, v = [i.strip() for i in line.split(':')]
            dmi_info[k] = v
        else:
            line_in = False
    return sub_dict(dmi_info, 'Manufacturer,Product Name,Serial Number')
#返回網卡及ip信息:網卡、IP址、MAC地址
def get_ipinfo(data):
    data = (i for i in data.split('\n\n') if i and not i.startswith('lo'))
    ip_info = []
    ifname = re.compile(r'(eth[\d:]*|wlan[\d:]*)')
    ipaddr = re.compile(r'(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[0-9]{1,2})(\.(25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[0-9]{1,2})){3}')
    macaddr = re.compile(r'[A-F0-9a-f:]{17}')
    for i in data:
        x = {}
        if ifname.match(i):
            device = ifname.match(i).group()
            x['Adapter'] = device
        if macaddr.search(i):
            mac = macaddr.search(i).group()
            x['MAC'] = mac
        if ipaddr.search(i):
            ip = ipaddr.search(i).group()
            x['IP'] = ip
        else:
            x['IP'] = None
        ip_info.append(x)
    return ip_info
#返回內存及swap大小
def get_meminfo():
    mem_info = {}
    with open('/proc/meminfo', 'r') as f:
        data = f.readlines()
        for i in data:
            k, v = [x.strip() for x in i.split(':')]
            mem_info[k] = int(v.split()[0])
    return sub_dict(mem_info, 'MemTotal,SwapTotal')
#返回操作信息
def get_osinfo():
    os_info = {}
    i = os.uname()
    os_info['os_type'] = i[0]
    os_info['node_name'] = i[1]
    os_info['kernel'] = i[2]
    return os_info
#唯一標識符
def get_indentity(data):
    match_serial = re.compile(r"Serial Number: .*", re.DOTALL)
    match_uuid = re.compile(r"UUID: .*", re.DOTALL)
    if match_serial.search(data):
        serial = match_serial.search(data).group()
    if match_uuid.search(data):
        uuid = match_uuid.search(data).group()
    if serial:
        serial_md5 = hashlib.md5(serial).hexdigest()
        return serial_md5
    elif uuid:
        uuid_md5 = hashlib.md5(uuid).hexdigest()
        return uuid_md5
if __name__ == "__main__":
    ipinfo = get_ipinfo(read_ifconfig())
    dmiinfo = get_dmiinfo(read_dmidecode())
    cpuinfo = get_cpuinfo(read_cpuinfo())
    diskinfo = get_diskinfo(read_fdisk())
    meminfo = get_meminfo()
    osinfo = get_osinfo()
    identity = get_indentity(read_dmidecode())

Copyright © Linux教程網 All Rights Reserved