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

la-famiglia-jst2324 · Jan 10, 2024 · 7dae571 · 7dae571
1 parent 2e9bdb2
commit 7dae571
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/parma_mining/discord/api/main.py b/parma_mining/discord/api/main.py
@@ -13,7 +13,15 @@
     ServersRequest,
 )
 
-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__)