refactor logging initialization into two steps:

(1) setup_logging_handler -> called prior to arg parsing
(2) setup_logging_priority -> called after arg parsing

Signed-off-by: Karl W Schulz <karl.schulz@amd.com>


[ROCm/rocprofiler-compute commit: e24e6c35a5]
这个提交包含在:
Karl W Schulz
2024-03-08 09:38:02 -06:00
提交者 Karl W. Schulz
父节点 2381bba1cf
当前提交 917b784ead
修改 2 个文件,包含 21 行新增12 行删除
@@ -51,8 +51,21 @@ class ColoredFormatter(logging.Formatter):
return logging.Formatter.format(self, record)
# Setup logger
def setup_logging(verbosity, quietmode):
# Setup logger handler - provided as separate function to be called
# prior to argument parsing
def setup_logging_handler():
if False:
formatter = ColoredFormatter("%(levelname)s %(message)s")
else:
formatter = logging.Formatter("%(message)s")
handler = logging.StreamHandler(sys.stdout)
handler.setFormatter(formatter)
logging.basicConfig(handlers=[handler])
# Setup logger priority
def setup_logging_priority(verbosity, quietmode):
# register a trace level logger
logging.TRACE = logging.DEBUG - 5
logging.addLevelName(logging.TRACE, "TRACE")
@@ -82,11 +95,6 @@ def setup_logging(verbosity, quietmode):
print("Ignoring unsupported OMNIPERF_LOGLEVEL setting (%s)" % loglevel)
sys.exit(1)
if False:
formatter = ColoredFormatter("%(levelname)s %(message)s")
else:
formatter = ColoredFormatter("%(message)s")
handler = logging.StreamHandler(sys.stdout)
handler.setFormatter(formatter)
logging.basicConfig(level=loglevel, handlers=[handler])
# update priority
logging.getLogger().setLevel(loglevel)
return loglevel