diff options
| author | SakuraMemory <[email protected]> | 2022-04-05 16:18:34 +0800 |
|---|---|---|
| committer | GitHub <[email protected]> | 2022-04-05 16:18:34 +0800 |
| commit | ceb7b3a5b11d3978c46185af1cf982a2eb3ca961 (patch) | |
| tree | 0e509c76a575db8a4ab485d503ea7bf3e73b5383 /ATRI/database | |
| parent | a2f3925fc63d12ad8daefa6ba30b18bc81693c99 (diff) | |
| parent | 93e71022a5987ca898ca42a84e10cf3a1ddacd0a (diff) | |
| download | ATRI-ceb7b3a5b11d3978c46185af1cf982a2eb3ca961.tar.gz ATRI-ceb7b3a5b11d3978c46185af1cf982a2eb3ca961.tar.bz2 ATRI-ceb7b3a5b11d3978c46185af1cf982a2eb3ca961.zip | |
Merge pull request #5 from Kyomotoi/main
sync main
Diffstat (limited to 'ATRI/database')
| -rw-r--r-- | ATRI/database/db.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/ATRI/database/db.py b/ATRI/database/db.py index c2aa015..68db8cf 100644 --- a/ATRI/database/db.py +++ b/ATRI/database/db.py @@ -1,7 +1,12 @@ +from pathlib import Path from tortoise import Tortoise +from nonebot import get_driver from ATRI.database import models -from nonebot import get_driver + + +DB_DIR = Path(".") / "data" / "database" / "sql" +DB_DIR.mkdir(exist_ok=True) # 关于数据库的操作类，只实现与数据库有关的CRUD @@ -17,7 +22,7 @@ class DB: from ATRI.database import models await Tortoise.init( - db_url="sqlite://ATRI/database/db.sqlite3", + db_url=f"sqlite://{DB_DIR}/db.sqlite3", modules={"models": [locals()["models"]]}, ) # Generate the schema |