testn / app /utils /log_utils.py
narendrasinghd's picture
Upload 18 files
7adba0b verified
raw
history blame contribute delete
803 Bytes
import logging
import colorlog
def setup_logger(logger_name: str, debug_color="white", info_color="green", propagate=False):
color_handler = colorlog.StreamHandler()
log_colors = {
"DEBUG": debug_color,
"INFO": info_color,
"WARNING": "yellow",
"ERROR": "red",
"CRITICAL": "bold_red",
}
color_handler.setFormatter(
colorlog.ColoredFormatter(
"%(log_color)s%(asctime)s [%(levelname)s] %(name)s: %(message)s",
datefmt="%Y-%m-%d %H:%M:%S",
log_colors=log_colors
)
)
logger = logging.getLogger(logger_name)
if logger.hasHandlers():
logger.handlers.clear()
logger.setLevel(logging.DEBUG)
logger.addHandler(color_handler)
logger.propagate = propagate
return logger