Supposons qu'un script fasse quelque chose comme ceci:
# module writer.py
import sys
def write():
sys.stdout.write("foobar")
Supposons maintenant que je veuille capturer la sortie de la write
fonction et la stocker dans une variable pour un traitement ultérieur. La solution naïve était:
# module mymodule.py
from writer import write
out = write()
print out.upper()
Mais ça ne marche pas. Je propose une autre solution et cela fonctionne, mais s'il vous plaît, faites-moi savoir s'il existe une meilleure façon de résoudre le problème. Merci
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