diff --git a/plugins/douyu/main.py b/plugins/douyu/main.py index 403d698..e315247 100644 --- a/plugins/douyu/main.py +++ b/plugins/douyu/main.py @@ -127,7 +127,7 @@ class DouyuPlugin(MessagePluginInterface): self.bot: WechatAPIClient = None self.feature = self.register_feature() self.redis_manager: Optional[DouyuRedisManager] = None - self._commands = ["斗鱼订阅", "取消斗鱼订阅", "斗鱼订阅列表", "斗鱼订阅提醒"] + self._commands = ["斗鱼订阅", "取消斗鱼订阅", "斗鱼订阅列表", "斗鱼订阅提醒", "取消斗鱼订阅提醒"] self._api_template = "https://www.douyu.com/betard/{room_id}" self._user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36" self._check_interval = 5 @@ -189,6 +189,13 @@ class DouyuPlugin(MessagePluginInterface): ok = self.redis_manager.add_group_subscriber(roomid, sender) await self.bot.send_at_message(roomid, "已加入斗鱼订阅提醒名单", [sender]) return True, "加入提醒名单成功" if ok else "加入提醒名单失败" + if first_token == "取消斗鱼订阅提醒": + if not roomid: + await self.bot.send_text_message(sender, "请在群聊中使用该命令", sender) + return True, "仅支持群聊" + ok = self.redis_manager.remove_group_subscriber(roomid, sender) + await self.bot.send_at_message(roomid, "已取消斗鱼订阅提醒", [sender]) + return True, "取消提醒成功" if ok else "取消提醒失败" if first_token == "斗鱼订阅": parts = content.split() if len(parts) < 2: