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

refactor: Update migration functions to accept connection parameter for improved flexibility and consistency #30

Merged
mo merged 1 commits from ph4 into prod 2025-06-01 17:39:22 +02:00
Conversation 0 Commits 1 Files Changed 1 +4 -4
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
be/alembic/migrations.py
View File

@ -26,9 +26,9 @@ async def run_migrations():
async with engine.connect() as connection: async with engine.connect() as connection:
# Get current database schema version # Get current database schema version
def do_get_current_rev(): def do_get_current_rev(conn):
migration_context = MigrationContext.configure( migration_context = MigrationContext.configure(
connection, conn,
opts={ opts={
'target_metadata': DatabaseBase.metadata, 'target_metadata': DatabaseBase.metadata,
'compare_type': True, 'compare_type': True,
@ -40,9 +40,9 @@ async def run_migrations():
current_rev = await connection.run_sync(do_get_current_rev) current_rev = await connection.run_sync(do_get_current_rev)
# Run migrations # Run migrations
def do_upgrade(): def do_upgrade(conn):
migration_context = MigrationContext.configure( migration_context = MigrationContext.configure(
connection, conn,
opts={ opts={
'target_metadata': DatabaseBase.metadata, 'target_metadata': DatabaseBase.metadata,
'compare_type': True, 'compare_type': True,