Source code for flexlate_dev.logger

import logging
from enum import Enum

from pydantic import BaseSettings, validator
from rich.logging import RichHandler


class LogLevel(str, Enum):
    INFO = "INFO"
    DEBUG = "DEBUG"


class LoggingConfig(BaseSettings):
    level: LogLevel = LogLevel.INFO

    class Config:
        env_prefix = "FLEXLATE_DEV_LOG_"

    @validator("level", pre=True)
    def cast_log_level(cls, v):
        if isinstance(v, LogLevel):
            return v
        level = str(v).casefold().strip()
        if level == "info":
            return LogLevel.INFO
        elif level == "debug":
            return LogLevel.DEBUG
        raise ValueError(f"invalid log level {level}")


LOGGING_CONFIG = LoggingConfig()

logging.basicConfig(
    level=LOGGING_CONFIG.level.value,
    format="%(message)s",
    datefmt="[%X]",
    handlers=[RichHandler(rich_tracebacks=True)],
)

log = logging.getLogger("flexlate-dev")

if __name__ == "__main__":
    log.info("info level")
    log.debug("debug level")
    try:
        raise ValueError("exception")
    except ValueError as e:
        log.exception(e)