Python 子进程：问题捕获错误，其中部分命令在管道在一起时失败

Question

在 bash 中执行一个简单的命令

cmd='ls -l | wc -l'

我知道我们可以使用 subprocess call/check_output/communicate 以多种方式运行此命令。如果初始命令由于某种原因不起作用或失败，则会出现问题。就像 [用 lsx 替换 ls]。

cmd='lsx -l | wc -l'

在这种情况下，我们如何捕获错误，或者我们只需要处理输出就可以了？这是我尝试过的。

import subprocess
>>> subprocess.call('lsx -l | wc -l', shell=True)
/bin/sh: lsx: command not found
       0
0
>>> subprocess.check_output('lsx -l | wc -l', shell=True)
/bin/sh: lsx: command not found
b'       0\n'

上面两条命令的错误码好像还是0。 我也试过https://docs.python.org/3.5/library/subprocess.html#replacing-shell-pipeline，但不知道如何获取第一个进程的错误代码。

Answer 1

您可以将stderr 分配给PIPE。考虑这个例子：

>>> from subprocess import PIPE, Popen
>>> sub = Popen('lsx -l | wc -l', shell=True, stderr=PIPE, stdout=PIPE)
>>> output, error_output = sub.communicate()
>>> error_output
b'/bin/sh: 1: lsx: not found\n'
>>> output
b'0\n'
>>> sub = Popen('ls -l | wc -l', shell=True, stderr=PIPE, stdout=PIPE)
>>> output, error_output = sub.communicate()
>>> error_output
b''

Answer 2

如果您指定bash 而不是sh，您可以设置pipefail 选项以在管道的任何部分失败时返回非零退出状态：

subprocess.check_output(['bash', '-c', 'set -o  pipefail; lsx -l | wc -l'])

---

也就是说，您当然可以完全避免使用shell=True：

import subprocess
try:
    p1 = subprocess.Popen(['ls', '--invalid-argument'], stdout=subprocess.PIPE)
    p2 = subprocess.Popen(['wc', '-l'], stdin=p1.stdout, stdout=subprocess.PIPE)
    wc_stdout = p2.communicate()[0]
    if p1.wait() != 0 or p2.wait() != 0:
        raise RuntimeError("Something failed!")
except FileNotFoundError as ex:
    raise RuntimeError("Something failed, because we couldn't find an executable!") from ex