有很多需求需要在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))
目录不存时: