diff options
| author | Kyomotoi <0w0@imki.moe> | 2022-10-05 20:03:42 +0800 |
|---|---|---|
| committer | Kyomotoi <0w0@imki.moe> | 2022-10-05 20:03:42 +0800 |
| commit | 8725dc30c520988b0f9e97f1b224bc030715207d (patch) | |
| tree | bbc935c42ea40cd173e242238e33026c0b84d12e /ATRI/plugins/rss/rss_mikanan/data_source.py | |
| parent | 43a8d02c4122a0d236c5cbd7b773a8c77715b244 (diff) | |
| download | ATRI-8725dc30c520988b0f9e97f1b224bc030715207d.tar.gz ATRI-8725dc30c520988b0f9e97f1b224bc030715207d.tar.bz2 ATRI-8725dc30c520988b0f9e97f1b224bc030715207d.zip | |
✨ 添加 rss 对 Mikan 的支持
Diffstat (limited to 'ATRI/plugins/rss/rss_mikanan/data_source.py')
| -rw-r--r-- | ATRI/plugins/rss/rss_mikanan/data_source.py | 105 |
1 files changed, 105 insertions, 0 deletions
diff --git a/ATRI/plugins/rss/rss_mikanan/data_source.py b/ATRI/plugins/rss/rss_mikanan/data_source.py index e69de29..eadf97b 100644 --- a/ATRI/plugins/rss/rss_mikanan/data_source.py +++ b/ATRI/plugins/rss/rss_mikanan/data_source.py @@ -0,0 +1,105 @@ +import xmltodict + +from nonebot.permission import SUPERUSER +from nonebot.adapters.onebot.v11 import GROUP_OWNER, GROUP_ADMIN + + +from ATRI.service import Service +from ATRI.rule import is_in_service +from ATRI.exceptions import RssError +from ATRI.utils import request, gen_random_str + + +from .db import DB + + +class RssMikananSubscriptor(Service): + def __init__(self): + Service.__init__( + self, + "rss.mikan", + "Rss的mikan支持", + rule=is_in_service("rss.mikan"), + permission=SUPERUSER | GROUP_OWNER | GROUP_ADMIN, + main_cmd="/rss.mikan", + ) + + async def __add_sub(self, _id: str, group_id: int): + try: + async with DB() as db: + await db.add_sub(_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) + 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}) + 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) + 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() + except Exception: + raise RssError("rss.mikan: 获取所有订阅失败") + + async def add_sub(self, url: str, group_id: int) -> str: + data = await self.get_mikan_info(url) + if not data: + return "该链接不含mikan内容" + + rss_link = data["link"] + + query_result = await self.get_sub_list( + {"rss_link": rss_link, "group_id": group_id} + ) + if query_result: + _id = query_result[0]._id + return f"该链接已经订阅过啦! ID: {_id}" + + _id = gen_random_str(6) + title = data["title"] + disc = data["description"] + + await self.__add_sub(_id, group_id) + await self.update_sub( + _id, group_id, {"title": title, "rss_link": rss_link, "discription": disc} + ) + return f"订阅成功! ID: {_id}" + + async def del_sub(self, _id: str, group_id: int) -> str: + query_result = await self.get_sub_list({"_id": _id, "group_id": group_id}) + if not query_result: + return "没有找到订阅..." + + await self.__del_sub(_id, group_id) + return f"成功取消ID为 {_id} 的订阅" + + async def get_mikan_info(self, url: str) -> dict: + if "mikanani.me" not in url: + return dict() + + try: + resp = await request.get(url) + except Exception: + RssError("rss.mikan: 请求链接失败") + + xml_data = resp.read() + data = xmltodict.parse(xml_data) + return data["rss"]["@version"]["channel"] |