hyperboria/library/logging/handlers.py

43 lines
1.0 KiB
Python
Raw Normal View History

import logging.handlers
import os
import queue
from izihawa_types.var import varstr
class QueueHandler(logging.handlers.QueueHandler):
def __init__(self, *handlers):
self._queue = queue.Queue(-1)
self._listener = logging.handlers.QueueListener(self._queue, *handlers, respect_handler_level=True)
self.setLevel('INFO')
super().__init__(self._queue)
self._listener.start()
def stop(self):
self._listener.stop()
def prepare(self, record):
return record
class BaseFileHandler(logging.handlers.WatchedFileHandler):
def _open(self):
file = super()._open()
os.chmod(self.baseFilename, 0o644)
return file
class BaseBinaryFileHandler(BaseFileHandler):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs, mode='ab+')
def emit(self, record):
try:
self.stream.write(varstr(record.msg))
self.flush()
except RecursionError:
raise
except Exception:
self.handleError(record)