mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-12-19 08:07:48 +01:00
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
|
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)
|