Comment capturer la sortie standard dans les scripts Python ?

Capture de la sortie standard du script

Dans un scénario dans lequel un script effectue des tâches spécifiques, telles que l'écriture de données sur sa sortie standard, la capture de cette sortie peut être crucial pour la suite du traitement. Une approche courante consiste à tenter de stocker la sortie dans une variable, comme illustré dans l'exemple suivant :

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

Cependant, cette approche ne parvient pas à capturer la sortie du script. Une solution alternative, utilisant une configuration d'objet et d'environnement StringIO, capture avec succès le résultat :

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

Solution Python 3.4

Pour les versions Python 3.4 et supérieures, une solution plus une approche simple est disponible en utilisant le gestionnaire de contexte 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>

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!