diff --git a/plugins/lucky_pot/main.py b/plugins/lucky_pot/main.py index f56426d..c79aef4 100644 --- a/plugins/lucky_pot/main.py +++ b/plugins/lucky_pot/main.py @@ -20,6 +20,7 @@ from utils.robot_cmd.robot_command import Feature, PermissionStatus class LuckyPotRoom: """奖池房间类""" + def __init__(self, room_id: str, entry_fee: int, max_players: int, countdown: int, creator: str, group_id: str): self.room_id = room_id self.status = "waiting" # waiting | drawing | finished @@ -38,36 +39,36 @@ class LuckyPotRoom: """添加玩家到房间""" if self.status != "waiting": return False - + if user_id in self.players: return False - + self.players.append(user_id) self.player_nicknames[user_id] = nickname return True - + def is_full(self) -> bool: """检查房间是否已满""" return len(self.players) >= self.max_players - + def is_empty(self) -> bool: """检查房间是否为空""" return len(self.players) == 0 - + def get_player_count(self) -> int: """获取玩家数量""" return len(self.players) - + def get_total_pot(self) -> int: """获取奖池总额""" return self.entry_fee * len(self.players) - + def get_remaining_time(self) -> int: """获取剩余时间（秒）""" elapsed = int(time.time()) - self.start_time remaining = max(0, self.countdown - elapsed) return remaining - + def cancel_timer(self): """取消倒计时""" if self.timer: @@ -80,7 +81,7 @@ class LuckyPotPlugin(MessagePluginInterface): # 功能权限常量 FEATURE_KEY = "LUCKY_POT" - FEATURE_DESCRIPTION = "🎯 拼手气大奖池 [大奖池, 开奖池, 拼手气]" + FEATURE_DESCRIPTION = "🎯 拼手气大奖池 [大奖池, 开奖池, 拼手气]" @property def name(self) -> str: @@ -103,9 +104,16 @@ class LuckyPotPlugin(MessagePluginInterface): return self._commands @property - def feature(self) -> Optional[Feature]: - """获取功能权限""" - return Feature.get_feature_by_key(self.FEATURE_KEY) + def feature_key(self) -> Optional[str]: + return self.FEATURE_KEY + + @property + def feature_description(self) -> Optional[str]: + return self.FEATURE_DESCRIPTION + + def __init__(self): + super().__init__() + self.feature = self.register_feature() def initialize(self, context: Dict[str, Any]) -> bool: """初始化插件""" @@ -121,7 +129,7 @@ class LuckyPotPlugin(MessagePluginInterface): self._commands = self._config.get("LuckyPot", {}).get("command", ["大奖池", "开奖池", "拼手气"]) self.command_format = self._config.get("LuckyPot", {}).get("command-format", "大奖池 创建/参与/查看") self.enable = self._config.get("LuckyPot", {}).get("enable", True) - + # 游戏参数配置 self.entry_fee = self._config.get("LuckyPot", {}).get("entry_fee", 100) # 入场费 self.max_players = self._config.get("LuckyPot", {}).get("max_players", 10) # 最大玩家数 @@ -130,12 +138,12 @@ class LuckyPotPlugin(MessagePluginInterface): self.lucky_bonus_chance = self._config.get("LuckyPot", {}).get("lucky_bonus_chance", 20) # 欧皇奖励概率 self.lucky_bonus_percent = self._config.get("LuckyPot", {}).get("lucky_bonus_percent", 20) # 欧皇额外奖励比例 self.cooldown = self._config.get("LuckyPot", {}).get("cooldown", 5) # 冷却时间（秒） - + # 活跃房间字典 {group_id: room} self.active_rooms = {} # 玩家冷却时间 {user_id: timestamp} self.player_cooldowns = {} - + self.LOG.info(f"[{self.name}] 插件初始化完成，指令：{self._commands}") return True @@ -201,21 +209,21 @@ class LuckyPotPlugin(MessagePluginInterface): if roomid in self.active_rooms: room = self.active_rooms[roomid] if room.status == "waiting": - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, - f"⚠️ 已有一个活跃的大奖池房间正在等待中！\n" - f"🕒 剩余时间: {room.get_remaining_time()}秒\n" - f"👥 当前人数: {room.get_player_count()}/{room.max_players}\n" - f"💰 当前奖池: {room.get_total_pot()} 积分", - sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + f"⚠️ 已有一个活跃的大奖池房间正在等待中！\n" + f"🕒 剩余时间: {room.get_remaining_time()}秒\n" + f"👥 当前人数: {room.get_player_count()}/{room.max_players}\n" + f"💰 当前奖池: {room.get_total_pot()} 积分", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "已有活跃房间" # 检查用户积分是否足够 user_points = self.points_db.get_user_points(sender, roomid) if user_points["total_points"] < self.entry_fee: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, - f"❌ 积分不足！创建大奖池需要 {self.entry_fee} 积分，您当前只有 {user_points['total_points']} 积分。", - sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + f"❌ 积分不足！创建大奖池需要 {self.entry_fee} 积分，您当前只有 {user_points['total_points']} 积分。", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "积分不足" @@ -241,7 +249,9 @@ class LuckyPotPlugin(MessagePluginInterface): ) if not deduct_success: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "❌ 扣除积分失败，无法创建大奖池！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "❌ 扣除积分失败，无法创建大奖池！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "扣除积分失败" @@ -252,14 +262,14 @@ class LuckyPotPlugin(MessagePluginInterface): self.active_rooms[roomid] = room # 发送创建成功消息 - await self.bot.send_text_message(target, - f"🎯 大奖池已创建！\n" - f"🆔 房间ID: {room_id}\n" - f"💰 入场费: {self.entry_fee} 积分\n" - f"👥 人数上限: {self.max_players} 人\n" - f"⏳ 倒计时: {self.countdown} 秒\n" - f"\n🎮 发送 \"{self._commands[0]} 参与\" 加入游戏！", - sender) + await self.bot.send_text_message(target, + f"🎯 大奖池已创建！\n" + f"🆔 房间ID: {room_id}\n" + f"💰 入场费: {self.entry_fee} 积分\n" + f"👥 人数上限: {self.max_players} 人\n" + f"⏳ 倒计时: {self.countdown} 秒\n" + f"\n🎮 发送 \"{self._commands[0]} 参与\" 加入游戏！", + sender) return True, "创建房间成功" @@ -271,9 +281,9 @@ class LuckyPotPlugin(MessagePluginInterface): # 检查是否有活跃房间 if roomid not in self.active_rooms: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, - f"❌ 当前没有活跃的大奖池房间！\n发送 \"{self._commands[0]} 创建\" 创建一个新的大奖池。", - sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + f"❌ 当前没有活跃的大奖池房间！\n发送 \"{self._commands[0]} 创建\" 创建一个新的大奖池。", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "没有活跃房间" @@ -281,19 +291,25 @@ class LuckyPotPlugin(MessagePluginInterface): # 检查房间状态 if room.status != "waiting": - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "❌ 当前房间已经开始开奖或已结束，无法加入！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "❌ 当前房间已经开始开奖或已结束，无法加入！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "房间状态不允许加入" # 检查是否已在房间中 if sender in room.players: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "⚠️ 您已经在大奖池中了，无需重复加入！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "⚠️ 您已经在大奖池中了，无需重复加入！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "已在房间中" # 检查房间是否已满 if room.is_full(): - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "❌ 大奖池已满员，无法加入！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "❌ 大奖池已满员，无法加入！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "房间已满" @@ -302,18 +318,18 @@ class LuckyPotPlugin(MessagePluginInterface): last_join_time = self.player_cooldowns[sender] if time.time() - last_join_time < self.cooldown: remaining = int(self.cooldown - (time.time() - last_join_time)) - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, - f"⏳ 操作过于频繁，请等待 {remaining} 秒后再试！", - sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + f"⏳ 操作过于频繁，请等待 {remaining} 秒后再试！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "冷却中" # 检查用户积分是否足够 user_points = self.points_db.get_user_points(sender, roomid) if user_points["total_points"] < room.entry_fee: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, - f"❌ 积分不足！参与大奖池需要 {room.entry_fee} 积分，您当前只有 {user_points['total_points']} 积分。", - sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + f"❌ 积分不足！参与大奖池需要 {room.entry_fee} 积分，您当前只有 {user_points['total_points']} 积分。", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "积分不足" @@ -324,7 +340,9 @@ class LuckyPotPlugin(MessagePluginInterface): ) if not deduct_success: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "❌ 扣除积分失败，无法加入大奖池！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "❌ 扣除积分失败，无法加入大奖池！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "扣除积分失败" @@ -333,17 +351,17 @@ class LuckyPotPlugin(MessagePluginInterface): # 添加用户到房间 room.add_player(sender, nickname) - + # 更新冷却时间 self.player_cooldowns[sender] = time.time() # 发送加入成功消息 - await self.bot.send_text_message(target, - f"✅ {nickname} 成功加入大奖池！\n" - f"👥 当前人数: {room.get_player_count()}/{room.max_players}\n" - f"💰 当前奖池: {room.get_total_pot()} 积分\n" - f"⏳ 剩余时间: {room.get_remaining_time()} 秒", - sender) + await self.bot.send_text_message(target, + f"✅ {nickname} 成功加入大奖池！\n" + f"👥 当前人数: {room.get_player_count()}/{room.max_players}\n" + f"💰 当前奖池: {room.get_total_pot()} 积分\n" + f"⏳ 剩余时间: {room.get_remaining_time()} 秒", + sender) # 如果房间已满，立即开奖 if room.is_full(): @@ -360,9 +378,9 @@ class LuckyPotPlugin(MessagePluginInterface): # 检查是否有活跃房间 if roomid not in self.active_rooms: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, - f"❌ 当前没有活跃的大奖池房间！\n发送 \"{self._commands[0]} 创建\" 创建一个新的大奖池。", - sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + f"❌ 当前没有活跃的大奖池房间！\n发送 \"{self._commands[0]} 创建\" 创建一个新的大奖池。", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "没有活跃房间" @@ -371,15 +389,15 @@ class LuckyPotPlugin(MessagePluginInterface): # 显示房间信息 if room.status == "waiting": player_list = "\n".join([f"👤 {room.player_nicknames[p]}" for p in room.players]) - await self.bot.send_text_message(target, - f"🎯 大奖池状态: 等待中\n" - f"🆔 房间ID: {room.room_id}\n" - f"💰 入场费: {room.entry_fee} 积分\n" - f"👥 当前人数: {room.get_player_count()}/{room.max_players}\n" - f"💰 当前奖池: {room.get_total_pot()} 积分\n" - f"⏳ 剩余时间: {room.get_remaining_time()} 秒\n\n" - f"参与玩家:\n{player_list}", - sender) + await self.bot.send_text_message(target, + f"🎯 大奖池状态: 等待中\n" + f"🆔 房间ID: {room.room_id}\n" + f"💰 入场费: {room.entry_fee} 积分\n" + f"👥 当前人数: {room.get_player_count()}/{room.max_players}\n" + f"💰 当前奖池: {room.get_total_pot()} 积分\n" + f"⏳ 剩余时间: {room.get_remaining_time()} 秒\n\n" + f"参与玩家:\n{player_list}", + sender) elif room.status == "drawing": await self.bot.send_text_message(target, "🎯 大奖池正在开奖中，请稍候...", sender) elif room.status == "finished" and room.result: @@ -397,7 +415,9 @@ class LuckyPotPlugin(MessagePluginInterface): # 检查是否有活跃房间 if roomid not in self.active_rooms: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "❌ 当前没有活跃的大奖池房间！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "❌ 当前没有活跃的大奖池房间！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "没有活跃房间" @@ -405,13 +425,17 @@ class LuckyPotPlugin(MessagePluginInterface): # 检查是否是创建者 if room.creator != sender: - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "❌ 只有房间创建者才能取消大奖池！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "❌ 只有房间创建者才能取消大奖池！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "不是创建者" # 检查房间状态 if room.status != "waiting": - client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, "❌ 房间已经开始开奖或已结束，无法取消！", sender) + client_msg_id, create_time, new_msg_id = await self.bot.send_text_message(target, + "❌ 房间已经开始开奖或已结束，无法取消！", + sender) self.revoke.add_message_to_revoke(target, client_msg_id, create_time, new_msg_id, 5) return True, "房间状态不允许取消" @@ -429,9 +453,9 @@ class LuckyPotPlugin(MessagePluginInterface): del self.active_rooms[roomid] # 发送取消成功消息 - await self.bot.send_text_message(target, - f"🚫 大奖池已被创建者取消！\n所有玩家的积分已退还。", - sender) + await self.bot.send_text_message(target, + f"🚫 大奖池已被创建者取消！\n所有玩家的积分已退还。", + sender) return True, "取消房间成功" @@ -479,43 +503,43 @@ class LuckyPotPlugin(MessagePluginInterface): try: # 更新房间状态 room.status = "drawing" - + # 发送开奖提示 target = room.group_id if room.group_id else room.creator - await self.bot.send_text_message(target, - f"🧧 大奖池开奖中...\n💥💥💥", - room.creator) - + await self.bot.send_text_message(target, + f"🧧 大奖池开奖中...\n💥💥💥", + room.creator) + # 等待一小段时间，增加悬念 await asyncio.sleep(2) - + # 计算奖池金额 total_pot = room.get_total_pot() - + # 计算抽水金额 house_cut = int(total_pot * (self.house_cut_percent / 100)) remaining_pot = total_pot - house_cut - + # 决定是否有欧皇 has_lucky_winner = random.randint(1, 100) <= self.lucky_bonus_chance lucky_winner = None lucky_bonus = 0 - + if has_lucky_winner and len(room.players) > 1: lucky_winner = random.choice(room.players) lucky_bonus = int(remaining_pot * (self.lucky_bonus_percent / 100)) remaining_pot -= lucky_bonus - + # 随机分配奖池 rewards = self._distribute_prizes(room.players, remaining_pot) - + # 如果有欧皇，增加额外奖励 if lucky_winner: for i, (player, reward) in enumerate(rewards): if player == lucky_winner: rewards[i] = (player, reward + lucky_bonus) break - + # 保存结果 room.result = { "total_pot": total_pot, @@ -524,26 +548,26 @@ class LuckyPotPlugin(MessagePluginInterface): "lucky_winner": lucky_winner, "lucky_bonus": lucky_bonus } - + # 更新玩家积分 for player, reward in rewards: self.points_db.add_points( player, room.group_id, reward, PointSource.GAME, f"大奖池 {room.room_id} 获得奖励" ) - + # 更新房间状态 room.status = "finished" - + # 发送结果 result_text = self._format_result_message(room) await self.bot.send_text_message(target, result_text, room.creator) - + # 延迟一段时间后清理房间 await asyncio.sleep(30) if room.group_id in self.active_rooms and self.active_rooms[room.group_id].room_id == room.room_id: del self.active_rooms[room.group_id] - + except Exception as e: self.LOG.error(f"开奖出错: {e}") # 尝试恢复房间状态 @@ -556,15 +580,15 @@ class LuckyPotPlugin(MessagePluginInterface): rewards = [] remaining = total_points player_count = len(players) - + # 确保每个玩家至少获得1积分 min_points = 1 - + # 如果总积分不足以给每人1分，则平均分配 if total_points < player_count: points_per_player = max(1, total_points // player_count) return [(player, points_per_player) for player in players] - + # 为每个玩家随机分配积分，除了最后一个 for i in range(player_count - 1): # 确保剩余玩家每人至少能分到1积分 @@ -573,13 +597,13 @@ class LuckyPotPlugin(MessagePluginInterface): share = random.randint(min_points, max_points) if max_points > min_points else min_points rewards.append(share) remaining -= share - + # 最后一个玩家获得剩余所有积分 rewards.append(remaining) - + # 随机打乱奖励顺序 random.shuffle(rewards) - + # 将玩家和奖励配对 return list(zip(players, rewards)) @@ -587,36 +611,36 @@ class LuckyPotPlugin(MessagePluginInterface): """格式化结果消息""" if not room.result: return "❌ 没有结果数据" - + result = room.result total_pot = result["total_pot"] house_cut = result["house_cut"] rewards = result["rewards"] lucky_winner = result["lucky_winner"] lucky_bonus = result["lucky_bonus"] - + # 按奖励金额排序 sorted_rewards = sorted(rewards, key=lambda x: x[1], reverse=True) - + message = f"🎯 大奖池开奖结果\n\n" message += f"💰 总奖池: {total_pot} 积分\n" message += f"💸 系统抽水: {house_cut} 积分 ({self.house_cut_percent}%)\n" message += f"🎁 实际分配: {total_pot - house_cut} 积分\n\n" - + # 如果有欧皇，先显示欧皇信息 if lucky_winner: lucky_name = room.player_nicknames.get(lucky_winner, lucky_winner) message += f"🏆 恭喜 {lucky_name} 成为本轮欧皇！额外获得 {lucky_bonus} 积分！\n\n" - + # 显示所有玩家的奖励 message += "🎮 玩家奖励:\n" - + for i, (player, reward) in enumerate(sorted_rewards): player_name = room.player_nicknames.get(player, player) entry_fee = room.entry_fee profit = reward - entry_fee profit_text = f"+{profit}" if profit >= 0 else f"{profit}" - + # 根据排名和盈亏添加不同的表情 if i == 0: emoji = "🥇" @@ -630,17 +654,17 @@ class LuckyPotPlugin(MessagePluginInterface): emoji = "😭" else: emoji = "😐" - + # 为欧皇添加特殊标记 if player == lucky_winner: emoji = "🏆" - + # 为特别惨的玩家添加特殊文案 if profit < -entry_fee * 0.9: # 亏损超过90% message += f"{emoji} {player_name}: {reward} 积分 ({profit_text}) 💔 这也太惨了！\n" else: message += f"{emoji} {player_name}: {reward} 积分 ({profit_text})\n" - + return message def _get_nickname(self, message: Dict[str, Any], user_id: str) -> str: @@ -648,4 +672,4 @@ class LuckyPotPlugin(MessagePluginInterface): all_contacts = message.get("all_contacts", {}) if user_id in all_contacts: return all_contacts[user_id].get("nickname", user_id) - return user_id \ No newline at end of file + return user_id