diff options
Diffstat (limited to 'shared/alembic/env.py')
| -rw-r--r-- | shared/alembic/env.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/shared/alembic/env.py b/shared/alembic/env.py index 14303f6..18687f2 100644 --- a/shared/alembic/env.py +++ b/shared/alembic/env.py @@ -1,6 +1,7 @@ """Alembic environment configuration for async PostgreSQL migrations.""" import asyncio +import os from logging.config import fileConfig from alembic import context @@ -14,6 +15,13 @@ config = context.config if config.config_file_name is not None: fileConfig(config.config_file_name) +# Override URL from DATABASE_URL env var if set +database_url = os.environ.get("DATABASE_URL") +if database_url: + if database_url.startswith("postgresql://"): + database_url = database_url.replace("postgresql://", "postgresql+asyncpg://", 1) + config.set_main_option("sqlalchemy.url", database_url) + target_metadata = Base.metadata |