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