多处理中的 Python 日志记录 : AttributeError: 'Logger' object has no attribute 'flush'

Question

基于此code我创建了一个 python 对象，它既将输出打印到终端，又将输出保存到一个日志文件，并在其名称后附加日期和时间:

import sys
import time

class Logger(object):
    """
    Creates a class that will both print and log any
    output text. See https://stackoverflow.com/a/5916874
    for original source code. Modified to add date and
    time to end of file name.
    """
    def __init__(self, filename="Default"):
        self.terminal = sys.stdout
        self.filename = filename + ' ' + time.strftime('%Y-%m-%d-%H-%M-%S') + '.txt'
        self.log = open(self.filename, "a")

    def write(self, message):
        self.terminal.write(message)
        self.log.write(message)


sys.stdout = Logger('TestLog')

这很好用，但是当我尝试将它与使用 Pool 多处理函数的脚本一起使用时，出现以下错误:

AttributeError: 'Logger' object has no attribute 'flush'

如何修改我的 Logger 对象，以便它可以与任何并行运行的脚本一起工作？

Answer 1

如果您要替换 sys.stdout，它必须是 file-like object ，这意味着您必须实现 flush。 flush can be a no-op :

def flush(self):
    pass