From e7e34cda5400031e3ed4a5c5902c1467e4fba989 Mon Sep 17 00:00:00 2001 From: shiyu Date: Tue, 23 Dec 2025 09:53:31 +0800 Subject: [PATCH] feat(db): add patch for aiosqlite to ensure proper connection handling --- db/session.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) 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()