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

Compare commits

base: mo:0a42d68853d487c6469381a22fe41e17406a85db
Branches Tags
mo:ph4
mo:prod
mo:ph5
mo:main
..
compare: mo:26315cd407460425953eeef62891648228033ecc
Branches Tags
mo:prod
mo:ph5
mo:ph4
mo:main

No commits in common. "0a42d68853d487c6469381a22fe41e17406a85db" and "26315cd407460425953eeef62891648228033ecc" have entirely different histories.
0a42d68853 ... 26315cd407

1 changed files with 3 additions and 13 deletions
Show Stats Expand all files Collapse all files

16
be/alembic/migrations.py
View File

@ -17,16 +17,6 @@ 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
@ -62,10 +52,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)