blob: c8782de37dd89ebde42d17c5b072438bd7494940 (
plain)
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
|
from pathlib import Path
from tortoise import Tortoise
from ATRI.log import logger as log
# 临时的实现，寻求更好的方式！欢迎pr
DB_DIR = Path(".") / "data" / "sql"
DB_DIR.mkdir(parents=True, exist_ok=True)
async def run():
from ATRI.database import models
await Tortoise.init(
db_url=f"sqlite://{DB_DIR}/db.sqlite3",
modules={"models": [locals()["models"]]},
)
await Tortoise.generate_schemas()
async def init_database():
log.info("正在初始化数据库...")
await run()
log.success("数据库初始化完成")
async def close_database_connection():
log.info("正在关闭数据库连接...")
await Tortoise.close_connections()
log.info("数据库成功关闭")
|
