From 19cde88acf968edf2b32026356d67f9b2504dbd9 Mon Sep 17 00:00:00 2001 From: ninemine <1371605831@qq.com> Date: Fri, 31 Oct 2025 11:07:41 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E7=BB=9F=E4=B8=80=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3BaseGame.db.get=5Fuser=5Fdisplay=5Fname=E8=8E=B7?= =?UTF-8?q?=E5=8F=96=E7=94=A8=E6=88=B7=E5=90=8D=E7=A7=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- core/database.py | 17 +++++++++++++++++ games/gift.py | 10 ++++------ games/idiom.py | 18 ++++++++++++------ games/points.py | 2 +- 4 files changed, 34 insertions(+), 13 deletions(-) diff --git a/core/database.py b/core/database.py index cd9680c..1a2c86c 100644 --- a/core/database.py +++ b/core/database.py @@ -220,6 +220,23 @@ class Database: logger.error(f"更新用户名失败: user_id={user_id}, username={username}, error={e}", exc_info=True) return False + def get_user_display_name(self, user_id: int) -> str: + """获取用户显示名称 + 如果用户已注册（username不为None），返回用户名；否则返回"用户{user_id}" + + Args: + user_id: 用户ID + + Returns: + 用户显示名称（用户名或"用户{user_id}"） + """ + user_dict = self.get_or_create_user(user_id) + username = user_dict.get('username') + if username: + return username + else: + return f"用户{user_id}" + # ===== 游戏状态相关操作 ===== def get_game_state(self, chat_id: int, user_id: int, game_type: str) -> Optional[Dict]: diff --git a/games/gift.py b/games/gift.py index 68a591c..376a906 100644 --- a/games/gift.py +++ b/games/gift.py @@ -82,12 +82,10 @@ class GiftGame(BaseGame): receiver_id = user['user_id'] # 获取接收者名称用于显示 - receiver_user = self.db.get_or_create_user(receiver_id) - receiver_name = receiver_user.get('username', f"用户{receiver_id}") + receiver_name = self.db.get_user_display_name(receiver_id) # 获取发送者名称用于显示 - sender_user = self.db.get_or_create_user(sender_id) - sender_name = sender_user.get('username', f"用户{sender_id}") + sender_name = self.db.get_user_display_name(sender_id) # 验证参数 if points <= 0: @@ -184,7 +182,7 @@ class GiftGame(BaseGame): for record in records: timestamp = datetime.fromtimestamp(record['created_at']).strftime('%m-%d %H:%M') - receiver_name = record.get('receiver_name', f"用户{record['receiver_id']}") + receiver_name = self.db.get_user_display_name(record['receiver_id']) points = record['points'] message = record.get('message', '') @@ -214,7 +212,7 @@ class GiftGame(BaseGame): for record in records: timestamp = datetime.fromtimestamp(record['created_at']).strftime('%m-%d %H:%M') - sender_name = record.get('sender_name', f"用户{record['sender_id']}") + sender_name = self.db.get_user_display_name(record['sender_id']) points = record['points'] message = record.get('message', '') diff --git a/games/idiom.py b/games/idiom.py index 15a0384..6968146 100644 --- a/games/idiom.py +++ b/games/idiom.py @@ -379,10 +379,12 @@ class IdiomGame(BaseGame): text += f"**当前链长**：{state_data['chain_length']}\n\n" user_count = state_data['participants'][str(user_id)] - text += f"@用户{user_id} 成功次数：{user_count}\n\n" + user_display_name = self.db.get_user_display_name(user_id) + text += f"@{user_display_name} 成功次数：{user_count}\n\n" if mentioned_user_id: - text += f"已指定 @用户{mentioned_user_id} 接龙\n\n" + mentioned_display_name = self.db.get_user_display_name(mentioned_user_id) + text += f"已指定 @{mentioned_display_name} 接龙\n\n" else: text += "任何人都可以接龙\n\n" @@ -416,7 +418,8 @@ class IdiomGame(BaseGame): state_data['next_user_id'] = next_user_id self.db.save_game_state(chat_id, 0, 'idiom', state_data) - return f"✅ 已指定 @用户{next_user_id} 接龙" + next_user_display_name = self.db.get_user_display_name(next_user_id) + return f"✅ 已指定 @{next_user_display_name} 接龙" def _reject_idiom(self, chat_id: int, user_id: int, idiom: str) -> str: """裁判拒绝词语 @@ -509,7 +512,8 @@ class IdiomGame(BaseGame): # 下一位 if state_data.get('next_user_id'): - text += f"**下一位**：@用户{state_data['next_user_id']}\n\n" + next_user_display_name = self.db.get_user_display_name(state_data['next_user_id']) + text += f"**下一位**：@{next_user_display_name}\n\n" else: text += f"**下一位**：任何人都可以接龙\n\n" @@ -522,7 +526,8 @@ class IdiomGame(BaseGame): reverse=True ) for idx, (uid, count) in enumerate(sorted_participants[:5], 1): - text += f"{idx}. @用户{uid} - {count}次\n" + user_display_name = self.db.get_user_display_name(int(uid)) + text += f"{idx}. @{user_display_name} - {count}次\n" text += "\n" # 最近成语 @@ -588,7 +593,8 @@ class IdiomGame(BaseGame): reverse=True ) for idx, (uid, count) in enumerate(sorted_participants, 1): - text += f"{idx}. @用户{uid} - {count}次\n" + user_display_name = self.db.get_user_display_name(int(uid)) + text += f"{idx}. @{user_display_name} - {count}次\n" # 更新统计 try: for _ in range(count): diff --git a/games/points.py b/games/points.py index d90a125..dbfa79a 100644 --- a/games/points.py +++ b/games/points.py @@ -126,7 +126,7 @@ class PointsGame(BaseGame): for i, user in enumerate(leaderboard): rank = i + 1 medal = medals[i] if i < len(medals) else "🏅" - username = user.get('username', f"用户{user['user_id']}") + username = self.db.get_user_display_name(user['user_id']) points = user.get('points', 0) text += f"{medal} **第 {rank} 名** {username}\n"