From 35a983b85adf85f77a4fea84d02b04b41d82e153 Mon Sep 17 00:00:00 2001 From: Karl W Schulz Date: Fri, 8 Mar 2024 17:42:22 -0600 Subject: [PATCH] allow color formatter via env variable OMNIPERF_COLOR=1 Signed-off-by: Karl W Schulz [ROCm/rocprofiler-compute commit: 34070394eb3364ce92f431be9ff5a9b7ddb679e7] --- projects/rocprofiler-compute/src/utils/logger.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/projects/rocprofiler-compute/src/utils/logger.py b/projects/rocprofiler-compute/src/utils/logger.py index 3fea66d788..312b886250 100644 --- a/projects/rocprofiler-compute/src/utils/logger.py +++ b/projects/rocprofiler-compute/src/utils/logger.py @@ -54,7 +54,13 @@ class ColoredFormatter(logging.Formatter): # Setup console handler - provided as separate function to be called # prior to argument parsing def setup_console_handler(): - if False: + color = False + + if "OMNIPERF_COLOR" in os.environ.keys(): + if os.environ["OMNIPERF_COLOR"] == "1": + color = True + + if color: formatter = ColoredFormatter("%(levelname)s %(message)s") else: formatter = logging.Formatter("%(message)s")