capturando dis.dis resultados

Question

¿Hay alguna forma de obtener la salida dis.dis() sin redirigir sys.stdout? He tratado:

out=str(dis.dis()) 

y

out="""""" 
out+=str(dis.dis()) 

Sin embargo pronto me di cuenta de que devuelve None. ¿Hay alguna forma de arreglar esto?

Answer 1

Desafortunadamente, en las versiones de Python anteriores a la 3.4 el módulo dis usa declaraciones de impresión para stdout, por lo que no devolverá nada directamente útil. Ya sea que tenga que volver a poner en práctica las funciones dis, disassemble y disassemble_string, o se reemplaza temporalmente sys.stdout con una alternativa para capturar la salida:

import sys 
from cStringIO import StringIO 

out = StringIO() 
stdout = sys.stdout 
sys.stdout = out 
try: 
    dis.dis() 
finally: 
    sys.stdout = stdout 
out = out.getvalue() 

Esto es en realidad el mejor hecho usando un gestor de contexto:

import sys 
from contextlib import contextmanager 
from cStringIO import StringIO 

@contextmanager 
def captureStdOut(output): 
    stdout = sys.stdout 
    sys.stdout = output 
    try: 
     yield 
    finally: 
     sys.stdout = stdout 

out = StringIO() 
with captureStdOut(out): 
    dis.dis() 
print out.getvalue() 

De esta manera, se garantiza que tendrá stdout restaurado incluso si algo sale mal con dis. Una pequeña demostración:

>>> out = StringIO() 
>>> with captureStdOut(out): 
...  dis.dis(captureStdOut) 
... 
>>> print out.getvalue() 
83   0 LOAD_GLOBAL    0 (GeneratorContextManager) 
       3 LOAD_DEREF    0 (func) 
       6 LOAD_FAST    0 (args) 
       9 LOAD_FAST    1 (kwds) 
      12 CALL_FUNCTION_VAR_KW  0 
      15 CALL_FUNCTION   1 
      18 RETURN_VALUE   

En Python 3.4 y arriba, las funciones pertinentes a echar un parámetro file para redirigir la salida a:

from io import StringIO 

with StringIO() as out: 
    dis.dis(file=out) 
    print(out.get_value())