Ensure 'load-data' from podman compose up can log to disk

konveyor · Jul 19, 2024 · 301208d · 301208d
1 parent 021c2f3
commit 301208d
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/kai/service/incident_store/psql.py b/kai/service/incident_store/psql.py
@@ -30,7 +30,18 @@ def json_exactly_equal(self, json_dict: dict):
 
 
 if __name__ == "__main__":
-    log_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), "../../../logs")
-    initLogging("INFO", "DEBUG", log_dir, "kai_psql.log")
+    if os.getenv("LOG_DIR") is not None:
+        log_dir = os.getenv("LOG_DIR")
+    else:
+        log_dir = os.path.join(
+            os.path.dirname(os.path.realpath(__file__)), "../../../logs"
+        )
+
+    if os.getenv("LOG_LEVEL") is not None:
+        log_level = os.getenv("LOG_LEVEL").upper()
+    else:
+        log_level = "INFO"
+
+    initLogging(log_level, "DEBUG", log_dir, "kai_psql.log")
 
     cmd("postgresql")