From 975c2dad8bebbe14d5633fff2f54bb2ac43b72bc Mon Sep 17 00:00:00 2001 From: Karl W Schulz Date: Thu, 7 Mar 2024 16:51:06 -0600 Subject: [PATCH] add quietmode option to setup_logging function Signed-off-by: Karl W Schulz [ROCm/rocprofiler-compute commit: abd3d20cb212d85934835f27795fa8a82b1d902e] --- projects/rocprofiler-compute/src/utils/logger.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/projects/rocprofiler-compute/src/utils/logger.py b/projects/rocprofiler-compute/src/utils/logger.py index 9a65894016..06bfce1975 100644 --- a/projects/rocprofiler-compute/src/utils/logger.py +++ b/projects/rocprofiler-compute/src/utils/logger.py @@ -52,17 +52,22 @@ class ColoredFormatter(logging.Formatter): # Setup logger -def setup_logging(verbose): +def setup_logging(verbosity, quietmode): # register a trace level logger logging.TRACE = logging.DEBUG - 5 logging.addLevelName(logging.TRACE, "TRACE") setattr(logging, "TRACE", logging.TRACE) setattr(logging, "trace", trace_logger) - # default: set loglevel based on selected verbosity + # set loglevel based on selected verbosity and quietmode levels = [logging.INFO, logging.DEBUG, logging.TRACE] - loglevel = levels[min(verbose, len(levels) - 1)] # cap to last level index - # optional: override of default loglevel via env variable + + if quietmode: + loglevel = logging.ERROR + else: + loglevel = levels[min(verbosity, len(levels) - 1)] # cap to last level index + + # optional: override of default loglevel via env variable which takes precedence if "OMNIPERF_LOGLEVEL" in os.environ.keys(): loglevel = os.environ["OMNIPERF_LOGLEVEL"] if loglevel in {"DEBUG", "debug"}: