首頁 > 後端開發 > Python教學 > 如何使用上下文管理器暫時重定向 Python 中的 stdout/stderr?

如何使用上下文管理器暫時重定向 Python 中的 stdout/stderr?

Linda Hamilton
發布: 2024-10-31 00:28:03
原創
678 人瀏覽過

How to Temporarily Redirect stdout/stderr in Python with a Context Manager?

暫時重定向stdout/stderr,重新訪問

雖然將stdout/stderr 重定向到外部檔案是常見的做法,但可能有必要在範圍內臨時這樣做

當前解決方案的限制

現有解決方案通常會替換整個輸出流,而使這些流的本地副本不受影響。當方法使用流的本機副本時,這可能會導致問題。

ContextManager 解決方案

為了解決此限制,我們可以使用 contextmanager 實現重定向邏輯:

<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>
登入後複製

此類允許臨時重定向 stdout 和 stderr。

使用範例

為了示範用法,我們可以將輸出重定向到/dev/null 並觀察行為:

<code class="python">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>
登入後複製

在此範例中,訊息「You 'll never see me」在重定向處於活動狀態時被抑制,但隨後可見,確認重定向是臨時的並且僅影響上下文管理器內的範圍。

以上是如何使用上下文管理器暫時重定向 Python 中的 stdout/stderr?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板