Bagaimana untuk Menangkap Output Standard dalam Skrip Python?

Menangkap Output Standard Skrip

Dalam senario di mana skrip melaksanakan tugas tertentu, seperti menulis data ke output standardnya, menangkap output tersebut boleh menjadi penting untuk pemprosesan selanjutnya. Pendekatan biasa melibatkan percubaan untuk menyimpan output dalam pembolehubah, seperti yang digambarkan dalam contoh berikut:

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

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

<code class="python"># mymodule.py
from writer import write

out = write()
print(out.upper())</code>

Walau bagaimanapun, pendekatan ini gagal menangkap output skrip. Penyelesaian alternatif, menggunakan objek StringIO dan persediaan persekitaran, berjaya menangkap output:

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

Penyelesaian Python 3.4

Untuk versi Python 3.4 dan ke atas, lebih banyak pendekatan mudah tersedia menggunakan pengurus konteks contextlib.redirect_stdout:

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

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

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