From 401e6e06acb76d6b452686e986c79a41d812341f Mon Sep 17 00:00:00 2001 From: David Trattnig <david@subsquare.at> Date: Sat, 4 Feb 2023 18:31:59 +0100 Subject: [PATCH] refact: slight debugging improvements --- src/aura_engine/base/logger.py | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) diff --git a/src/aura_engine/base/logger.py b/src/aura_engine/base/logger.py index 0e364b34..206a38b7 100644 --- a/src/aura_engine/base/logger.py +++ b/src/aura_engine/base/logger.py @@ -54,17 +54,20 @@ class AuraLogger: Retrieve the configured log level (default=INFO). """ lvl = self.config.get("log_level") - - if lvl == "debug": - return logging.DEBUG - elif lvl == "info": - return logging.INFO - elif lvl == "warning": - return logging.WARNING - elif lvl == "error": - return logging.ERROR - else: - return logging.CRITICAL + mapping = { + "debug": logging.DEBUG, + "info": logging.INFO, + "warning": logging.WARNING, + "error": logging.ERROR, + "critical": logging.CRITICAL, + } + + log_level = mapping.get(lvl) + if not log_level: + print("No log level configured. Using INFO.") + log_level = logging.INFO + print(f"Setting log level {log_level} ({lvl})") + return log_level def create_logger(self, name, lvl): """ -- GitLab