位置：首页-资讯-后端开发

人生苦短我用python（04），Pyt

2023-01-31 02:11

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

文章内容为原创，欢迎转载请注明出处

作者： EflyPro->晦明禅师

文章来源：公众号“睿江云计算”

今天为大家介绍如何使用python的commands模块调用linux shell命令，同时附上一个实际例子给大家参考。

commands库介绍

commands库很简单，只有3个函数，其中一个是不建议使用的，所以今天就跟大家介绍2个函数

commands.getstatusoutput(cmd)
#这个函数执行cmd传递进来的命令，然后返回一个2-tuple (status, output)，其中status是执行cmd的返回结果，output是执行cmd的输出内容。

commands.getoutput(cmd)
#这个是上面函数的简化版，只返回执行cmd的输出内容output

实现一个获取本机磁盘分区使用率的脚本

平时我们可以使用 cat /proc/mounts 命令来查看挂载的设备分区列表

其中我们只统计/dev/开头的分区。至于如何获取分区的磁盘信息，则用os模块的 os.statvfs 函数，具体函数用法请看代码注释。

#!/usr/bin/env python
import commands
import os

def get_disk_uper(path):
    #计算磁盘分区使用率，
    #其中f_blocks是分区的总块数，换算成G
    #f_bavail是分区的可用块数，换算成G
    info = os.statvfs(path)
    total = round(float(info.f_bsize * info.f_blocks)/1024/1024/1024, 2)
    free = round(float(info.f_bsize * info.f_bavail)/1024/1024/1024, 2)
    return 100 - int(float(free) / float(total) * 100)

def get_disk_info():    
    disk_info = {}
    (status, output) = commands.getstatusoutput('cat /proc/mounts')
    #status是执行命令的返回值，默认情况下0是成功，非0是失败
    if status != 0:
        print ststus, output
        return
    #分析返回内容，对返回内容以换行符作为分割
    for line in output.split("\n"):
        #只对/dev/ 分区设备获取使用率
        if not line.startswith('/dev/'):
            continue
        #对一行内容按照空格或者tab进行分割，以提取分区字段名称
        temp = line.split()
        if len(temp) < 2:
            continue
        #获取分区名
        dev_name = temp[1]
        dev_used = get_disk_uper(dev_name)
        print "[%s] = %d%%"%(dev_name, dev_used)

if __name__ == "__main__":
    get_disk_info()

执行返回内容如下