我有一个 Python 脚本，它使用了我的雇主提供的一些封闭式 Python 函数(即我无法编辑这些函数)。当我调用这些函数时，它们会将输出打印到我想要​​抑制的 linux 终端。我试过重定向 stdout/stderr via;

orig_out = sys.stdout
sys.stdout = StringIO()
rogue_function()
sys.stdout = orig_out

但这无法捕获输出。我认为我通过 Python 调用的函数(上面的 rogue_function())实际上是编译后的 C 代码的包装器，它们实际上是在进行打印。

有谁知道我可以通过函数(以及函数调用的任何子函数)对传递给 stdout/stderr 的任何打印进行“深度捕获”？

更新:

我最终采用了下面所选答案中概述的方法并编写了一个上下文管理器来抑制 stdout 和 stderr:

# Define a context manager to suppress stdout and stderr.
class suppress_stdout_stderr(object):
    '''
    A context manager for doing a "deep suppression" of stdout and stderr in 
    Python, i.e. will suppress all print, even if the print originates in a 
    compiled C/Fortran sub-function.
       This will not suppress raised exceptions, since exceptions are printed
    to stderr just before a script exits, and after the context manager has
    exited (at least, I think that is why it lets exceptions through).      

    '''
    def __init__(self):
        # Open a pair of null files
        self.null_fds =  [os.open(os.devnull,os.O_RDWR) for x in range(2)]
        # Save the actual stdout (1) and stderr (2) file descriptors.
        self.save_fds = [os.dup(1), os.dup(2)]

    def __enter__(self):
        # Assign the null pointers to stdout and stderr.
        os.dup2(self.null_fds[0],1)
        os.dup2(self.null_fds[1],2)

    def __exit__(self, *_):
        # Re-assign the real stdout/stderr back to (1) and (2)
        os.dup2(self.save_fds[0],1)
        os.dup2(self.save_fds[1],2)
        # Close all file descriptors
        for fd in self.null_fds + self.save_fds:
            os.close(fd)

要使用它，您只需:

with suppress_stdout_stderr():
    rogue_function()

这“非常好”。它确实抑制了使我的脚本困惑的流氓函数的打印输出。我在测试它时注意到它允许通过引发的异常以及一些记录器打印，我并不完全清楚为什么。我认为这与 when 这些消息被发送到 stdout/stderr 有关(我认为它发生在我的上下文管理器退出之后)。如果有人能证实这一点，我很想听听细节......

最佳答案

从 python 3.5 开始，我们可以使用 contextlib 中的内置函数(即 redirect_stdout 和 redirect_stderr )以最少的工作完成此操作。我们只需要将这两个内置的上下文管理器组合在我们的自定义上下文管理器中，使用 the nice pattern in Martijn's answer here 即可轻松完成。将两个输出重定向到 os.devnull 应该足够安全和​​便携。

from contextlib import contextmanager,redirect_stderr,redirect_stdout
from os import devnull

@contextmanager
def suppress_stdout_stderr():
    """A context manager that redirects stdout and stderr to devnull"""
    with open(devnull, 'w') as fnull:
        with redirect_stderr(fnull) as err, redirect_stdout(fnull) as out:
            yield (err, out)

请注意，当出现问题时，抑制 stderr 仍会为您提供完整的回溯，这是一件好事:

import sys

def rogue_function():
    print('spam to stdout')
    print('important warning', file=sys.stderr)
    1 + 'a'
    return 42

with suppress_stdout_stderr():
    rogue_function()

当运行上面只打印时

Traceback (most recent call last):
  File "tmp.py", line 20, in <module>
    rogue_function()
  File "foo.py", line 16, in rogue_function
    1 + 'a'
TypeError: unsupported operand type(s) for +: 'int' and 'str'

到终端。不应该忽视未处理的异常。

关于python - 抑制 Python 函数的 stdout/stderr 打印，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/11130156/