Spaces:
Running
Running
| 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) | |