diff --git a/plugins/douyu/main.py b/plugins/douyu/main.py index b7eb4be..81154d2 100644 --- a/plugins/douyu/main.py +++ b/plugins/douyu/main.py @@ -170,6 +170,9 @@ class DouyuPlugin(MessagePluginInterface): await self.bot.send_text_message(roomid or sender, "请提供房间号,例如:订阅斗鱼 7718843", sender) return True, "命令格式错误" room_id = parts[1].strip() + if not room_id.isdigit(): + await self.bot.send_text_message(roomid or sender, "房间号必须是数字,例如:斗鱼订阅 52876", sender) + return True, "命令格式错误" ok = self.redis_manager.add_group_room(roomid or sender, room_id) await self.bot.send_text_message(roomid or sender, f"✅ 已订阅斗鱼房间 {room_id}", sender) return True, "订阅成功" if ok else "订阅失败" @@ -179,6 +182,9 @@ class DouyuPlugin(MessagePluginInterface): await self.bot.send_text_message(roomid or sender, "请提供房间号,例如:取消订阅斗鱼 7718843", sender) return True, "命令格式错误" room_id = parts[1].strip() + if not room_id.isdigit(): + await self.bot.send_text_message(roomid or sender, "房间号必须是数字,例如:取消斗鱼订阅 52876", sender) + return True, "命令格式错误" ok = self.redis_manager.remove_group_room(roomid or sender, room_id) await self.bot.send_text_message(roomid or sender, f"✅ 已取消订阅斗鱼房间 {room_id}", sender) return True, "取消成功" if ok else "取消失败"