logger.py

import logging
from pathlib import Path
# logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# logger = logging.getLogger("FusionGAN")

# 此处修改颜色
FMTDCIT = {
    'ERROR': "\033[31mERROR\033[0m",
    'INFO': "\033[37mINFO\033[0m",
    'DEBUG': "\033[1mDEBUG\033[0m",
    'WARN': "\033[33mWARN\033[0m",
    'WARNING': "\033[33mWARNING\033[0m",
    'CRITICAL': "\033[35mCRITICAL\033[0m",
}


class Filter(logging.Filter):
    def __init__(self, **kwargs) -> None:
        super().__init__(**kwargs)

    def filter(self, record: logging.LogRecord) -> bool:
        record.levelname = FMTDCIT.get(record.levelname)
        return True


filter = Filter()


def getLogger(
        name: str,
        level: int = logging.INFO,
        fmt: str = "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
        fmt_date: str = "%H:%M:%S",
        log_dir:str = "./log"
) -> logging.Logger:
    fmter = logging.Formatter(fmt, fmt_date)
    ch = logging.StreamHandler()
    ch.setLevel(level)
    ch.setFormatter(fmter)
    ch.addFilter(filter)

    logger = logging.getLogger(name)
    logger.setLevel(level)
    logger.addHandler(ch)

    fh = logging.FileHandler(log_dir)
    logger.addHandler(fh)
    logger.setLevel(level)
    return logger