File size: 1,305 Bytes
872b0a0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | 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
|