Python 中的上下文流重定向

标准输出和错误流（stdout 和 stderr）的重定向在许多场景中证明是有用的。然而，当函数持有对这些流的内部引用时，传统方法通常会出现不足。

需要动态解决方案

传统的重定向技术，如 sys.stdout，永久重定向流。当方法本质上在内部复制这些变量之一时，就会出现此问题。因此，这些方法无法正确重定向流。

解决方案：上下文管理器扩展

为了有效解决此问题，可以采用上下文管理器方法。此方法涉及将重定向逻辑包装在上下文管理器中：

import os
import sys

class RedirectStdStreams(object):
    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

通过利用此上下文管理器，您可以无缝地重定向上下文块内的流：

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

print('Fubar')

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

print("I'm back!")

结论

提供的解决方案利用上下文管理器模式临时重定向 stdout 和 stderr，规避了以前方法的限制。在处理拥有对这些流的本地引用的函数时，该技术被证明特别有用。