diff options
| author | Kyomotoi <0w0@imki.moe> | 2023-04-06 16:07:46 +0800 |
|---|---|---|
| committer | Kyomotoi <0w0@imki.moe> | 2023-04-06 16:07:46 +0800 |
| commit | 60be1eb2bd16fc89827a50ad18377642aae176be (patch) | |
| tree | 657a08ad802b2b67c2f5d941d93b67c67160b315 /ATRI/plugins/rss/rss_mikanan/data_source.py | |
| parent | 170211bc72311fc4524e7c731303645eac2aa041 (diff) | |
| download | ATRI-60be1eb2bd16fc89827a50ad18377642aae176be.tar.gz ATRI-60be1eb2bd16fc89827a50ad18377642aae176be.tar.bz2 ATRI-60be1eb2bd16fc89827a50ad18377642aae176be.zip | |
🎨 优化数据库代码
Diffstat (limited to 'ATRI/plugins/rss/rss_mikanan/data_source.py')
| -rw-r--r-- | ATRI/plugins/rss/rss_mikanan/data_source.py | 18 |
1 files changed, 7 insertions, 11 deletions
diff --git a/ATRI/plugins/rss/rss_mikanan/data_source.py b/ATRI/plugins/rss/rss_mikanan/data_source.py index 6aec1b5..289d4f3 100644 --- a/ATRI/plugins/rss/rss_mikanan/data_source.py +++ b/ATRI/plugins/rss/rss_mikanan/data_source.py @@ -2,44 +2,40 @@ import xmltodict from ATRI.exceptions import RssError from ATRI.utils import request, gen_random_str +from ATRI.database import DatabaseWrapper, RssMikananiSubcription -from .db import DB +DB = DatabaseWrapper(RssMikananiSubcription) class RssMikananSubscriptor: async def __add_sub(self, _id: str, group_id: int): try: - async with DB() as db: - await db.add_sub(_id, group_id) + await DB.add_sub(_id=_id, group_id=group_id) except Exception: raise RssError("rss.mikan: 添加订阅失败") async def update_sub(self, _id: str, group_id: int, update_map: dict): try: - async with DB() as db: - await db.update_sub(_id, group_id, update_map) + await DB.update_sub(update_map=update_map, _id=_id, group_id=group_id) except Exception: raise RssError("rss.mikan: 更新订阅失败") async def __del_sub(self, _id: str, group_id: int): try: - async with DB() as db: - await db.del_sub({"_id": _id, "group_id": group_id}) + await DB.del_sub({"_id": _id, "group_id": group_id}) except Exception: raise RssError("rss.mikan: 删除订阅失败") async def get_sub_list(self, query_map: dict) -> list: 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 RssError("rss.mikan: 获取订阅列表失败") 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 RssError("rss.mikan: 获取所有订阅失败") |