diff --git a/services/logger.py b/services/logger.py index 5ec288c1..0cdf2458 100644 --- a/services/logger.py +++ b/services/logger.py @@ -1,32 +1,34 @@ import logging import colorlog -def get_colorful_logger(name='root'): - # Define the color scheme - color_scheme = { - 'DEBUG': 'light_black', - 'INFO': 'green', - 'WARNING': 'yellow', - 'ERROR': 'red', - 'CRITICAL': 'red,bg_white', - } - # Define secondary log colors - secondary_colors = { - 'log_name': {'DEBUG': 'blue'}, - 'asctime': {'DEBUG': 'cyan'}, - 'process': {'DEBUG': 'purple'}, - 'module': {'DEBUG': 'light_black,bg_blue'}, - } +# Define the color scheme +color_scheme = { + 'DEBUG': 'light_black', + 'INFO': 'green', + 'WARNING': 'yellow', + 'ERROR': 'red', + 'CRITICAL': 'red,bg_white', +} - # Create a ColoredFormatter object for colorized logging - formatter = colorlog.ColoredFormatter( - "%(log_color)s%(levelname)s: %(log_color)s[%(module)s]%(reset)s %(white)s%(message)s", - log_colors=color_scheme, - secondary_log_colors=secondary_colors, # Set secondary log colors - style='%', - reset=True - ) +# Define secondary log colors +secondary_colors = { + 'log_name': {'DEBUG': 'blue'}, + 'asctime': {'DEBUG': 'cyan'}, + 'process': {'DEBUG': 'purple'}, + 'module': {'DEBUG': 'light_black,bg_blue'}, +} + +# Create a ColoredFormatter object for colorized logging +formatter = colorlog.ColoredFormatter( + "%(log_color)s%(levelname)s: %(log_color)s[%(module)s]%(reset)s %(white)s%(message)s", + log_colors=color_scheme, + secondary_log_colors=secondary_colors, # Set secondary log colors + style='%', + reset=True +) + +def get_colorful_logger(name='main'): # Create a stream handler for logging output stream = logging.StreamHandler() @@ -39,4 +41,8 @@ def get_colorful_logger(name='root'): return logger -root_logger = get_colorful_logger() +root_logger = logging.getLogger() +root_logger.setLevel(logging.DEBUG) # Set the log level +stream_handler = logging.StreamHandler() +stream_handler.setFormatter(formatter) +root_logger.addHandler(stream_handler)