diff --git a/gso/migrations/env.py b/gso/migrations/env.py index a1f9b9fcdbb9088dc846a4273a9a12eff786baf6..4d84cfb15787fc357dd96857fb97b4cee13b80a8 100644 --- a/gso/migrations/env.py +++ b/gso/migrations/env.py @@ -1,11 +1,9 @@ import logging -import os -import orchestrator from alembic import context from orchestrator.db.database import BaseModel from orchestrator.settings import app_settings -from sqlalchemy import engine_from_config, pool +from sqlalchemy import engine_from_config, pool, text # this is the Alembic Config object, which provides # access to the values within the .ini file in use. @@ -17,17 +15,8 @@ logger = logging.getLogger("alembic.env") config.set_main_option("sqlalchemy.url", app_settings.DATABASE_URI) -# add your model's MetaData object here -# for 'autogenerate' support -# from myapp import mymodel -# target_metadata = mymodel.Base.metadata target_metadata = BaseModel.metadata -# other values from the config, defined by the needs of env.py, -# can be acquired: -# my_important_option = config.get_main_option("my_important_option") -# ... etc. - def run_migrations_offline() -> None: """Run migrations in 'offline' mode. @@ -85,8 +74,10 @@ def run_migrations_online() -> None: ) try: with context.begin_transaction(): + connection.execute(text("SELECT pg_advisory_xact_lock(1000);")) context.run_migrations() finally: + connection.execute(text("SELECT pg_advisory_unlock(1000);")) connection.close()