la salida estándar de registro y stderr al archivo usando Python conectarse 'con' declaración

Question

Quiero registrar la salida std de un trozo de código Python a un archivo, utilizando el 'con' declaración:

with log_to_file('log'): 
    # execute code 

es el más fácil manera de hacer esto para definir la log_to_file manualmente, por ejemplo:

import sys 

class log_to_file(): 
    def __init__(self, filename): 
     self.f = open(filename, 'wb') 

    def __enter__(self): 
     self.stdout = sys.stdout 
     self.stderr = sys.stderr 
     sys.stdout = self.f 
     sys.stderr = self.f 

    def __exit__(self, type, value, traceback): 
     sys.stdout = self.stdout 
     sys.stderr = self.stderr 

o hay una clase incorporada que puede hacer esto ya?

Answer 1

Lo único que podría sugerir es usar el decorador contextmanager, pero no estoy seguro de que esto sea realmente mejor.

from contextlib import contextmanager 
@contextmanager 
def stdouterrlog(logfile): 
    with open(logfile, 'wb') as lf: 
    stdout = sys.stdout 
    stderr = sys.stderr 
    sys.stdout = lf 
    sys.stderr = lf 
    yield lf # support 'with stdouterrlog(x) as logfile' 
    sys.stdout = stdout 
    sys.stderr = stderr