检索 subprocess.call() 的输出

Question

如何使用 subprocess.call() 获取进程运行的输出？

将 StringIO.StringIO 对象传递给 stdout 会出现以下错误：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Library/Frameworks/Python.framework/Versions/2.6/lib/python2.6/subprocess.py", line 444, in call
    return Popen(*popenargs, **kwargs).wait()
  File "/Library/Frameworks/Python.framework/Versions/2.6/lib/python2.6/subprocess.py", line 588, in __init__
    errread, errwrite) = self._get_handles(stdin, stdout, stderr)
  File "/Library/Frameworks/Python.framework/Versions/2.6/lib/python2.6/subprocess.py", line 945, in _get_handles
    c2pwrite = stdout.fileno()
AttributeError: StringIO instance has no attribute 'fileno'

Answer 1

如果您使用的是 Python 2.7 或更高版本，则可以使用 subprocess.check_output 基本上完全符合您的要求（它返回 标准输出 作为字符串）。

一个简单的示例（Linux 版本；请参阅注释）：

import subprocess

print subprocess.check_output(["ping", "-c", "1", "8.8.8.8"])

请注意 ping命令使用 Linux 表示法（-c 表示计数）。如果您在 Windows 上尝试此操作，请记住将其更改为 -n 以获得相同的结果。

如下评论所示，您可以在另一个答案中找到更详细的解释。

Answer 2

subprocess.call() 的输出只能重定向到文件。

您应该使用 subprocess.Popen() 来代替。然后，您可以传递 stderr、stdout 和/或 stdin 参数的 subprocess.PIPE 并使用 communicate() 方法从管道中读取：

from subprocess import Popen, PIPE

p = Popen(['program', 'arg1'], stdin=PIPE, stdout=PIPE, stderr=PIPE)
output, err = p.communicate(b"input data that is passed to subprocess' stdin")
rc = p.returncode

原因是subprocess.call() 使用的类文件对象必须有一个真实的文件描述符，从而实现 fileno() 方法。仅使用任何类似文件的对象并不能解决问题。

请参阅此处了解更多信息。

Answer 3

对于 Python 3.5 或更高版本，建议您使用 run 函数子流程模块。这将返回一个 CompletedProcess 对象，您可以从中轻松获取输出和返回代码。

from subprocess import PIPE, run

command = ['echo', 'hello']
result = run(command, stdout=PIPE, stderr=PIPE, universal_newlines=True)
print(result.returncode, result.stdout, result.stderr)

Answer 4

我有以下解决方案。它捕获执行的外部命令的退出代码、标准输出和标准错误：

import shlex
from subprocess import Popen, PIPE

def get_exitcode_stdout_stderr(cmd):
    """
    Execute the external command and get its exitcode, stdout and stderr.
    """
    args = shlex.split(cmd)

    proc = Popen(args, stdout=PIPE, stderr=PIPE)
    out, err = proc.communicate()
    exitcode = proc.returncode
    #
    return exitcode, out, err

cmd = "..."  # arbitrary external command, e.g. "python mytest.py"
exitcode, out, err = get_exitcode_stdout_stderr(cmd)

我还有一篇关于它的博客文章 此处。

编辑：该解决方案已更新为不需要写入临时文件的较新解决方案。文件。

Answer 5

我最近刚刚弄清楚如何做到这一点，这是我当前项目中的一些示例代码：

#Getting the random picture.
#First find all pictures:
import shlex, subprocess
cmd = 'find ../Pictures/ -regex ".*\(JPG\|NEF\|jpg\)" '
#cmd = raw_input("shell:")
args = shlex.split(cmd)
output,error = subprocess.Popen(args,stdout = subprocess.PIPE, stderr= subprocess.PIPE).communicate()
#Another way to get output
#output = subprocess.Popen(args,stdout = subprocess.PIPE).stdout
ber = raw_input("search complete, display results?")
print output
#... and on to the selection process ...

现在，您将命令的输出存储在变量“output”中。 “stdout = subprocess.PIPE”告诉类从 Popen 中创建一个名为“stdout”的文件对象。据我所知，communicate() 方法只是作为一种便捷的方式来返回输出的元组以及您运行的进程中的错误。此外，该进程在实例化 Popen 时运行。

Answer 6

关键是使用函数 subprocess.check_output

例如，以下函数捕获进程的 stdout 和 stderr 并返回它们以及调用是否成功。它与 Python 2 和 3 兼容：

from subprocess import check_output, CalledProcessError, STDOUT

def system_call(command):
    """ 
    params:
        command: list of strings, ex. `["ls", "-l"]`
    returns: output, success
    """
    try:
        output = check_output(command, stderr=STDOUT).decode()
        success = True 
    except CalledProcessError as e:
        output = e.output.decode()
        success = False
    return output, success

output, success = system_call(["ls", "-l"])

如果您想将命令作为字符串而不是数组传递，请使用此版本：

from subprocess import check_output, CalledProcessError, STDOUT
import shlex

def system_call(command):
    """ 
    params:
        command: string, ex. `"ls -l"`
    returns: output, success
    """
    command = shlex.split(command)
    try:
        output = check_output(command, stderr=STDOUT).decode()
        success = True 
    except CalledProcessError as e:
        output = e.output.decode()
        success = False
    return output, success

output, success = system_call("ls -l")

Answer 7

在 Ipython shell 中：

In [8]: import subprocess
In [9]: s=subprocess.check_output(["echo", "Hello World!"])
In [10]: s
Out[10]: 'Hello World!\n'

基于 sargue 的回答。归功于萨尔格。