From 6cc8f0154fdb4c05b50004fd58739259b6a313d2 Mon Sep 17 00:00:00 2001 From: colramos-amd Date: Fri, 29 Mar 2024 11:23:06 -0500 Subject: [PATCH] Restore OMNIPERF_COLOR global to disable default log coloring Signed-off-by: colramos-amd --- src/utils/logger.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/src/utils/logger.py b/src/utils/logger.py index 50c13de51f..fc49d92d38 100644 --- a/src/utils/logger.py +++ b/src/utils/logger.py @@ -73,7 +73,22 @@ def setup_console_handler(): setattr(logging, "TRACE", logging.TRACE) setattr(logging, "trace", trace_logger) - formatter = ColoredFormatter() + color_setting = 1 + if "OMNIPERF_COLOR" in os.environ.keys(): + if type(os.environ["OMNIPERF_COLOR"]) != int: + raise TypeError( + "OMNIPERF_COLOR must be of type int. Detected: {}".format( + type(os.environ["OMNIPERF_COLOR"]) + ) + ) + color_setting = int(os.environ["OMNIPERF_COLOR"]) + + if color_setting == 1: + # colored loglevel and message + formatter = ColoredFormatter() + else: + # non-colored + formatter = PlainFormatter() console_handler = logging.StreamHandler(sys.stdout) console_handler.setFormatter(formatter)