很多时候,需要用Python来执行外部命令,然后通过管理获取命令的执行结果(不是ret code)。
这时其实有多种选择,os.system当然是不行,因为它直接把结果打印出来了,返回的结果只是一个ret code。这就意味着我们没法通过变量来获取执行结果。
一个可行的方法是使用popen2模块。下面是示例代码:
def execute_cmd(cmd):
stdout, stdin, stderr = popen2.popen3(cmd)
lines = [line.strip() for line in stdout]
return lines
popen2模块下的popen3方法默认接受的就是一个shell命令行,并且能够处理管道符,因此非常好用。
唯一的问题是,这个模块从python2.3开始就被标记为Deprecated。因此我们考虑使用python建议的subprocess模块。
subprocess模块下提供了一个Popen方法来替换原来的popen模块。代码差别不太大:
def execute_cmd(cmd):
stdout = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE).stdout
lines = [line.strip() for line in stdout]
stdout.close()
return lines
这个替换可以去掉了恼人的Deprecated Warning了。但是使用的时候有两点需要注意:
1. shell参数和stdout参数。shell=True表示这是一个shell命令,可能包含了管道操作等。如果不加这个参数,那么执行下列方法就会抛出异常:
stdout = subprocess.Popen('ls -al', stdout=subprocess.PIPE).stdout
这时需要写成:
stdout = subprocess.Popen(['ls','-al'], stdout=subprocess.PIPE).stdout
当然,这时要处理管道就会比较蛋疼了。
而加了shell=True之后,这个问题就没有了。
而stdout=subprocess.PIPE则指示对输出进行重定向,以便我们可以读取执行结果。
2. 读完执行结果之后,注意关闭stdout这个FD。