UnSAMFlow/utils/logger.py

import logging
import logging.config
import logging.handlers

# from pathlib import Path


def init_logger(level="INFO", log_name="main_logger"):

    logger = logging.getLogger(log_name)
    logger.setLevel(level)

    fh = logging.StreamHandler()
    formatter = logging.Formatter(
        "[%(levelname)s] %(message)s",
    )
    fh.setFormatter(formatter)
    logger.addHandler(fh)

    logger.info("Start logging!")
    return logger


# def init_logger(
#     level="INFO", log_dir="./", log_name="main_logger", filename="main.log"
# ):

#     logger = logging.getLogger(log_name)

#     fh = logging.handlers.RotatingFileHandler(
#         Path(log_dir) / filename, "w", 20 * 1024 * 1024, 5
#     )
#     formatter = logging.Formatter(
#         "%(asctime)s %(levelname)5s - %(name)s "
#         "[%(filename)s line %(lineno)d] - %(message)s",
#         datefmt="%m-%d %H:%M:%S",
#     )
#     fh.setFormatter(formatter)
#     logger.addHandler(fh)

#     # logging to screen
#     if "DEBUG" in log_dir:
#         fh = logging.StreamHandler()
#         formatter = logging.Formatter(
#             "[%(levelname)s] %(message)s",
#         )
#         fh.setFormatter(formatter)
#         logger.addHandler(fh)

#     logger.setLevel(level)
#     logger.info("Start training")
#     return logger