🔧 Reuse database url from config in alembic setup (#1229)

This commit is contained in:
Patrick Arminio
2024-07-30 19:46:41 +02:00
committed by GitHub
parent b15d482605
commit b5b2929104
2 changed files with 3 additions and 7 deletions

View File

@@ -19,6 +19,7 @@ fileConfig(config.config_file_name)
# target_metadata = None # target_metadata = None
from app.models import SQLModel # noqa from app.models import SQLModel # noqa
from app.core.config import settings # noqa
target_metadata = SQLModel.metadata target_metadata = SQLModel.metadata
@@ -29,12 +30,7 @@ target_metadata = SQLModel.metadata
def get_url(): def get_url():
user = os.getenv("POSTGRES_USER", "postgres") return str(settings.SQLALCHEMY_DATABASE_URI)
password = os.getenv("POSTGRES_PASSWORD", "")
server = os.getenv("POSTGRES_SERVER", "db")
port = os.getenv("POSTGRES_PORT", "5432")
db = os.getenv("POSTGRES_DB", "app")
return f"postgresql+psycopg://{user}:{password}@{server}:{port}/{db}"
def run_migrations_offline(): def run_migrations_offline():

View File

@@ -51,7 +51,7 @@ class Settings(BaseSettings):
POSTGRES_SERVER: str POSTGRES_SERVER: str
POSTGRES_PORT: int = 5432 POSTGRES_PORT: int = 5432
POSTGRES_USER: str POSTGRES_USER: str
POSTGRES_PASSWORD: str POSTGRES_PASSWORD: str = ""
POSTGRES_DB: str = "" POSTGRES_DB: str = ""
@computed_field # type: ignore[misc] @computed_field # type: ignore[misc]