如何動態重定向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中文網其他相關文章！