禁止从 Python 函数打印 stdout / stderr

我有一个 Python 脚本，它使用我的雇主提供的一些封闭式 Python 函数（即我无法编辑这些函数）。当我调用这些函数时，它们会将输出打印到我想禁止的 Linux 终端。我尝试通过重定向 stdout / stderr；

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

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

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

UPDATE:

我最终采用了下面所选答案中概述的方法，并编写了一个上下文管理器来抑制 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 and redirect_stderr。我们只需要将这两个内置上下文管理器组合到我们的自定义上下文管理器中，这可以使用以下命令轻松完成Martijn 的回答中的好模式在这里。将两个输出重定向到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'

到航站楼。未处理的异常永远不应该被忽视。