diff --git a/db/session.py b/db/session.py index 373860a..659ee8a 100644 --- a/db/session.py +++ b/db/session.py @@ -13,7 +13,21 @@ TORTOISE_ORM = { } +def patch_aiosqlite_for_tortoise() -> None: + import aiosqlite + + if hasattr(aiosqlite.Connection, "start"): + return + + def start(self) -> None: # type: ignore[no-redef] + if not self._thread.is_alive(): + self._thread.start() + + aiosqlite.Connection.start = start # type: ignore[attr-defined] + + async def init_db(): + patch_aiosqlite_for_tortoise() await Tortoise.init(config=TORTOISE_ORM) await Tortoise.generate_schemas() await runtime_registry.refresh()