有很多需求需要在Python中执行shell命令、启动子进程,并捕获命令的输出和退出状态码,类似于Java中的Runtime类库。
subprocess模块的使用:
Python使用最广泛的是标准库的subprocess模块,用来替换os.system(),os.spawn*(),os.popen*()和commands.*等模块与函数。
使用subprocess最简单的方式就是用它提供的便利函数,call,check_all与check_output,当便利函数满足不了要求再使用Popen类。
1. call
subprocess(args,*,stdin=None,stout=None,stderr=None,shell=False)
shell=True,Python会先运行一个shell,再用shell解释字符串,而不是传递一个列表。
2. check_call
check_all与call类似,只是遇到异常情况返回的形式不同,它会抛出subprocess.CalledProcessError异常
3. check_output
这个便利函数是使用最多了,它可以获取命令的结果,而不是退出状态码
如果想要捕捉退出状态码,可以通过抛出的subprocess.CalledProcessError异常
import subprocess
try:
output = subprocess.check_output('ls /zz',shell=True)
except subprocess.CalledProcessError as e:
output = e.output
code = e.returncode
print (code,output)
执行结果:
如果想捕获命令的错误输出,需将错误输出重定向到标准输出
subprocess.check_output(['cmd','arg1','arg2'],stderr=subprocess.STDOUT)
4. 使用Popen
下面的exexute_cmd函数对Popen,进行了封装,执行成功,返回标准输出和状态码,失败是返回状态码和错误输出。
#!/usr/bin/python
#coding=utf8
import subprocess def execute_cmd(cmd):
p = subprocess.Popen(cmd,
shell=True,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
stdout, stderr = p.communicate()
if p.returncode != 0:
return p.returncode, stderr
return p.returncode, stdout if __name__=='__main__':
cmd='ls /u01'
returncode,out=execute_cmd(cmd)
if returncode != 0:
raise SystemExit('execute {0} err :{1}'.format(cmd,out))
else:
print("execute command ({0} sucessful)".format(cmd))
目录不存时: