Bagaimana untuk Menangkap Output stdout Skrip dalam Python?

Merakam Output stdout daripada Skrip dalam Python

Apabila bekerja dengan skrip luaran, selalunya menjadi perlu untuk menangkap output stdout mereka untuk pemprosesan selanjutnya. Pendekatan biasa ialah menggunakan fungsi write() untuk mencetak output, yang kemudiannya boleh disimpan dalam pembolehubah. Walau bagaimanapun, pendekatan ini tidak selalu berfungsi, seperti yang ditunjukkan oleh skrip berikut:

<code class="python"># module writer.py
import sys

def write():
    sys.stdout.write("foobar")</code>

Untuk menangkap output fungsi write(), penyelesaian yang melibatkan modul cStringIO telah dicadangkan:

<code class="python">import sys
from cStringIO import StringIO

# setup the environment
backup = sys.stdout

# ####
sys.stdout = StringIO()     # capture output
write()
out = sys.stdout.getvalue() # release output
# ####

sys.stdout.close()  # close the stream 
sys.stdout = backup # restore original stdout

print out.upper()   # post processing</code>

Walau bagaimanapun, terdapat cara yang lebih cekap dan Pythonic untuk mencapai matlamat ini:

Menggunakan contextlib.redirect_stdout (Python 3.4 )

Kaedah ini menggunakan contextlib pengurus konteks .redirect_stdout untuk mengubah hala output stdout buat sementara waktu ke objek StringIO:

<code class="python">from contextlib import redirect_stdout
import io

f = io.StringIO()
with redirect_stdout(f):
    help(pow)
s = f.getvalue()</code>

Menggunakan io.StringIO secara langsung

Pendekatan lain ialah dengan mencipta objek StringIO secara terus dan halakan output stdout secara manual kepadanya:

<code class="python">import io

output = io.StringIO()
sys.stdout = output
write()
sys.stdout.seek(0)
captured_output = output.read()</code>

Dengan melaksanakan kaedah ini, pembangun boleh menangkap dan memproses output skrip dengan berkesan tanpa menggunakan penyelesaian yang rumit.

Atas ialah kandungan terperinci Bagaimana untuk Menangkap Output stdout Skrip dalam Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!