commit 1998aac349a36f4a810e1f07f271eab55e80c7b4
author Yanray Wang <yanray.wang@arm.com> Mon Aug 14 10:33:37 2023 +0800
committer Yanray Wang <yanray.wang@arm.com> Mon Aug 14 11:22:34 2023 +0800
tree 2fa78865abf476123517aac0a3ca50a7078ac7cb
parent ee07afa2051a1d7d8a20e46035b3cc8e4c94fc80

logging_util: support to tweak loglevel directed to stderr/stdout

Previously we set loglevel >= WARNING printed to stderr and
loglevel < WARNING printed to stdout. To be more flexible, we
replace this `WARNING` value with an argument: split_level and leave
`WARNING` as default split_level if not set.

Signed-off-by: Yanray Wang <yanray.wang@arm.com>

scripts/code_size_compare.py

diff --git a/scripts/code_size_compare.py b/scripts/code_size_compare.py
index 72c69e4..672b803 100755
--- a/scripts/code_size_compare.py
+++ b/scripts/code_size_compare.py
@@ -906,11 +906,8 @@
     comp_args = parser.parse_args()
 
     logger = logging.getLogger()
-    logging_util.configure_logger(logger)
-    if comp_args.stdout and not comp_args.verbose:
-        logger.setLevel(logging.ERROR)
-    else:
-        logger.setLevel(logging.DEBUG if comp_args.verbose else logging.INFO)
+    logging_util.configure_logger(logger, split_level=logging.NOTSET)
+    logger.setLevel(logging.DEBUG if comp_args.verbose else logging.INFO)
 
     if os.path.isfile(comp_args.record_dir):
         logger.error("record directory: {} is not a directory"

scripts/mbedtls_dev/logging_util.py

diff --git a/scripts/mbedtls_dev/logging_util.py b/scripts/mbedtls_dev/logging_util.py
index 85a3f19..db1ebfe 100644
--- a/scripts/mbedtls_dev/logging_util.py
+++ b/scripts/mbedtls_dev/logging_util.py
@@ -21,14 +21,16 @@
 
 def configure_logger(
         logger: logging.Logger,
-        log_format="[%(levelname)s]: %(message)s"
+        log_format="[%(levelname)s]: %(message)s",
+        split_level=logging.WARNING
     ) -> None:
     """
     Configure the logging.Logger instance so that:
         - Format is set to any log_format.
             Default: "[%(levelname)s]: %(message)s"
-        - loglevel >= WARNING are printed to stderr.
-        - loglevel <  WARNING are printed to stdout.
+        - loglevel >= split_level are printed to stderr.
+        - loglevel <  split_level are printed to stdout.
+            Default: logging.WARNING
     """
     class MaxLevelFilter(logging.Filter):
         # pylint: disable=too-few-public-methods
@@ -41,14 +43,14 @@
 
     log_formatter = logging.Formatter(log_format)
 
-    # set loglevel >= WARNING to be printed to stderr
+    # set loglevel >= split_level to be printed to stderr
     stderr_hdlr = logging.StreamHandler(sys.stderr)
-    stderr_hdlr.setLevel(logging.WARNING)
+    stderr_hdlr.setLevel(split_level)
     stderr_hdlr.setFormatter(log_formatter)
 
-    # set loglevel <= INFO to be printed to stdout
+    # set loglevel < split_level to be printed to stdout
     stdout_hdlr = logging.StreamHandler(sys.stdout)
-    stdout_hdlr.addFilter(MaxLevelFilter(logging.INFO))
+    stdout_hdlr.addFilter(MaxLevelFilter(split_level - 1))
     stdout_hdlr.setFormatter(log_formatter)
 
     logger.addHandler(stderr_hdlr)