🐛 fix(alembic): dispose engine

fix #33

Author: ProgramRipper

nonebot_plugin_orm/templates/generic/env.py

@@ -70,6 +70,8 @@ async def run_migrations_online() -> None:
     async with engine.connect() as connection:
         await connection.run_sync(do_run_migrations)
 
+    await engine.dispose()
+
 
 if context.is_offline_mode():
     run_migrations_offline()

nonebot_plugin_orm/templates/multidb/env.py

@@ -113,7 +113,7 @@ async def run_migrations_online() -> None:
         raise
     finally:
         await asyncio.gather(*(conn.close() for conn in conns.values()))
-
+        await asyncio.gather(*(engine.dispose() for engine in engines.values()))
 
 if context.is_offline_mode():
     run_migrations_offline()