Bagaimana untuk Mengalihkan Sementara stdout/stderr dalam Python dengan Pengurus Konteks?

Mengubah hala stdout/stderr buat sementara waktu, Dilawati Semula

Walaupun amalan biasa untuk mengubah hala stdout/stderr ke fail luaran, mungkin perlu berbuat demikian buat sementara waktu, dalam skop kaedah tertentu.

Batasan Penyelesaian Semasa

Penyelesaian sedia ada biasanya menggantikan keseluruhan aliran keluaran, meninggalkan salinan tempatan aliran ini tidak terjejas. Ini boleh membawa kepada isu apabila kaedah menggunakan salinan tempatan strim.

Penyelesaian ContextManager

Untuk menangani had ini, kami boleh melaksanakan logik pengalihan menggunakan pengurus konteks:

<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>

Kelas ini membenarkan pengalihan sementara kedua-dua stdout dan stderr.

Contoh Penggunaan

Untuk menunjukkan penggunaan, kita boleh mengubah hala output ke /dev/null dan memerhatikan gelagat:

<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>

Dalam contoh ini, mesej "Anda tidak akan pernah melihat saya" ditindas semasa ubah hala aktif, tetapi boleh dilihat selepas itu, mengesahkan bahawa ubah hala adalah sementara dan hanya mempengaruhi skop dalam pengurus konteks.

Atas ialah kandungan terperinci Bagaimana untuk Mengalihkan Sementara stdout/stderr dalam Python dengan Pengurus Konteks?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!