使用 Python 'with' 语句将 stdout 和 stderr 记录到日志文件

Question

我想使用'with'语句将一段Python代码的std输出记录到一个文件中:

with log_to_file('log'):
    # execute code

是手动定义log_to_file最简单的方法，例如:

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

还是已经有一个内置的类可以这样做?

Answer 1

我唯一能建议的是使用contextmanager装饰器，但我不相信这真的更好。

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