File size: 1,145 Bytes
9d4bd7c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
from tortoise import Tortoise
DATABASE_URL = "sqlite://db.sqlite3"
TORTOISE_ORM = {
"connections": {"default": DATABASE_URL},
"apps": {
"models": {
"models": [
'App.routers.stocks.models',
'App.routers.tasks.models',
'App.routers.utt.models',
'App.routers.users.models',
'App.routers.portfolio.models',
'App.routers.bonds.models',
"aerich.models"
],
"default_connection": "default",
}
}
}
async def init_db():
await Tortoise.init(
db_url=DATABASE_URL,
modules={'models': [
'App.routers.stocks.models',
'App.routers.tasks.models',
'App.routers.utt.models',
'App.routers.users.models',
'App.routers.portfolio.models',
'App.routers.bonds.models'
]}
)
await Tortoise.generate_schemas()
async def close_db():
await Tortoise.close_connections()
async def clear_db():
for model in Tortoise.apps.get('models').values():
await model.all().delete() |