[SAM-260011]: feat: refactor imports and enhance configuration management

Author: Heisenberg208

.vscode/settings.json

@@ -1 +1,9 @@
-{}
+{
+  "python-envs.pythonProjects": [
+    {
+      "path": "",
+      "envManager": "ms-python.python:poetry",
+      "packageManager": "ms-python.python:poetry"
+    }
+  ]
+}

pyproject.toml

@@ -28,19 +28,14 @@ mongo = ["pymongo"]
 [tool.poetry]
 packages = [
   { include = "sample", from = "src" },
-  { include = "features", from = "src" },
-  { include = "utils", from = "src" },
-  { include = "api", from = "src" },
-  { include = "db", from = "src" },
-  { include = "templates", from = "." },
   { include = "assets", from = "." },
 ]
-
+include = [{ path = "templates/**/*", format = ["sdist", "wheel"] }]
 exclude = ["src/**/__pycache__", "tests", "docs", "*.log"]
 
 [project.scripts]
 sample = "sample.cli:cli"
-lint = "utils.lint:main"
+lint = "sample.utils.lint:main"
 
 
 [tool.poetry.group.dev.dependencies]

src/__init__.py

@@ -1,3 +1,3 @@
-from utils._version import get_version
+from sample.utils._version import get_version
 
 __version__ = get_version()

src/sample/__init__.py

@@ -1,3 +1,3 @@
-from utils._version import get_version
+from sample.utils._version import get_version
 
 __version__ = get_version()

src/sample/__main__.py

@@ -2,12 +2,12 @@
 from contextlib import asynccontextmanager
 from pathlib import Path
 
-from db.connection import connect_db
 from fastapi import FastAPI, Request
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-from utils.constants import PORT
+from sample.db.connection import connect_db
+from sample.utils.constants import PORT
 
 from . import __version__
 

src/sample/api/__init__.py

@@ -1,3 +1,3 @@
-from utils._version import get_version
+from sample.utils._version import get_version
 
 __version__ = get_version()

src/sample/api/fast_api.py

@@ -1,9 +1,12 @@
+import logging
+
 from fastapi import APIRouter, FastAPI, HTTPException, Request
 from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel
+from sample.db.connection import connect_db
+from sample.utils.constants import API_PREFIX, GREETING
+from sample.utils.helper import normalize_name
 from starlette.exceptions import HTTPException as StarletteHTTPException
-from utils.constants import API_PREFIX, GREETING
-from utils.helper import normalize_name
 
 from . import __version__
 
@@ -139,8 +142,15 @@ async def chrome_devtools_probe():
 def start():
     import uvicorn
 
+    try:
+        _ = connect_db()
+    except Exception:
+        logging.warning(
+            "⚠️ Could not connect to the database. Please check database configuration."
+        )
+
     print(f"🧵 {__version__}\n")
-    uvicorn.run("api.fast_api:app", host="127.0.0.1", port=5000, reload=True)
+    uvicorn.run("sample.api.fast_api:app", host="127.0.0.1", port=5000, reload=True)
 
 
 if __name__ == "__main__":

src/sample/cli.py

@@ -24,6 +24,6 @@ def dev():
 @cli.command()
 def api():
     """Run the Sample FastAPI backend."""
-    from api.fast_api import start
+    from sample.api.fast_api import start
 
     start()

src/sample/db/connection.py

@@ -1,7 +1,7 @@
 import io
 from datetime import datetime
 
-from utils.constants import MONGO_CONFIG
+from sample.utils.constants import MONGO_CONFIG
 
 _mongo_client = None
 _db = None
@@ -28,7 +28,6 @@ def connect_db():
     try:
         uri = MONGO_CONFIG["MONGODB_URI"]
         name = MONGO_CONFIG["DATABASE_NAME"]
-        print("Mongo URI =", repr(uri))
 
         _mongo_client = MongoClient(uri, serverSelectionTimeoutMS=3000)
         # Force an actual connection attempt

src/sample/utils/config.py

@@ -0,0 +1,15 @@
+import os
+
+from dotenv import load_dotenv
+
+
+def load_env():
+    env = os.getenv("ENV", "development")
+
+    file_map = {
+        "production": ".env.production",
+        "development": ".env.development",
+    }
+
+    load_dotenv(file_map.get(env, ".env.development"), override=True)
+    return env