你好
我是你们的雨嘉叔

14-Python Linux系统管理之Python中执行外部命令

一、简介

有很多需求需要在Python中执行shell命令、启动子进程,并捕获命令的输出和退出状态码,类似于Java中的Runtime类库。本文将介绍subprocess模块的定位,然后介绍subprocess模块提供的便利函数,最后介绍Popen这个类的使用方法。

二、subprocess模块的使用

1、call

call函数的定义如下:

subprocess.call(args,*,stdin=None,stout=None,stderr=None,shell=False)

call函数将允许由args参数制定的命令直到命令结束。call函数的返回值是命令的退出状态码,工程师可以通过退出状态码判断命令是否执行成功。如下所示:

In [1]: import subprocess
In [2]: subprocess.call([‘ls’,’-l’])

total 8

-rw-r–r–. 1 root root 0 Sep 24 10:52 1.txt

-rw-r–r–. 1 root root 0 Sep 24 10:52 2.txt

-rw-r–r–. 1 root root 3736 Sep 24 10:51 demo.tar.gz

-rw-r–r–. 1 root root 929 Sep 24 10:52 demo.zip

Out[2]: 0

In [3]: subprocess.call(“exit 1”,shell = True)

Out[3]: 1


注:如果设置shell=True,Python会先运行一个shell,再用shell解释字符串,而不是传递一个列表。

2、check_call

check_all与call类似,只是遇到异常情况返回的形式不同。如果命令执行成功,下面会返回0;如果命令执行失败,它会抛出subprocess.CalledProcessError异常,相关案例如下所示:

In [5]: subprocess.check_call([“ls”,”-l”])
total 8

-rw-r–r–. 1 root root 0 Sep 24 10:52 1.txt

-rw-r–r–. 1 root root 0 Sep 24 10:52 2.txt

-rw-r–r–. 1 root root 3736 Sep 24 10:51 demo.tar.gz

-rw-r–r–. 1 root root 929 Sep 24 10:52 demo.zip

Out[5]: 0

In [6]: subprocess.check_call(“exit 1”,shell = True)

—————————————————————————

CalledProcessError Traceback (most recent call last)

<ipython-input-6-f13e6678cc55> in <module>()

—-> 1 subprocess.check_call(“exit 1”,shell = True)

/usr/local/lib/python3.7/subprocess.py in check_call(*popenargs, **kwargs)

326 if cmd is None:

327 cmd = popenargs[0]

–> 328 raise CalledProcessError(retcode, cmd)

329 return 0

330

CalledProcessError: Command ‘exit 1’ returned non-zero exit status 1.


3、check_output

从上面的输出结果可以看到,call和check_call函数直接将命令的输出结果输出到命令行终端,这种返回结果的形式很有可能是你不想要的。在实际工作过程中,一般会对获取命令的结果进行进一步处理,或者将命令的输出打印到日志文件中。如下所示:

In [1]: import subprocess
In [2]: output = subprocess.check_output([‘df’,’-h’])

In [3]: print(output)

b’Filesystem Size Used Avail Use% Mounted on\n/dev/mapper/centos-root 50G 1.7G 49G 4% /\ndevtmpfs 1.9G 0 1.9G 0% /dev\ntmpfs 1.9G 0 1.9G 0% /dev/shm\ntmpfs 1.9G 12M 1.9G 1% /run\ntmpfs 1.9G 0 1.9G 0% /sys/fs/cgroup\n/dev/mapper/centos-home 106G 246M 105G 1% /home\n/dev/sda1 1014M 184M 831M 19% /boot\ntmpfs 378M 0 378M 0% /run/user/0\n’

In [4]: lines = output.split(b’\n’)

In [5]: lines

Out[5]:

[b’Filesystem Size Used Avail Use% Mounted on’,

b’/dev/mapper/centos-root 50G 1.7G 49G 4% /’,

b’devtmpfs 1.9G 0 1.9G 0% /dev’,

b’tmpfs 1.9G 0 1.9G 0% /dev/shm’,

b’tmpfs 1.9G 12M 1.9G 1% /run’,

b’tmpfs 1.9G 0 1.9G 0% /sys/fs/cgroup’,

b’/dev/mapper/centos-home 106G 246M 105G 1% /home’,

b’/dev/sda1 1014M 184M 831M 19% /boot’,

b’tmpfs 378M 0 378M 0% /run/user/0′,

b”]

In [6]: for line in lines[1:-1]:

…: if line:

…: print(line.split()[-2])

…:

b’4%’

b’0%’

b’0%’

b’1%’

b’0%’

b’1%’

b’19%’

b’0%’



如果想要捕捉退出状态码,可以通过抛出的subprocess.CalledProcessError异常。

import subprocess
try:

output = subprocess.check_output(‘cmd’,’arg1′ ,’arg2′)

except subprocess.CalledProcessError as e:

output = e.output

code = e.returncode

默认情况下,check_output命令只会捕捉命令的标准输出。如果想捕捉命令的错误输出,需要将错误输出重定向到标准输出。如下所示:

output = subprocess.check_output([‘cmd’,’arg1′ ,’arg2′],stderr= subprocess.STDOUT)
赞(2)
未经允许不得转载:嘉叔的博客 » 14-Python Linux系统管理之Python中执行外部命令
分享到: 更多 (0)

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址