Function set_logger_level

codecarbon/external/logger.py:25–59

(level: Optional[str] = None)

def set_logger_level(level: Optional[str] = None):
    if level is None:
        lower_envs = {k.lower(): v for k, v in os.environ.items()}
        level = lower_envs.get("codecarbon_log_level", "INFO")

    logger = logging.getLogger("codecarbon")

    if isinstance(level, int):
        known_levels_int = {0, 10, 20, 30, 40, 50}
        if level not in known_levels_int:
            logger.error(f"Unknown int log level: {level}. Doing nothing.")
            return
        logger.setLevel(level)
        return

    if isinstance(level, str):
        level = level.upper()
        known_levels_str = {
            "CRITICAL",
            "FATAL",
            "ERROR",
            "WARN",
            "WARNING",
            "INFO",
            "DEBUG",
            "NOTSET",
        }

        if level not in known_levels_str:
            logger.error(f"Unknown str log level: {level}. Doing nothing.")
            return
        logger.setLevel(level)
        return

    logger.error(f"Unknown log level: {level}. Doing nothing.")


set_logger_format()