Source code for sbmlutils.log

"""Module for logging.

Using rich for output formating.
"""
import logging
from typing import List

from rich.logging import RichHandler

from sbmlutils.console import console


def get_logger(name: str, level: int = logging.INFO) -> logging.Logger:
    """Get new custom logger for name."""
    formatter = logging.Formatter(
        fmt="%(message)s",
        datefmt="[%X]",
    )

    # handler = logging.StreamHandler()
    handler = RichHandler(
        markup=False, rich_tracebacks=True, show_time=False, console=console
    )
    handler.setFormatter(formatter)

    logger = logging.getLogger(name)
    logger.setLevel(level)
    logger.addHandler(handler)
    return logger


def set_level_for_all_loggers(level: int = logging.DEBUG) -> None:
    """Set log level for all loggers.

    This is mainly used in debugging to get additional information.
    FIXME: add resetting functionality
    """
    loggers: List[logging.Logger] = [
        logging.getLogger(name) for name in logging.root.manager.loggerDict
    ]
    for logger in loggers:
        logger.setLevel(level)