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>
This commit is contained in:
Yanray Wang 2023-08-14 10:33:37 +08:00
parent ee07afa205
commit 1998aac349
2 changed files with 11 additions and 12 deletions

View File

@ -906,11 +906,8 @@ def main():
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"

View File

@ -21,14 +21,16 @@ import sys
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 @@ def configure_logger(
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)