64 lines
2.8 KiB
Python
64 lines
2.8 KiB
Python
import os
|
|
import time
|
|
from logging.handlers import TimedRotatingFileHandler
|
|
|
|
|
|
class LufffyTimedRotatingFileHandler(TimedRotatingFileHandler):
|
|
def doRollover(self):
|
|
"""
|
|
do a rollover; in this case, a date/time stamp is appended to the filename
|
|
when the rollover happens. However, you want the file to be named for the
|
|
start of the interval, not the current time. If there is a backup count,
|
|
then we have to get a list of matching filenames, sort them and remove
|
|
the one with the oldest suffix.
|
|
"""
|
|
if self.stream:
|
|
self.stream.close()
|
|
self.stream = None
|
|
# get the time that this sequence started at and make it a TimeTuple
|
|
currentTime = int(time.time())
|
|
dstNow = time.localtime(currentTime)[-1]
|
|
t = self.rolloverAt - self.interval
|
|
if self.utc:
|
|
timeTuple = time.gmtime(t)
|
|
else:
|
|
timeTuple = time.localtime(t)
|
|
dstThen = timeTuple[-1]
|
|
if dstNow != dstThen:
|
|
if dstNow:
|
|
addend = 3600
|
|
else:
|
|
addend = -3600
|
|
timeTuple = time.localtime(t + addend)
|
|
"""
|
|
dfn = self.rotation_filename(self.baseFilename + "." +
|
|
time.strftime(self.suffix, timeTuple))
|
|
if os.path.exists(dfn):
|
|
os.remove(dfn)
|
|
self.rotate(self.baseFilename, dfn)
|
|
"""
|
|
# 多进程会导致误删日志，将上面代码重写为如下代码（判断如果不存在则重命名）
|
|
# 注意：如果改写的代码会影响其他模块则不能采用该方法
|
|
dfn = self.rotation_filename(self.baseFilename + "." +
|
|
time.strftime(self.suffix, timeTuple))
|
|
if not os.path.exists(dfn):
|
|
self.rotate(self.baseFilename, dfn)
|
|
|
|
if self.backupCount > 0:
|
|
for s in self.getFilesToDelete():
|
|
os.remove(s)
|
|
if not self.delay:
|
|
self.stream = self._open()
|
|
newRolloverAt = self.computeRollover(currentTime)
|
|
while newRolloverAt <= currentTime:
|
|
newRolloverAt = newRolloverAt + self.interval
|
|
#If DST changes and midnight or weekly rollover, adjust for this.
|
|
if (self.when == 'MIDNIGHT' or self.when.startswith('W')) and not self.utc:
|
|
dstAtRollover = time.localtime(newRolloverAt)[-1]
|
|
if dstNow != dstAtRollover:
|
|
if not dstNow: # DST kicks in before next rollover, so we need to deduct an hour
|
|
addend = -3600
|
|
else: # DST bows out before next rollover, so we need to add an hour
|
|
addend = 3600
|
|
newRolloverAt += addend
|
|
self.rolloverAt = newRolloverAt |