Comment capturer la sortie standard dans les scripts Python ?

Patricia Arquette
Libérer: 2024-10-17 14:57:02
original
998 Les gens l'ont consulté

How to Capture Standard Output in Python Scripts?

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>
Copier après la connexion
<code class="python"># mymodule.py
from writer import write

out = write()
print(out.upper())</code>
Copier après la connexion

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>
Copier après la connexion

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>
Copier après la connexion

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!

source:php
Déclaration de ce site Web
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn
Derniers articles par auteur
Tutoriels populaires
Plus>
Derniers téléchargements
Plus>
effets Web
Code source du site Web
Matériel du site Web
Modèle frontal