人生苦短我用python(04),Pyt

2023-01-31 02:01:21 人生 我用 苦短

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

作者: 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()

执行返回内容如下

[/] = 10%
[/boot] = 21%



下期预告 

人生苦短我用python【0x02】yield浅析


相关文章