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，規避了以前方法的限制。在處理擁有對這些流的本地引用的函數時，該技術被證明特別有用。