48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
# Copyright (c) 2023-present Plane Software, Inc. and contributors
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# See the LICENSE file for details.
|
|
|
|
import logging.handlers as handlers
|
|
import time
|
|
|
|
|
|
class SizedTimedRotatingFileHandler(handlers.TimedRotatingFileHandler):
|
|
"""
|
|
Handler for logging to a set of files, which switches from one file
|
|
to the next when the current file reaches a certain size, or at certain
|
|
timed intervals
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
filename,
|
|
maxBytes=0,
|
|
backupCount=0,
|
|
encoding=None,
|
|
delay=0,
|
|
when="h",
|
|
interval=1,
|
|
utc=False,
|
|
):
|
|
handlers.TimedRotatingFileHandler.__init__(self, filename, when, interval, backupCount, encoding, delay, utc)
|
|
self.maxBytes = maxBytes
|
|
|
|
def shouldRollover(self, record):
|
|
"""
|
|
Determine if rollover should occur.
|
|
|
|
Basically, see if the supplied record would cause the file to exceed
|
|
the size limit we have.
|
|
"""
|
|
if self.stream is None: # delay was set...
|
|
self.stream = self._open()
|
|
if self.maxBytes > 0: # are we rolling over?
|
|
msg = "%s\n" % self.format(record)
|
|
# due to non-posix-compliant Windows feature
|
|
self.stream.seek(0, 2)
|
|
if self.stream.tell() + len(msg) >= self.maxBytes:
|
|
return 1
|
|
t = int(time.time())
|
|
if t >= self.rolloverAt:
|
|
return 1
|
|
return 0
|