python: получить вывод печати в инструкции exec

Question

У меня есть небольшая проблема. Вот мой код:

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

Как я могу получить то, что напечатал вывод? Как я могу получить что-то вроде:

0 
1 
2 

С уважением и благодарностью,

Answer 1

Я была такая же идея, как Фредерик, но я написал менеджер контекста для обработки заменяющий стандартный вывод:

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

Что-то вроде:

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

должен выполнять код в другом процессе и трубы выход обратно в основной программе через codeproc.stdout. Но я лично не использовал его, так что если есть что-то я сделал не так, не стесняйтесь указать на это: P

Answer 3

Вы можете перенаправить стандартный вывод в строку на время ехес вызова:

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

Вот PY3 дружественный вариант @ ответ Йохена. Я также добавил try-except пункт, чтобы восстановить в случае ошибок в 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

Вот небольшая коррекция ответа Фредерика. Нам нужно обработать возможное исключение в exec() для возврата назад stdout. В противном случае мы не могли видеть дальше print выходов:

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