FE2E/library/utils.py

import logging
import sys


def setup_logging(args=None, log_level=None, reset=False):
    if logging.root.handlers:
        if reset:
            for handler in logging.root.handlers[:]:
                logging.root.removeHandler(handler)
        else:
            return

    if log_level is None and args is not None:
        log_level = getattr(args, "console_log_level", None)
    if log_level is None:
        log_level = "INFO"
    log_level = getattr(logging, str(log_level).upper())

    handler = logging.StreamHandler(sys.stdout)
    handler.setFormatter(logging.Formatter("%(message)s"))

    logging.root.setLevel(log_level)
    logging.root.addHandler(handler)


setup_logging()