diff options
author | Kyomotoi <[email protected]> | 2022-06-14 21:12:20 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2022-06-14 21:12:20 +0800 |
commit | b1a78c026ec409bea48d767b65b5b7e4e182d904 (patch) | |
tree | 9a2aa5e25406875ae1569d21d06c1202bc9dac80 /ATRI/plugins/bilibili_dynamic/database/db.py | |
parent | d439e315c45aa805c8f2ce75430afe577eaa7fff (diff) | |
download | ATRI-b1a78c026ec409bea48d767b65b5b7e4e182d904.tar.gz ATRI-b1a78c026ec409bea48d767b65b5b7e4e182d904.tar.bz2 ATRI-b1a78c026ec409bea48d767b65b5b7e4e182d904.zip |
🚚💩 更改数据库位置 & 优化代码
Diffstat (limited to 'ATRI/plugins/bilibili_dynamic/database/db.py')
-rw-r--r-- | ATRI/plugins/bilibili_dynamic/database/db.py | 49 |
1 files changed, 0 insertions, 49 deletions
diff --git a/ATRI/plugins/bilibili_dynamic/database/db.py b/ATRI/plugins/bilibili_dynamic/database/db.py deleted file mode 100644 index 37ed223..0000000 --- a/ATRI/plugins/bilibili_dynamic/database/db.py +++ /dev/null @@ -1,49 +0,0 @@ -from pathlib import Path -from tortoise import Tortoise - -from ATRI import driver -from .models import Subscription - - -DB_DIR = Path(".") / "data" / "database" / "bilibili_dynamic" -DB_DIR.mkdir(parents=True, exist_ok=True) - - -class DB: - async def __aenter__(self): - return self - - async def __aexit__(self, exc_type, exc_val, exc_tb): - pass - - async def init(self): - from . import models - - await Tortoise.init( - db_url=f"sqlite://{DB_DIR}/db.sqlite3", - modules={"models": [locals()["models"]]}, - ) - await Tortoise.generate_schemas() - - async def add_sub(self, uid: int, group_id: int): - await Subscription.create(uid=uid, group_id=group_id) - - async def update_sub(self, uid: int, update_map: dict): - await Subscription.filter(uid=uid).update(**update_map) - - async def del_sub(self, query_map: dict): - await Subscription.filter(**query_map).delete() - - async def get_sub_list(self, query_map: dict) -> list: - return await Subscription.filter(**query_map) - - async def get_all_subs(self) -> list: - return await Subscription.all() - - -async def init(): - async with DB() as db: - await db.init() - - -driver().on_startup(init) |