Class ColoredFormatter

datastudio/utils/logging.py:47–59

Colored formatter for console output.

class ColoredFormatter(logging.Formatter):
    """Colored formatter for console output."""

    def __init__(self, fmt: str = None, datefmt: str = None, use_color: bool = True):
        super().__init__(fmt, datefmt)
        self.use_color = use_color

    def format(self, record: logging.LogRecord) -> str:
        if self.use_color:
            levelname = record.levelname
            if levelname in COLORS:
                record.levelname = f"{COLORS[levelname]}[{levelname}]{COLORS['RESET']}"
        return super().format(record)


@HOOKS.register_module()