Logging

Logging helper functions.

get_logger(name)

Get logger with rich configuration.

Source code in databooks/logging.py

def get_logger(name: str) -> logging.Logger:
    """Get logger with rich configuration."""
    level = os.getenv("LOG_LEVEL", logging.INFO)
    logging.basicConfig(
        level=level,
        format="%(message)s",
        datefmt="[%X]",
        handlers=[RichHandler(rich_tracebacks=True)],
    )
    return logging.getLogger(name)

set_verbose(logger)

Set logger to DEBUG level when user requests verbosity.

Source code in databooks/logging.py

def set_verbose(logger: logging.Logger) -> None:
    """Set logger to DEBUG level when user requests verbosity."""
    verbose_level = logging.DEBUG
    if logger.level < verbose_level:
        logger.setLevel(logging.DEBUG)
        logger.debug(
            f"Verbose mode: setting log level to {logging.getLevelName(verbose_level)}"
        )