41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
class RotateHandler(logging.Handler):
|
||
|
|
"""
|
||
|
|
A handler class which buffers logging records in memory. A rotating
|
||
|
|
buffer is used so that the last X records are always available.
|
||
|
|
"""
|
||
|
|
def __init__(self, capacity):
|
||
|
|
"""
|
||
|
|
Initialize the handler with the buffer size.
|
||
|
|
"""
|
||
|
|
logging.Handler.__init__(self)
|
||
|
|
|
||
|
|
self.set_capacity(capacity)
|
||
|
|
|
||
|
|
|
||
|
|
def emit(self, record):
|
||
|
|
"""
|
||
|
|
Add the record to the rotating buffer.
|
||
|
|
|
||
|
|
"""
|
||
|
|
self._buffer[self._index] = record
|
||
|
|
self._index = (self._index + 1 ) % self._capacity
|
||
|
|
|
||
|
|
|
||
|
|
def get_buffer(self):
|
||
|
|
"""
|
||
|
|
Return the buffer with the records in the correct order.
|
||
|
|
"""
|
||
|
|
|
||
|
|
return [record for record in self._buffer[self._index:] + self._buffer[:self._index]
|
||
|
|
if record is not None]
|
||
|
|
|
||
|
|
def set_capacity(self,capacity):
|
||
|
|
"""
|
||
|
|
Set the number of log records that will be stored.
|
||
|
|
"""
|
||
|
|
self._capacity = capacity
|
||
|
|
self._index = 0
|
||
|
|
self._buffer = self._capacity * [None]
|