Spaces:
Running
Running
File size: 3,165 Bytes
5301c48 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
import os
import sys
from enum import IntEnum
from loguru import logger
simple_log_format_enabled = os.getenv("SIMPLE_LOG_FORMAT", "true").lower() in ("true", "1", "yes")
default_log_level = os.getenv("LOG_LEVEL", "INFO")
# Define custom log levels
class LogLevel(IntEnum):
"""Custom log levels."""
VERBOSE = 5
DEBUG = 10
INFO = 20
WARNING = 30
ERROR = 40
CRITICAL = 50
# Configuration Constants
COLORED_FORMAT = (
"<green>{time:YYYY-MM-DD HH:mm:ss}</green> | "
"<level>{level: <8}</level> | "
"<cyan>{name}</cyan> | "
"<blue>{file}:{line}</blue> | "
"<level>{message}</level>"
)
SIMPLE_COLORED_FORMAT = "<green>{time:YYYY-MM-DD HH:mm:ss}</green> | " "<level>{level: <8}</level> | " "<level>{message}</level>"
class LogManager:
"""Manages logger configuration."""
_instance = None
def __new__(cls):
"""Create a singleton instance."""
if cls._instance is None:
cls._instance = super(LogManager, cls).__new__(cls)
cls._instance.handler_id = None
cls._instance.current_level = default_log_level
cls._instance._initialize()
return cls._instance
def _get_format_string(self):
"""Return the appropriate format string based on LOG_FORMAT_MODE."""
if simple_log_format_enabled:
if self.current_level == "DEBUG":
return COLORED_FORMAT
return SIMPLE_COLORED_FORMAT
return COLORED_FORMAT
def _initialize(self):
"""Initialize logging with console handler."""
logger.remove() # Remove default handler
log_format = self._get_format_string()
self.handler_id = logger.add(sys.stdout, format=log_format, level=self.current_level, colorize=True)
# Add custom level only if it doesn't exist
try:
logger.level("VERBOSE", no=LogLevel.VERBOSE, color="<magenta>")
except ValueError:
# Level already exists, ignore the error
pass
def get_current_log_level(self):
"""Get the current log level."""
return self.current_level
def update_log_level(self, level):
"""Update the log level of the console handler.
This can be called at any time during runtime to change the log level.
"""
level = level.upper()
if level not in ["VERBOSE", "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]:
raise ValueError(f"Invalid log level: {level}")
logger.remove(self.handler_id)
self.current_level = level
log_format = self._get_format_string()
self.handler_id = logger.add(sys.stdout, format=log_format, level=self.current_level, colorize=True)
# Instantiate LogManager to ensure logging is initialized on module import
log_manager = LogManager()
# Add verbose method to logger
def verbose(self, message, *args, **kwargs):
"""Log a verbose message."""
self.log("VERBOSE", message, *args, **kwargs)
logger.__class__.verbose = verbose
# Function to get the logger
def get_logger(name):
"""Get a logger instance bound with a name."""
return logger.bind(name=name)
|