Initial commit

This commit is contained in:
shiyu
2026-02-09 13:19:28 +08:00
commit 17d13999b0
300 changed files with 48565 additions and 0 deletions

3
db/__init__.py Normal file
View File

@@ -0,0 +1,3 @@
from .session import init_db, close_db
__all__ = ["init_db", "close_db"]

22
db/session.py Normal file
View File

@@ -0,0 +1,22 @@
from tortoise import Tortoise
from domain.adapters import runtime_registry
TORTOISE_ORM = {
"connections": {"default": "sqlite://data/db/db.sqlite3"},
"apps": {
"models": {
"models": ["models.database"],
"default_connection": "default",
}
},
}
async def init_db():
await Tortoise.init(config=TORTOISE_ORM)
await Tortoise.generate_schemas()
await runtime_registry.refresh()
async def close_db():
await Tortoise.close_connections()