From 1556106cf0afd6fa2c1e81f8723a0d9b4273e08f Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Tue, 28 Jun 2022 18:27:22 +0200 Subject: [PATCH] Make colors optional --- src/great_ai/utilities/logger/get_logger.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/great_ai/utilities/logger/get_logger.py b/src/great_ai/utilities/logger/get_logger.py index 7f83634..fdae0d6 100644 --- a/src/great_ai/utilities/logger/get_logger.py +++ b/src/great_ai/utilities/logger/get_logger.py @@ -6,7 +6,9 @@ from .custom_formatter import CustomFormatter loggers: Dict[str, logging.Logger] = {} -def get_logger(name: str, level: int = logging.INFO) -> logging.Logger: +def get_logger( + name: str, level: int = logging.INFO, disable_colors: bool = False +) -> logging.Logger: if name not in loggers: logger = logging.getLogger(name) logger.setLevel(level) @@ -14,8 +16,9 @@ def get_logger(name: str, level: int = logging.INFO) -> logging.Logger: fmt = "%(asctime)s | %(levelname)8s | %(message)s" stdout_handler = logging.StreamHandler() - stdout_handler.setLevel(logging.DEBUG) - stdout_handler.setFormatter(CustomFormatter(fmt)) + stdout_handler.setLevel(level) + if not disable_colors: + stdout_handler.setFormatter(CustomFormatter(fmt)) logger.addHandler(stdout_handler) loggers[name] = logger