python: obtenga la salida de impresión en una declaración del ejecutivo

Question

Tengo un pequeño problema. Aquí está mi código:

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

¿Cómo puedo obtener las cosas que imprimen superadas? ¿Cómo puedo obtener algo como:

0 
1 
2 

Un saludo y gracias,

Answer 1

que tenían la misma idea que Frédéric, pero escribí un gestor de contexto para manejar la sustitución de la salida estándar:

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

Algo así como:

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

debe ejecutar el código en un proceso y tubería diferente la salida de nuevo a su programa principal a través de codeproc.stdout. Pero yo no he usado personalmente, por lo que si hay algo que he hecho mal dude en señalarlo: P

Answer 3

Puede redirigir la salida estándar a una cadena para la duración de la llamada 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

Aquí es el AP3-amigable versión de @ respuesta de Jochen. También agregué try-except cláusula para recuperar en caso de errores en el code.

import sys 
from io import StringIO 
import contextlib 

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

code = """ 
i = [0,1,2] 
for j in i : 
    print(j) 
""" 
with stdoutIO() as s: 
    try: 
     exec(code) 
    except: 
     print("Something wrong with the code") 
print "out:", s.getvalue() 

Answer 5

Aquí hay una pequeña corrección de la respuesta de Frédéric. Necesitamos manejar una posible excepción en exec() para regresar normal stdout. De otra manera no podríamos ver más lejos print salidas:

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