加入斗鱼开播下播提醒

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: