From c7202923b04372985d5909f7725b2983ffbaa150 Mon Sep 17 00:00:00 2001 From: xuchen-amd Date: Wed, 2 Apr 2025 20:29:39 -0400 Subject: [PATCH] remove flask debug msg (#655) * Suppress Flask warning message in quiet mode. * Init args.gui if dne. --- src/rocprof_compute_base.py | 3 ++- src/utils/logger.py | 7 ++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/src/rocprof_compute_base.py b/src/rocprof_compute_base.py index 2bcffa3775..1a9556e6b5 100644 --- a/src/rocprof_compute_base.py +++ b/src/rocprof_compute_base.py @@ -79,8 +79,9 @@ class RocProfCompute: self.set_version() self.parse_args() self.__mode = self.__args.mode + gui_value = getattr(self.__args, "gui", None) self.__loglevel = setup_logging_priority( - self.__args.verbose, self.__args.quiet, self.__mode + self.__args.verbose, self.__args.quiet, self.__mode, gui_value ) setattr(self.__args, "loglevel", self.__loglevel) set_locale_encoding() diff --git a/src/utils/logger.py b/src/utils/logger.py index 5e61e4b6c5..e8c450095e 100644 --- a/src/utils/logger.py +++ b/src/utils/logger.py @@ -168,7 +168,7 @@ def setup_file_handler(loglevel, workload_dir): # Setup logger priority - called after argument parsing -def setup_logging_priority(verbosity, quietmode, appmode): +def setup_logging_priority(verbosity, quietmode, appmode, guimode): # set loglevel based on selected verbosity and quietmode levels = [logging.INFO, logging.DEBUG, logging.TRACE] @@ -178,6 +178,11 @@ def setup_logging_priority(verbosity, quietmode, appmode): else: loglevel = levels[min(verbosity, len(levels) - 1)] # cap to last level index + # optional: suppress Werkzeug's messages in analyze GUIs. + if quietmode and "analyze" in appmode and guimode: + werkzeug_logger = logging.getLogger("werkzeug") + werkzeug_logger.setLevel(logging.ERROR) + # optional: override of default loglevel via env variable which takes precedence if "ROCPROFCOMPUTE_LOGLEVEL" in os.environ.keys(): loglevel = os.environ["ROCPROFCOMPUTE_LOGLEVEL"]