如何动态重定向Python函数中的标准输出和错误流？

Python 中的上下文流重定向

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

需要动态解决方案

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

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

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

<code class="python">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</code>

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

<code class="python">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，从而规避了以前方法的限制。事实证明，这种技术在处理拥有这些流的本地引用的函数时特别有用。

以上是如何动态重定向Python函数中的标准输出和错误流？的详细内容。更多信息请关注PHP中文网其他相关文章！