如何在 Python 方法中临时重定向 stdout 和 stderr？

临时重定向 stdout 和 stderr

是否可以临时重定向 stdout 和 stderr，特别是在 Python 中的方法期间？

现有解决方案的陷阱：

不幸的是，CurrentUserSolutions 并没有真正重定向流；相反，他们只是完全替换它们。但是，如果方法拥有任一流的本地副本（例如，由于其用作参数），则此方法会失败。

基于上下文管理器的解决方案：

将重定向逻辑封装在上下文管理器中可以解决此问题。下面是一个示例：

<code class="python">import os
import sys

class RedirectStdStreams:
    def __init__(self, stdout=None, stderr=None):
        self._stdout = stdout or sys.stdout
        self._stderr = stderr or sys.stderr

    def __enter__(self):
        self.old_stdout, self.old_stderr = sys.stdout, sys.stderr
        self.old_stdout.flush(); self.old_stderr.flush()
        sys.stdout, sys.stderr = self._stdout, self._stderr

    def __exit__(self, exc_type, exc_value, traceback):
        self._stdout.flush(); self._stderr.flush()
        sys.stdout = self.old_stdout
        sys.stderr = self.old_stderr

if __name__ == '__main__':

    devnull = open(os.devnull, 'w')
    print('Fubar')

    with RedirectStdStreams(stdout=devnull, stderr=devnull):
        print("You'll never see me")

    print("I'm back!")</code>

如示例中所示使用时，此上下文管理器会临时将 stdout 和 stderr 重定向到 /dev/null，确保来自“You'll never see me”行的任何输出当重定向处于活动状态时，它被有效隐藏。

以上是如何在 Python 方法中临时重定向 stdout 和 stderr？的详细内容。更多信息请关注PHP中文网其他相关文章！