Class Logger

logger.py:42–123

        return 'Vertical' not in record.getMessage()

class Logger:
    LOGS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data', 'logs')

    def __init__(self, name, level=logging.DEBUG, enable_file_logging=True):
        self.logger = logging.getLogger(name)
        self.logger.setLevel(level)
        self.logger.propagate = False  # Prevent duplicate logs via root logger
        self.enable_file_logging = enable_file_logging

        vertical_filter = VerticalFilter()

        if _HAS_RICH:
            # Define custom log level styles
            custom_theme = Theme({
                "debug": "yellow",
                "info": "blue",
                "warning": "yellow",
                "error": "bold red",
                "critical": "bold magenta",
                "success": "bold green"
            })

            console = Console(theme=custom_theme)

            # Create console handler with rich and set level
            console_handler = RichHandler(console=console, show_time=False, show_level=False, show_path=False, log_time_format="%Y-%m-%d %H:%M:%S")
        else:
            # Fallback to plain StreamHandler when rich is not available (e.g. Pager)
            console_handler = logging.StreamHandler()

        console_handler.setLevel(level)
        console_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
        console_handler.setFormatter(console_formatter)

        # Add filter to console handler
        console_handler.addFilter(vertical_filter)

        # Add console handler to the logger
        self.logger.addHandler(console_handler)

        if self.enable_file_logging:
            # Ensure the log folder exists
            os.makedirs(self.LOGS_DIR, exist_ok=True)
            log_file_path = os.path.join(self.LOGS_DIR, f"{name}.log")

            # Create file handler and set level
            file_handler = RotatingFileHandler(log_file_path, maxBytes=5*1024*1024, backupCount=2)
            file_handler.setLevel(level)
            file_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
            file_handler.setFormatter(file_formatter)

            # Add filter to file handler
            file_handler.addFilter(vertical_filter)

            # Add file handler to the logger
            self.logger.addHandler(file_handler)
    
    def set_level(self, level):