From 920173c8faaab5a65459ce176d36812bac6feb08 Mon Sep 17 00:00:00 2001 From: Yuki-Asuuna <10174503104@stu.ecnu.edu.cn> Date: Fri, 25 Feb 2022 21:22:48 +0800 Subject: =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0b=E7=AB=99=E5=8A=A8=E6=80=81?= =?UTF-8?q?=E8=AE=A2=E9=98=85=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Change-Id: I8b74e3a286901379b8337e33d1b581524cb80d97 --- ATRI/plugins/bilibili_dynamic/__init__.py | 169 ++++++++++++++++++++++++++++++ 1 file changed, 169 insertions(+) create mode 100644 ATRI/plugins/bilibili_dynamic/__init__.py (limited to 'ATRI/plugins/bilibili_dynamic/__init__.py') diff --git a/ATRI/plugins/bilibili_dynamic/__init__.py b/ATRI/plugins/bilibili_dynamic/__init__.py new file mode 100644 index 0000000..87854c0 --- /dev/null +++ b/ATRI/plugins/bilibili_dynamic/__init__.py @@ -0,0 +1,169 @@ +from ATRI.utils.apscheduler import scheduler +from ATRI.utils import timestamp2datetime + +from nonebot.params import State +from nonebot.adapters.onebot.v11 import MessageSegment, GroupMessageEvent, Message +from nonebot.typing import T_State +from nonebot import get_bot + +from .data_source import BilibiliDynamicSubscriptor +import re +from tabulate import tabulate + +bilibili_dynamic = BilibiliDynamicSubscriptor().on_command( + "/bilibili_dynamic", "b站动态订阅助手", aliases={"b站动态"} +) + +__help__ = """欢迎使用【b站动态订阅助手】～ +目前支持的功能如下...请选择： +1.添加订阅 +2.取消订阅 +3.订阅列表 +----------------------------------- +用法示例1：/bilibili_dynamic 添加订阅 +用法示例2：/bilibili_dynamic 取消订阅 401742377（数字uid） +用法示例3：/bilibili_dynamic 订阅列表""" + + +def help() -> str: + return __help__ + + +@bilibili_dynamic.handle() +async def _menu(event: GroupMessageEvent, state: T_State = State()): + args = str(event.get_plaintext()).strip().lower().split()[1:] + # print(args) + if not args: + await bilibili_dynamic.finish(help()) + elif args and len(args) == 1: + state["sub_command"] = args[0] + elif args and len(args) == 2: + state["sub_command"] = args[0] + state["uid"] = args[1] + else: + await bilibili_dynamic.finish("参数错误QAQ 请检查您的输入～") + + +@bilibili_dynamic.got("sub_command", prompt="您要执行操作是?\n【添加订阅/取消订阅/订阅列表】") +async def handle_subcommand(event: GroupMessageEvent, state: T_State = State()): + if state["sub_command"] not in ["添加订阅", "取消订阅", "订阅列表"]: + await bilibili_dynamic.finish("没有这个命令哦, 请在【添加订阅/取消订阅/订阅列表】中选择并重新发送") + + if state["sub_command"] == "订阅列表": + subscriptor = BilibiliDynamicSubscriptor() + # print(event.group_id) + r = await subscriptor.get_subscriptions(query_map={"groupid": event.group_id}) + subs = [] + for s in r: + subs.append([s.nickname, s.uid, s.last_update]) + output = "本群订阅的UP列表如下～\n" + tabulate( + subs, headers=["up名称", "UID", "上次更新时间"], tablefmt="plain", showindex=True + ) + await bilibili_dynamic.finish(output) + + +@bilibili_dynamic.got("uid", prompt="请输入b站UID（输入-1取消）:") +async def handle_uid(event: GroupMessageEvent, state: T_State = State()): + sub_command = state["sub_command"] + if isinstance(state["uid"], list): + uid = str(state["uid"][0]) + else: + uid = state["uid"] + + if uid == "-1": + await bilibili_dynamic.finish("已经成功退出订阅~") + # print(state) + # print(uid) + if not re.match(r"^\d+$", uid): + await bilibili_dynamic.reject("这似乎不是UID呢, 请重新输入:") + uid = int(uid) + subscriptor = BilibiliDynamicSubscriptor() + up_name = await subscriptor.get_upname_by_uid(uid) + if up_name == "": + await bilibili_dynamic.finish(f"无法获取uid={uid}的up信息...订阅失败了".format(uid=uid)) + else: + await bilibili_dynamic.send( + f"uid为{uid}的UP主是【{up_name}】\n{sub_command}操作中...".format( + uid=uid, up_name=up_name, sub_command=sub_command + ) + ) + query_result = await subscriptor.get_subscriptions( + query_map={"uid": uid, "groupid": event.group_id} + ) + success = True + if sub_command == "添加订阅": + if len(query_result) > 0: + await bilibili_dynamic.finish( + f"订阅失败，因为uid={uid}的UP主【{up_name}】已在本群订阅列表中".format( + uid=uid, up_name=up_name + ) + ) + success = await subscriptor.add_subscription(uid, event.group_id) + print(success) + success = success and ( + await subscriptor.update_subscription_by_uid( + uid=uid, update_map={"nickname": up_name} + ) + ) + elif sub_command == "取消订阅": + if len(query_result) == 0: + await bilibili_dynamic.finish( + f"取消订阅失败，因为uid={uid}的UP主【{up_name}】不在本群订阅列表中".format( + uid=uid, up_name=up_name + ) + ) + success = await subscriptor.remove_subscription(uid, event.group_id) + if success: + await bilibili_dynamic.finish( + f"成功{sub_command}【{up_name}】的动态!".format( + sub_command=sub_command, up_name=up_name + ) + ) + else: + await bilibili_dynamic.finish("诶...因为神奇的原因失败了") + + +from queue import Queue + +# 任务队列（taskQueue） +tq = Queue() + + +# 业务逻辑 +# 每10s从任务队列中拉一个uid出来，调用api进行查询 +# 当任务队列为空时，从数据库读取订阅列表，并塞入任务队列tq中 +@scheduler.scheduled_job( + "interval", name="b站动态检查", seconds=10, max_instances=3, misfire_grace_time=60 +) +async def _check_dynamic(): + from ATRI.database.models import Subscription + + subscriptor = BilibiliDynamicSubscriptor() + all_dynamic = await subscriptor.get_all_subscriptions() + if tq.empty(): + for d in all_dynamic: + tq.put(d) + else: + d: Subscription = tq.get() + ts = int(d.last_update.timestamp()) + info: dict = await subscriptor.get_recent_dynamic_by_uid(d.uid) + res = [] + if info: + if info.get("cards") is not None: + res = subscriptor.extract_dynamics_detail(info.get("cards")) + res = res[::-1] + for i in res: + i["name"] = d.nickname + if ts < i["timestamp"]: + text, pic_url = subscriptor.generate_output(pattern=i) + # print(text,pic_url) + output = Message( + [MessageSegment.text(text), MessageSegment.image(pic_url)] + ) + bot = get_bot() + await bot.send_group_msg(group_id=d.groupid, message=output) + _ = await subscriptor.update_subscription_by_uid( + uid=d.uid, + update_map={"last_update": timestamp2datetime(i["timestamp"])}, + ) + break -- cgit v1.2.3 From 3db938fc748a56e6de666e7c298b1e6858052bdb Mon Sep 17 00:00:00 2001 From: Yuki-Asuuna <10174503104@stu.ecnu.edu.cn> Date: Tue, 1 Mar 2022 15:44:04 +0800 Subject: fix: sort to ensure dynamic timestamp with time chronological order Change-Id: I105fc15904591cd89bef7bade38f48c7a7b834eb --- ATRI/plugins/bilibili_dynamic/__init__.py | 1 - 1 file changed, 1 deletion(-) (limited to 'ATRI/plugins/bilibili_dynamic/__init__.py') diff --git a/ATRI/plugins/bilibili_dynamic/__init__.py b/ATRI/plugins/bilibili_dynamic/__init__.py index 87854c0..fa9f8c3 100644 --- a/ATRI/plugins/bilibili_dynamic/__init__.py +++ b/ATRI/plugins/bilibili_dynamic/__init__.py @@ -151,7 +151,6 @@ async def _check_dynamic(): if info: if info.get("cards") is not None: res = subscriptor.extract_dynamics_detail(info.get("cards")) - res = res[::-1] for i in res: i["name"] = d.nickname if ts < i["timestamp"]: -- cgit v1.2.3