Python函数捕获子进程stdout和stderr到日志文件

时间:2017-05-19 14:43:58

标签: python bash subprocess io-redirection tee

这里有很多代码。我会尽量保持这种简洁。

我有一个运行外部程序的python函数,并将stdout和stderr发送到日志文件。

我正在使用doctest来测试这个功能。我需要测试输出捕获功能。下面的代码显示了我编写函数和测试的尝试。测试失败,没有写入日志文件。我不确定问题是在测试中还是在测试中,或者两者兼而有之。建议?

from __future__ import print_function

import subprocess

def run(command_line, log_file):
    """
    # Verify stdout and stderr are both written to log file in chronological order
    >>> run("echo text to stdout; echo text to stderr 1>&2", "log")
    >>> f = open("log"); out = f.read(); f.close()
    >>> print(out.strip())
    text to stdout
    text to stderr
    """
    command_line = "set -o pipefail; " + command_line + " 2>&1 | tee " + log_file

    # Run command. Wait for command to complete. If the return code was zero then return, otherwise raise CalledProcessError
    subprocess.check_call(command_line, shell=True, executable="bash")

测试结果:

$ python -m doctest testclass.py
text to stdout
text to stderr
**********************************************************************
File "testclass.py", line 10, in testclass.run
Failed example:
    print(out.strip())
Expected:
    text to stdout
    text to stderr
Got:
    <BLANKLINE>
**********************************************************************
1 items had failures:
   1 of   3 in testclass.run
***Test Failed*** 1 failures.

1 个答案:

答案 0 :(得分:2)

由于使用subprocess.check_call执行shell=True,使用2 stdout / stderr重定向和tee不是执行命令和捕获输出的最佳方式(实际上它最接近< em>最差方式),我对它失败并不感到惊讶。

我的解决方案是删除set -o pipefail作为初学者(你不需要在这里检查返回代码)并将两个命令包装在括号中,否则重定向/ tee仅适用于最后一个(我还是令人困惑的是,为什么你没有得到任何输出,说实话,但是):

command_line = "(" + command_line + ") 2>&1 | tee " + log_file

如果您必须恢复pipefail,请在括号内执行:

command_line = "(set -o pipefail; " + command_line + ") 2>&1 | tee " + log_file