基于此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
对象,以便它可以与任何并行运行的脚本一起使用?
答案 0 :(得分:21)
如果您要替换sys.stdout
,则必须使用file-like object,这意味着您必须实施flush
。 flush
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 +中测试过。