python - 子进程检查输出错误

Question

我正在尝试使用下面的脚本检查 check_output 的用法并遇到编译错误，我哪里出错了？

import os
import subprocess
from subprocess import check_output

#result = subprocess.check_output(['your_program.exe', 'arg1', 'arg2'])
SCRIPT_ROOT=subprocess.check_output(["pwd","shell=True"])
print SCRIPT_ROOT

def main ():
    pass

if __name__ == '__main__':
    main()

Traceback (most recent call last):
  File "test.py", line 3, in <module>
    from subprocess import check_output
ImportError: cannot import name check_output

Answer 1

check_output 已在 Python 2.7 中引入。如果您使用的是早期版本的 python，它就不会存在。

另一种方法是使用 Popen。

output = subprocess.Popen(['pwd'], stdout=subprocess.PIPE).communicate()[0]

可以找到这方面的证明 here .

New function: the subprocess module’s check_output() runs a command with a specified set of arguments and returns the command’s output as a string when the command runs without error, or raises a CalledProcessError exception otherwise.

替代品演示。

import subprocess
cmd = subprocess.Popen(['pwd'], stdout=subprocess.PIPE)
output = cmd.communicate()[0]
print cmd.returncode
print output

输出

> python p.py
/Users/vlazarenko/tests

唯一真正的区别是 Popen 在命令返回非零代码时不会抛出异常。