多处理中的Python日志记录:AttributeError:'Logger'对象没有属性'flush'

时间:2013-12-11 17:12:51

标签: python logging multiprocessing

基于此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对象,以便它可以与任何并行运行的脚本一起使用?

2 个答案:

答案 0 :(得分:21)

如果您要替换sys.stdout,则必须使用file-like object,这意味着您必须实施flushflush can be a no-op

def flush(self):
    pass

答案 1 :(得分:0)

@ ecatmur的回答只会修复丢失的同花顺,一旦我收到了这个:

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

original code帖子中的评论提供了一个修改,它将解释任何其他缺失的属性,为了完整性,我将发布此代码的完整工作版本:

class Logger(object):
    def __init__(self, filename = "logfile.log"):
        self.terminal = sys.stdout
        self.log = open(filename, "a")

    def __getattr__(self, attr):
        return getattr(self.terminal, attr)

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

    def flush(self):
        pass

这在Python 2.7中没有错误地运行。尚未在3.3 +中测试过。