update config logic

Author: byjoh

Makefile

@@ -1,7 +1,7 @@
+export ENV ?= stage
 VENV_NAME ?= venv
 PIP ?= pip
 PYTHON ?= python3.11
-ENV ?= stage
 
 venv: $(VENV_NAME)/bin/activate
 
@@ -13,19 +13,7 @@ $(VENV_NAME)/bin/activate: pyproject.toml
 test: venv
 	$(VENV_NAME)/bin/python -m unittest discover tests
 
-build: set-config venv
+build: venv
 	cp LICENSE LICENSE.bak
 	$(VENV_NAME)/bin/python -m build
 	rm LICENSE.bak
-
-set-config:
-ifeq ($(ENV), stage)
-	@cp extend/config/config_stage.py extend/config/config.py
-	@echo "Using STAGING configuration"
-else ifeq ($(ENV), prod)
-	@cp extend/config/config_prod.py extend/config/config.py
-	@echo "Using PRODUCTION configuration"
-else
-	@echo "Invalid ENV value. Use 'stage' or 'prod'."
-	@exit 1
-endif

extend/config/__init__.py

@@ -1,4 +1,17 @@
-from .config import API_HOST, API_VERSION
+import os
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+env = os.getenv("ENV", "dev")
+
+if env == "stage":
+    from .config_stage import API_HOST, API_VERSION
+elif env == "prod":
+    from .config_prod import API_HOST, API_VERSION
+else:
+    from .config_stage import API_HOST, API_VERSION
 
 __all__ = [
     "API_HOST",