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

您可以将标准输出重定向到的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

这里是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

这是Frédéric的回答的一个小小的更正。我们需要在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