简介

这个脚本通过获取/etc/os-release文件中的系统版本信息，通过Linux系统中的/proc目录中的cpuinfo和meminfo文件获取cpu和内存的信息，在使用系统内置的os和platform库获取系统名称和主机名，内核版本等信息。并使用prettytable库将结果在命令行中形成表格进行输出。本脚本使用与Linux系统，并在Debian和ReaHat系的发行版本下测试运行正常。

下面是脚本运行的结果：

完整的代码

下面是全部的代码：

import os
import platform
import prettytable

def systemInfo():
    system = os.popen("cat /etc/os-release|grep -E \"^NAME\"|awk -F \"\\\"\" '{print $2}'").readline()
    sysversion = os.popen("cat /etc/os-release|grep -E \"^VERSION\"|awk -F \"\\\"\" '{print $2}'").readline()
    hostname = platform.node()
    version = platform.release()
    framework = platform.machine()
    cpumodel = os.popen("cat /proc/cpuinfo |grep \"model name\"|awk -F : '{print $2}'").readline()
    cpucores = os.popen('grep "processor" /proc/cpuinfo |wc -l').readline()
    memory = os.popen("cat /proc/meminfo |grep MemTotal|awk '{print $2$3}'").readline()
    swapmemory = os.popen("cat /proc/meminfo |grep SwapTotal|awk '{print $2$3}'").readline()

    table = prettytable.PrettyTable()
    table.field_names = ['名称','值']
    table.add_row(['系统名称',system.replace('\n','')])
    table.add_row(['系统版本',sysversion.replace('\n','')])
    table.add_row(['主机名',hostname])
    table.add_row(['内核版本',version])
    table.add_row(['CPU架构',framework])
    table.add_row(['CPU型号',cpumodel.replace('\n','')])
    table.add_row(['CPU核心数',cpucores.replace('\n','')])
    table.add_row(['物理内存',memory.replace('\n','')])
    table.add_row(['交换空间',swapmemory.replace('\n','')])

    print(table)
    
if __name__ == '__main__':
    systemInfo()

代码解析

第1~3行代码

import os
import platform
import prettytable

这是三个需要用到的库的导入，os和platform是系统自带的库，不用安装，直接import导入就可以。prettytable是一个第三方的库，需要安装才可以使用，否则脚本会报错。安装的命令是pip install prettytable。

第30~31行代码

if __name__ == '__main__':
    systemInfo()

可以理解为运行这个文件时的主入口，在python中，文件独立运行时，它的魔法函数 __name__ 会返回 __main__ 而在该文件被其他文件引用调用的时候，则不是返回 __main__ 。这个入口的意思就是，运行这个脚本文件的时候，就执行 systemInfo() 函数。函数在上面的第5~28行中已经定义了。

第5行代码

def systemInfo():

定义了一个名为systemInfo的函数，这个函数的内容都必须是在缩进（默认为四个空格）里面，如果没有写在缩进里面的内容，会被解析器认为是函数外的内容。下面6~28行的内容，就是因为行首有四个空格的缩进，所以是函数systemInfo的内容。

第6~7行代码

    system = os.popen("cat /etc/os-release|grep -E \"^NAME\"|awk -F \"\\\"\" '{print $2}'").readline()
    sysversion = os.popen("cat /etc/os-release|grep -E \"^VERSION\"|awk -F \"\\\"\" '{print $2}'").readline()

python中，要直接执行命令行命令可以使用os库中的popen()函数，然后在里面写命令行的命令就行。通过readline()可以获取到执行后的结果 Linux系统里面的/etc/os-release文件记录了系统的信息，包括系统名称、系统版本号、该系统的官方网址等信息。 上面的代码，大部分的功能是直接通过shell命令来进行查看文件，查找出NAME、VERSION开头的行，再将得到的信息用awk分隔取第二列的内容。python指负责获取结果后赋值给变量，system的数据是获取系统名称，sysversion的数据是获取系统的版本号。

第8~10行代码

    hostname = platform.node()
    version = platform.release()
    framework = platform.machine()

系统的主机名、系统内核版本号、CPU的架构，可以直接使用python内置的platform库的node()、release()、machine()三个函数来获得，就可以不用去查看系统的文件。

第11~14行代码

    cpumodel = os.popen("cat /proc/cpuinfo |grep \"model name\"|awk -F : '{print $2}'").readline()
    cpucores = os.popen('grep "processor" /proc/cpuinfo |wc -l').readline()
    memory = os.popen("cat /proc/meminfo |grep MemTotal|awk '{print $2$3}'").readline()
    swapmemory = os.popen("cat /proc/meminfo |grep SwapTotal|awk '{print $2$3}'").readline()

获取CPU和内存的信息是查看/proc目录下的对应cpuinfo和meminfo文件对应的信息行。cpu的类型直接看model name即可，但是多少核心数需要自己统计，因为cpuinfo里面。一个核心或者线程会记录一份信息，从processor 0开始，一直累加下去。所以统计有多少行processor，就可以知道有多少个核心。meminfo里面记录的内存大小的计数单位是kb。

第16~24行代码

    table = prettytable.PrettyTable()
    table.field_names = ['名称','值']
    table.add_row(['系统名称',system.replace('\n','')])
    table.add_row(['系统版本',sysversion.replace('\n','')])
    table.add_row(['主机名',hostname])
    table.add_row(['内核版本',version])
    table.add_row(['CPU架构',framework])
    table.add_row(['CPU型号',cpumodel.replace('\n','')])
    table.add_row(['CPU核心数',cpucores.replace('\n','')])
    table.add_row(['物理内存',memory.replace('\n','')])
    table.add_row(['交换空间',swapmemory.replace('\n','')])

赋值一个名为table的变量，这个变量是初始化一个prettytable的表格。第14行，给表格的field_names赋值，可以理解为给这个表格的表格头的列名值，用一个list表示，有多少列就在list中写多少个元素；第15~24行add_row()是将上面获取到的数据组合后逐行插入到表格中，一行数据需要执行一遍，里面的数据同样是写成一个list。每个元素对应上面field_names的列。

第28行代码

print(table)

将上面组合并美化为表格的内容输出到命令行。也就是最开始图片里面那样的输出结果。