🐛 fix(alembic): remove await_fallback() use

Author: ProgramRipper

nonebot_plugin_orm/param.py

@@ -2,9 +2,9 @@
 
 import sys
 from itertools import repeat
-from typing import Any, cast
 from dataclasses import dataclass
 from operator import methodcaller
+from typing import Any, Type, cast
 from inspect import Parameter, isclass
 
 from pydantic.fields import FieldInfo
@@ -166,7 +166,7 @@ def _check_param(
         if depends_inner is not None:
             dependency = compile_dependency(depends_inner.dependency, option)
         elif all(map(isclass, models)) and all(map(issubclass, models, repeat(Model))):
-            models = cast(Tuple[Model, ...], models)
+            models = cast(Tuple[Type[Model], ...], models)
             dependency = compile_dependency(
                 select(*models).where(
                     *(

nonebot_plugin_orm/templates/generic/env.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
+import asyncio
 from typing import cast
 
 from alembic import context
 from sqlalchemy import Connection
-from sqlalchemy.util import await_fallback
+from sqlalchemy.util import await_only
 from sqlalchemy.ext.asyncio import AsyncEngine
 
 from nonebot_plugin_orm.env import no_drop_table
@@ -81,4 +82,9 @@ async def run_migrations_online() -> None:
 if context.is_offline_mode():
     run_migrations_offline()
 else:
-    await_fallback(run_migrations_online())
+    coro = run_migrations_online()
+
+    try:
+        asyncio.run(coro)
+    except RuntimeError:
+        await_only(coro)

nonebot_plugin_orm/templates/multidb/env.py

@@ -2,9 +2,10 @@
 
 import asyncio
 from typing import cast
+from contextlib import suppress
 
 from alembic import context
-from sqlalchemy.util import await_fallback
+from sqlalchemy.util import await_only
 from sqlalchemy.ext.asyncio import AsyncEngine, AsyncConnection
 from sqlalchemy import MetaData, Connection, TwoPhaseTransaction
 
@@ -124,4 +125,9 @@ async def run_migrations_online() -> None:
 if context.is_offline_mode():
     run_migrations_offline()
 else:
-    await_fallback(run_migrations_online())
+    coro = run_migrations_online()
+
+    with suppress(RuntimeError):
+        asyncio.run(coro)
+
+    await_only(coro)