mo/mitlist
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'refactor: Introduce migration function to streamline upgrade steps in Alembic migrations' (#32) from ph4 into prod

Browse Source 
Reviewed-on: #32
This commit is contained in:
mo 2025-06-01 17:45:43 +02:00
commit 0a42d68853
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
be/alembic/migrations.py
View File

@ -17,6 +17,16 @@ sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__), '..'
from app.database import Base as DatabaseBase from app.database import Base as DatabaseBase
from app.config import settings from app.config import settings
def _get_migration_fn(script_directory, current_rev):
"""Create a migration function that knows how to upgrade from current revision."""
def migration_fn(rev, context):
# Get all upgrade steps from current revision to head
revisions = script_directory._upgrade_revs("head", current_rev)
for revision in revisions:
script = script_directory.get_revision(revision)
script.module.upgrade(context)
return migration_fn
async def run_migrations(): async def run_migrations():
"""Run database migrations asynchronously.""" """Run database migrations asynchronously."""
# Get alembic configuration and script directory # Get alembic configuration and script directory
@ -52,10 +62,10 @@ async def run_migrations():
} }
) )
# Set the migration function
migration_context._migrations_fn = _get_migration_fn(script_directory, current_rev)
with migration_context.begin_transaction(): with migration_context.begin_transaction():
migration_context._migrations_fn = script_directory._upgrade_revs(
"head", current_rev
)
migration_context.run_migrations() migration_context.run_migrations()
await connection.run_sync(upgrade_to_head) await connection.run_sync(upgrade_to_head)