我想要一个匹配subprocess.check_call
接口的好方法 - 即,当它失败时抛出CalledProcessError
,是同步的,& c - 但不是返回返回代码命令(如果它甚至这样做)返回程序的输出,只有stdout或(stdout,stderr)的元组。
有人有方法可以做到吗?
答案 0 :(得分:23)
from subprocess import check_output as qx
import subprocess
def check_output(*popenargs, **kwargs):
if 'stdout' in kwargs:
raise ValueError('stdout argument not allowed, it will be overridden.')
process = subprocess.Popen(stdout=subprocess.PIPE, *popenargs, **kwargs)
output, unused_err = process.communicate()
retcode = process.poll()
if retcode:
cmd = kwargs.get("args")
if cmd is None:
cmd = popenargs[0]
raise subprocess.CalledProcessError(retcode, cmd, output=output)
return output
class CalledProcessError(Exception):
def __init__(self, returncode, cmd, output=None):
self.returncode = returncode
self.cmd = cmd
self.output = output
def __str__(self):
return "Command '%s' returned non-zero exit status %d" % (
self.cmd, self.returncode)
# overwrite CalledProcessError due to `output` keyword might be not available
subprocess.CalledProcessError = CalledProcessError
另请参阅Capturing system command output as a string了解可能check_output()
实施的另一个示例。
答案 1 :(得分:1)
我无法在评论中进行格式化,因此这是对J.F. Sebastian的答案的回应
我发现这非常有用,所以我想我会加入这个答案。我希望能够在不检查版本的情况下无缝地在代码中工作。这就是我做的......
我将上面的代码放入名为' subprocess_compat.py'的文件中。然后在我使用子进程的代码中。
import sys
if sys.version_info < (2, 7):
import subprocess_compat
subprocess.check_output = subprocess_compat.check_output
现在我可以在代码中的任何地方调用&sub;。process_check_output&#39;使用我想要的参数,无论我使用哪种版本的python,它都能正常工作。
答案 2 :(得分:-1)
此函数以字符串列表的形式返回终端输出。
import subprocess,sys
def output(cmd):
cmd_l = cmd.split()
output = subprocess.Popen(cmd_l, stdout=subprocess.PIPE).communicate()[0]
output = output.decode("utf-8")
return (output)