Python でのコンテキスト ストリーム リダイレクト

標準出力およびエラー ストリーム (stdout および stderr) のリダイレクトは、多くのシナリオで役立つことがわかります。ただし、関数がこれらのストリームへの内部参照を保持している場合、従来の方法では不十分なことがよくあります。

動的ソリューションの必要性

sys.stdout などの従来のリダイレクト手法ストリームを永続的にリダイレクトします。この問題は、メソッドが本質的にこれらの変数の 1 つを内部的にコピーする場合に発生します。その結果、これらのメソッドはストリームを適切にリダイレクトできません。

解決策: Context Manager Extension

この問題に効果的に対処するには、コンテキスト マネージャーのアプローチを使用できます。この方法には、コンテキスト マネージャー内でリダイレクト ロジックをラップすることが含まれます:

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 を一時的にリダイレクトし、以前のアプローチの制限を回避します。 。この手法は、これらのストリームへのローカル参照を持つ関数を扱う場合に特に役立ちます。