首頁 >後端開發 >Python教學 >如何在 Python 方法中暫時重定向 stdout 和 stderr?

如何在 Python 方法中暫時重定向 stdout 和 stderr?

DDD
DDD原創
2024-10-28 12:09:01549瀏覽

How to Temporarily Redirect stdout and stderr in Python Methods?

暫時重定向 stdout 和 stderr

是否可以暫時重定向 stdout 和 stderr,特別是在 Python 中的方法期間?

現有解決方案的陷阱:

不幸的是,CurrentUserSolutions 並沒有真正重定向流;相反,他們只是完全替換它們。但是,如果方法擁有任一流的本地副本(例如,由於其用作參數),則此方法會失敗。

基於上下文管理器的解決方案:

將重定向邏輯封裝在上下文管理器中可以解決此問題。以下是一個範例:

<code class="python">import os
import sys

class RedirectStdStreams:
    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

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>

如範例所示使用時,此情境管理器會暫時將stdout 和stderr 重新導向到/dev/null,確保來自“You'll never see me”行的任何輸出當重定向處於活動狀態時,它被有效隱藏。

以上是如何在 Python 方法中暫時重定向 stdout 和 stderr?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn