Fix logging level setting issue
[pti/o2.git] / o2common / helper / o2logging.py
index 31584f5..e1196d2 100644 (file)
@@ -20,17 +20,25 @@ import yaml
 
 
 def get_logger(name=None):
-    CONFIG_FILE = os.environ.get(
-        "LOGGING_CONFIG_FILE", "/etc/o2/log.yaml")
-    if os.path.exists(CONFIG_FILE):
-        with open(file=CONFIG_FILE, mode='r', encoding="utf-8") as file:
-            config_yaml = yaml.load(stream=file, Loader=yaml.FullLoader)
-        logging.config.dictConfig(config=config_yaml)
+    CONFIG_FILE = os.environ.get(
+        "LOGGING_CONFIG_FILE", "/etc/o2/log.yaml")
+    if os.path.exists(CONFIG_FILE):
+        with open(file=CONFIG_FILE, mode='r', encoding="utf-8") as file:
+            config_yaml = yaml.load(stream=file, Loader=yaml.FullLoader)
+        logging.config.dictConfig(config=config_yaml)
 
     logger = logging.getLogger(name)
-
-    # override logging level
+    # override root logger's logging level
     LOGGING_CONFIG_LEVEL = os.environ.get("LOGGING_CONFIG_LEVEL", None)
     if LOGGING_CONFIG_LEVEL:
         logger.setLevel(LOGGING_CONFIG_LEVEL)
     return logger
+
+
+def configure_logger():
+    CONFIG_FILE = os.environ.get(
+        "LOGGING_CONFIG_FILE", "/etc/o2/log.yaml")
+    if os.path.exists(CONFIG_FILE):
+        with open(file=CONFIG_FILE, mode='r', encoding="utf-8") as file:
+            config_yaml = yaml.load(stream=file, Loader=yaml.FullLoader)
+            logging.config.dictConfig(config=config_yaml)