diff --git a/backend/app/migrations/run.py b/backend/app/migrations/run.py index 8321175..de16fc5 100644 --- a/backend/app/migrations/run.py +++ b/backend/app/migrations/run.py @@ -1,16 +1,21 @@ """Entry point for running migrations: python -m app.migrations.run""" import asyncio +from app.core.database import connect_to_mongo, close_mongo_connection from app.migrations.migrator import MigrationManager async def main() -> None: - mgr = MigrationManager() - applied = await mgr.migrate_up() - if applied: - print(f"Applied {len(applied)} migration(s): {applied}") - else: - print("Already up to date — no pending migrations.") + await connect_to_mongo() + try: + mgr = MigrationManager() + applied = await mgr.migrate_up() + if applied: + print(f"Applied {len(applied)} migration(s): {applied}") + else: + print("Already up to date — no pending migrations.") + finally: + await close_mongo_connection() if __name__ == "__main__":