使用subprocess代替popen

来源:互联网 发布:用java编写人机猜拳 编辑:程序博客网 时间:2024/05/20 06:28

很多时候,需要用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。


原创粉丝点击