python: obtenir la sortie d'impression dans une instruction exec

Question

J'ai un petit problème. Voici mon code:

code = """ 
i = [0,1,2] 
for j in i : 
    print j 
""" 
result = exec(code) 

Comment puis-je obtenir les éléments imprimés? Comment puis-je obtenir quelque chose comme:

0 
1 
2 

Cordialement et merci,

Answer 1

J'ai eu la même idée que Frédéric, mais je l'ai écrit un gestionnaire de contexte pour gérer le remplacement stdout:

import sys 
import StringIO 
import contextlib 

@contextlib.contextmanager 
def stdoutIO(stdout=None): 
    old = sys.stdout 
    if stdout is None: 
     stdout = StringIO.StringIO() 
    sys.stdout = stdout 
    yield stdout 
    sys.stdout = old 

code = """ 
i = [0,1,2] 
for j in i : 
    print j 
""" 
with stdoutIO() as s: 
    exec code 

print "out:", s.getvalue() 

Answer 2

Quelque chose comme:

codeproc = subprocess.Popen(code, stdout=subprocess.PIPE) 
print(codeproc.stdout.read()) 

doit exécuter le code dans un processus différent et redirigez la sortie à votre programme principal via codeproc.stdout. Mais je ne l'ai pas personnellement utilisé donc s'il y a quelque chose que j'ai fait de mal ne hésitez pas à le signaler: P

Answer 3

Vous pouvez rediriger la sortie standard à une chaîne pendant toute la durée de l'appel exec:

code = """ 
i = [0,1,2] 
for j in i : 
print j 
""" 

from cStringIO import StringIO 
old_stdout = sys.stdout 
redirected_output = sys.stdout = StringIO() 
exec(code) 
sys.stdout = old_stdout 

print redirected_output.getvalue() 

Answer 4

Voici la version PY3-friendly de @ réponse de Jochen. J'ai également ajouté try-except clause à récupérer en cas d'erreurs dans le code.

Answer 5

Voici une petite correction de la réponse de Frédéric. Nous devons gérer une exception possible dans exec() pour revenir à la normale stdout. Sinon, nous ne pouvions pas voir plus loin print sorties:

code = """ 
i = [0,1,2] 
for j in i : 
print j 
""" 

from cStringIO import StringIO 
old_stdout = sys.stdout 
redirected_output = sys.stdout = StringIO() 
try: 
    exec(code) 
except: 
    raise 
finally: # ! 
    sys.stdout = old_stdout # ! 

print redirected_output.getvalue() 
... 
print 'Hello, World!' # now we see it in case of the exception above