首頁  >  文章  >  後端開發  >  如何在Python方法中暫時重定向stdout/stderr而不影響本地流副本?

如何在Python方法中暫時重定向stdout/stderr而不影響本地流副本?

Barbara Streisand
Barbara Streisand原創
2024-10-27 00:12:30235瀏覽

How to Temporarily Redirect stdout/stderr in Python Methods Without Affecting Local Stream Copies?

在方法中暫時重定向stdout/stderr:現有解決方案不完美

Python 中暫時重定向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

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>

在此範例中,RedirectStdStreams 類別旨在將stdout 和stderr 重定向到指定的目的地(例如, /dev/null) 在定義的上下文中。這種方法可以有效地重定向流,而不會中斷本地流副本。

以上是如何在Python方法中暫時重定向stdout/stderr而不影響本地流副本?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

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