From 60be1eb2bd16fc89827a50ad18377642aae176be Mon Sep 17 00:00:00 2001
From: Kyomotoi <0w0@imki.moe>
Date: Thu, 6 Apr 2023 16:07:46 +0800
Subject: =?UTF-8?q?=F0=9F=8E=A8=20=E4=BC=98=E5=8C=96=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=BA=93=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/bilibili_dynamic/data_source.py | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

(limited to 'ATRI/plugins/bilibili_dynamic/data_source.py')

diff --git a/ATRI/plugins/bilibili_dynamic/data_source.py b/ATRI/plugins/bilibili_dynamic/data_source.py
index d629454..15babe5 100644
--- a/ATRI/plugins/bilibili_dynamic/data_source.py
+++ b/ATRI/plugins/bilibili_dynamic/data_source.py
@@ -5,8 +5,8 @@ from operator import itemgetter
 from ATRI.message import MessageBuilder
 from ATRI.utils import TimeDealer
 from ATRI.exceptions import BilibiliDynamicError
+from ATRI.database import DatabaseWrapper, BilibiliSubscription
 
-from .db import DB
 from .api import API
 
 
@@ -17,27 +17,25 @@ _OUTPUT_FORMAT = (
     .text("链接: {up_dy_link}")
     .done()
 )
+DB = DatabaseWrapper(BilibiliSubscription)
 
 
 class BilibiliDynamicSubscriptor:
     async def __add_sub(self, uid: int, group_id: int):
         try:
-            async with DB() as db:
-                await db.add_sub(uid, group_id)
+            await DB.add_sub(uid=uid, group_id=group_id)
         except Exception:
             raise BilibiliDynamicError("添加订阅失败")
 
     async def update_sub(self, uid: int, group_id: int, update_map: dict):
         try:
-            async with DB() as db:
-                await db.update_sub(uid, group_id, update_map)
+            await DB.update_sub(update_map=update_map, uid=uid, group_id=group_id)
         except Exception:
             raise BilibiliDynamicError("更新订阅失败")
 
     async def __del_sub(self, uid: int, group_id: int):
         try:
-            async with DB() as db:
-                await db.del_sub({"uid": uid, "group_id": group_id})
+            await DB.del_sub({"uid": uid, "group_id": group_id})
         except Exception:
             raise BilibiliDynamicError("删除订阅失败")
 
@@ -48,15 +46,13 @@ class BilibiliDynamicSubscriptor:
             query_map = {"uid": uid, "group_id": group_id}
 
         try:
-            async with DB() as db:
-                return await db.get_sub_list(query_map)
+            return await DB.get_sub_list(query_map)
         except Exception:
             raise BilibiliDynamicError("获取订阅列表失败")
 
     async def get_all_subs(self) -> list:
         try:
-            async with DB() as db:
-                return await db.get_all_subs()
+            return await DB.get_all_subs()
         except Exception:
             raise BilibiliDynamicError("获取全部订阅列表失败")
 
-- 
cgit v1.2.3