diff --git a/alembic/env.py b/alembic/env.py index b277d38..e260ebe 100644 --- a/alembic/env.py +++ b/alembic/env.py @@ -4,16 +4,15 @@ from logging.config import fileConfig from sqlalchemy import engine_from_config, pool from alembic import context -from db import Base +from db import Base, _resolve_database_url config = context.config if config.config_file_name is not None: fileConfig(config.config_file_name) -db_url = os.getenv("PRY_DATABASE_URL") -if db_url is not None: - config.set_main_option("sqlalchemy.url", db_url) +# Always use the same URL resolution logic as the application. +config.set_main_option("sqlalchemy.url", os.getenv("PRY_DATABASE_URL") or _resolve_database_url()) target_metadata = Base.metadata exclude_tables = {"spatial_ref_sys"}