From d07348c2a2dd551c00aa330fcc1015155854e874 Mon Sep 17 00:00:00 2001
From: Kyomotoi <kyomotoiowo@gmail.com>
Date: Mon, 4 Apr 2022 17:03:25 +0800
Subject: =?UTF-8?q?=F0=9F=9A=B8=20=E6=9B=B4=E6=94=B9=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=BA=93=E4=BD=8D=E7=BD=AE=20=E6=96=B9=E9=9D=A2=E6=9B=B4?=
 =?UTF-8?q?=E6=96=B0=E6=97=B6=E4=B8=80=E9=94=AE=E8=BF=81=E7=A7=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/database/db.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'ATRI/database')

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
-- 
cgit v1.2.3