Cara Menangkap Output Program dengan Berkesan dalam Python: Melangkaui Penyelesaian Asas

Merakam Output Program: Melangkaui Penyelesaian Naif

Dalam skrip Python, menangkap output program untuk pemprosesan selanjutnya adalah keperluan biasa. Walaupun penyelesaian naif mungkin kelihatan mudah, mereka sering gagal. Pertimbangkan skrip berikut yang menulis kepada stdout:

# writer.py
import sys

def write():
    sys.stdout.write("foobar")

Percubaan untuk menangkap output menggunakan kod berikut gagal:

# mymodule.py
from writer import write

out = write()
print(out.upper())

Untuk menangkap output dengan berkesan, penyelesaian yang lebih mantap diperlukan . Satu pendekatan melibatkan pengubahsuaian strim stdout sistem:

import sys
from cStringIO import StringIO

# Redirect stdout to a StringIO object
backup = sys.stdout
sys.stdout = StringIO()

# Perform the write operation
write()

# Retrieve and restore stdout
out = sys.stdout.getvalue()
sys.stdout.close()
sys.stdout = backup

# Process the captured output
print(out.upper())

Pengurus Konteks untuk Python 3.4 :

Untuk Python 3.4 dan yang lebih baru, penyelesaian yang lebih ringkas dan ringkas tersedia menggunakan pengurus konteks contextlib.redirect_stdout:

from contextlib import redirect_stdout
import io

f = io.StringIO()
# Redirect stdout to f using the context manager
with redirect_stdout(f):
    help(pow)

# Retrieve captured output from f
s = f.getvalue()

Pendekatan elegan ini memudahkan proses menangkap output, menjadikannya lebih mudah dikendalikan dalam skrip Python anda.

Atas ialah kandungan terperinci Cara Menangkap Output Program dengan Berkesan dalam Python: Melangkaui Penyelesaian Asas. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!