Comment capturer la sortie d'un script Python à l'aide de la manipulation d'E/S ?

Capturer la sortie d'un script

Supposons que vous disposiez d'un script qui effectue des opérations d'écriture à l'aide de sys.stdout. Une solution initiale pour capturer la sortie et la stocker dans une variable pour un traitement ultérieur pourrait être :

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

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

# mymodule.py
from writer import write

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

Cependant, cette approche ne parvient pas à capturer la sortie.

Une solution possible consiste à modifier les scripts comme suit :

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

Cette approche utilise un tampon pour capturer le flux de sortie.

À partir de Python 3.4, il existe une manière plus concise de capturer la sortie à l'aide de contextlib.redirect_stdout gestionnaire de contexte :

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

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

Cette solution est plus simple et élimine le besoin de gérer les sauvegardes et de fermer le flux manuellement.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!