|
|
|
|
@@ -10,6 +10,7 @@ import logging
|
|
|
|
|
from datetime import datetime
|
|
|
|
|
import time
|
|
|
|
|
from pathlib import Path
|
|
|
|
|
import atexit
|
|
|
|
|
from typing import MutableMapping, TextIO, TypedDict, Any, TYPE_CHECKING, cast
|
|
|
|
|
from corelibs.logging_handling.logging_level_handling.logging_level import LoggingLevel
|
|
|
|
|
from corelibs.string_handling.text_colors import Colors
|
|
|
|
|
@@ -121,6 +122,9 @@ class LogParent:
|
|
|
|
|
self.log_queue: 'Queue[str] | None' = None
|
|
|
|
|
self.handlers: dict[str, Any] = {}
|
|
|
|
|
|
|
|
|
|
def __del__(self):
|
|
|
|
|
self._cleanup()
|
|
|
|
|
|
|
|
|
|
# FIXME: we need to add a custom formater to add stack level listing if we want to
|
|
|
|
|
# Important note, although they exist, it is recommended to use self.logger.NAME directly
|
|
|
|
|
# so that the correct filename, method and row number is set
|
|
|
|
|
@@ -282,6 +286,15 @@ class LogParent:
|
|
|
|
|
return False
|
|
|
|
|
return True
|
|
|
|
|
|
|
|
|
|
def _cleanup(self):
|
|
|
|
|
"""cleanup for any open queues in case we have an abort"""
|
|
|
|
|
if not self.log_queue:
|
|
|
|
|
return
|
|
|
|
|
self.flush()
|
|
|
|
|
# Close the queue properly
|
|
|
|
|
self.log_queue.close()
|
|
|
|
|
self.log_queue.join_thread()
|
|
|
|
|
|
|
|
|
|
# MARK: log level handling
|
|
|
|
|
def set_log_level(self, handler_name: str, log_level: LoggingLevel) -> bool:
|
|
|
|
|
"""
|
|
|
|
|
@@ -634,6 +647,7 @@ class Log(LogParent):
|
|
|
|
|
if log_queue is None:
|
|
|
|
|
return
|
|
|
|
|
self.log_queue = log_queue
|
|
|
|
|
atexit.register(self._cleanup)
|
|
|
|
|
self.listener = logging.handlers.QueueListener(
|
|
|
|
|
self.log_queue,
|
|
|
|
|
*self.handlers.values(),
|
|
|
|
|
|