云服务器

人生苦短我用python[0x02] python执行shell命令(附例子)

2017-06-22 14:53:59 0

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

1.commands库介绍

commands库很简单,只有3个函数,其中一个是不建议使用的,所以今天就跟大家介绍2个函数
commands.getstatusoutput(cmd)
#这个函数执行cmd传递进来的命令,然后返回一个2-tuple (status, output),其中status是执行cmd的返回结果,output是执行cmd的输出内容。

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

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

平时我们可以使用 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%

 

上一篇: 无

微信关注

获取更多技术咨询