Logging stdout và stderr để đăng nhập tập tin sử dụng Python 'với' tuyên bố

Question

Tôi muốn đăng nhập đầu ra std của một đoạn mã Python vào một tập tin, sử dụng 'với' tuyên bố:

with log_to_file('log'): 
    # execute code 

là dễ nhất cách để thực hiện việc này để xác định thủ công log_to_file, ví dụ:

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 

hoặc có lớp tích hợp sẵn có thể làm điều này chưa?

Answer 1

Điều duy nhất tôi có thể đề xuất là sử dụng trang trí contextmanager nhưng tôi không tin điều này thực sự tốt hơn.

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