Removes direct env access from env.py and keeps runtime DB selection logic intact. (#70)

Co-authored-by: RAMAMUTHUKUMARAN <130954718+RAMAMUTHUKUMARAN@users.noreply.github.com>

Author: srmk-2005

migrations/env.py

@@ -1,9 +1,7 @@
 from logging.config import fileConfig
-import os
-
 from sqlalchemy import engine_from_config, create_engine
 from sqlalchemy import pool
-
+from src.config import settings
 from alembic import context
 
 # this is the Alembic Config object, which provides
@@ -22,9 +20,9 @@
 
 # map Alembic environment names to DB URLs
 DB_URLS = {
-    "development": os.getenv("DEV_DATABASE_URL"),
-    "production": os.getenv("PROD_DATABASE_URL"),
-    "custom": os.getenv("CUSTOM_DATABASE_URL"),
+    "development": settings.dev_database_url,
+    "production": settings.prod_database_url,
+    "custom": settings.custom_database_url,
 }
 
 def get_url() -> str:
@@ -39,9 +37,8 @@ def get_url() -> str:
 # can be acquired:
 # my_important_option = config.get_main_option("my_important_option")
 # ... etc.
-from os import environ as env
 def env_url():
-    return env.get("CTI_POSTGRES_URL")
+    return settings.cti_postgres_url
 
 def run_migrations_offline() -> None:
     """Run migrations in 'offline' mode.

src/config.py

@@ -20,6 +20,11 @@ class Settings(BaseSettings):
     cti_mongo_url: Optional[str] = Field(validation_alias="CTI_MONGO_URL", default=None)
     cti_postgres_url: Optional[str] = Field(validation_alias="CTI_POSTGRES_URL", default=None)
 
+    # Alembic / Migration Database URLs
+    dev_database_url: Optional[str] = Field(validation_alias="DEV_DATABASE_URL", default=None)
+    prod_database_url: Optional[str] = Field(validation_alias="PROD_DATABASE_URL", default=None)
+    custom_database_url: Optional[str] = Field(validation_alias="CUSTOM_DATABASE_URL", default=None)
+
     # Canvas Variables, only required for Canvas specific-operations
     # NOTE: SIS ID's are needed for SIS operations, must be manually defined ahead of time
     # TODO: Rename CTI_ACCESS_TOKEN to CTI_CANVAS_TOKEN, improve clarity on token purpose and differentiate from server token