python: pobierz wydruk w instrukcji exec

Question

Mam mały problem. Oto mój kod:

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

Jak mogę uzyskać wydruk wyjściowy? Jak mogę dostać coś takiego:

0 
1 
2 

Pozdrowienia i podziękowania

Answer 1

miałem ten sam pomysł, Frédéric, ale napisałem menedżera kontekstowe do obsługi zastępując 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

Coś jak:

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

należy wykonać kod w innym procesie i rury wyjście z powrotem do programu głównego przez codeproc.stdout. Ale ja nie osobiście używał go więc jeśli jest coś zrobiłem źle czuć swobodnie wskazać go: P

Answer 3

można przekierować standardowe wyjście na ciąg na czas trwania rozmowy: 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

Oto Py3 wersja z @ Jochen za odpowiedź. Dodałem także klauzulę try-except, aby odzyskać w przypadku błędów w 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

Oto mała korekta odpowiedzi Frédéric. Musimy obsłużyć ewentualny wyjątek w exec(), aby powrócić normalnie stdout. W przeciwnym razie nie moglibyśmy widzieć dalej print wyjść:

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