refactor: Support different log levels for different environments (#28)

la-famiglia-jst2324 · Jan 10, 2024 · 55affb3 · 55affb3
1 parent 5a552d9
commit 55affb3
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/parma_mining/github/api/main.py b/parma_mining/github/api/main.py
@@ -15,7 +15,15 @@
 )
 from parma_mining.github.normalization_map import GithubNormalizationMap
 
-logging.basicConfig(level=logging.INFO)
+env = os.getenv("env", "local")
+
+if env == "prod":
+    logging.basicConfig(level=logging.INFO)
+elif env in ["staging", "local"]:
+    logging.basicConfig(level=logging.DEBUG)
+else:
+    logging.warning(f"Unknown environment '{env}'. Defaulting to INFO level.")
+    logging.basicConfig(level=logging.INFO)
 
 logger = logging.getLogger(__name__)