liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加猜歌名功能，从API中提取歌曲前奏，然后猜歌名

Browse Source
This commit is contained in:
liuwei
2025-06-03 11:48:01 +08:00
parent 2a5c789412
commit 833d0005fa
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
plugins/guess_song/main.py
View File

@@ -36,6 +36,7 @@ class GuessSongRedisDB:
"""保存游戏会话数据""" """保存游戏会话数据"""
try: try:
with self.get_redis_connection() as redis_client: with self.get_redis_connection() as redis_client:
logger.info(f"保存猜歌游戏数据{game_data}")
# 设置过期时间为10分钟 # 设置过期时间为10分钟
redis_client.setex( redis_client.setex(
f'{self.prefix}{room_id}', f'{self.prefix}{room_id}',
@@ -110,7 +111,7 @@ class GuessSongPlugin(MessagePluginInterface):
# 保存上下文对象 # 保存上下文对象
self.event_system = context.get("event_system") self.event_system = context.get("event_system")
self.db_manager = context.get("db_manager") self.db_manager = DBConnectionManager.get_instance()
# 初始化Redis操作类 # 初始化Redis操作类
if self.db_manager: if self.db_manager:
@@ -150,7 +151,7 @@ class GuessSongPlugin(MessagePluginInterface):
"""处理消息""" """处理消息"""
content = str(message.get("content", "")).strip() content = str(message.get("content", "")).strip()
self.LOG.debug(f"插件执行： {self.name}：{content}") self.LOG.debug(f"插件执行： {self.name}：{content}")
# 简化命令处理，只检查开头是否包含命令前缀 # 简化命令处理，只检查开头是否包含命令前缀
command_found = False command_found = False
for cmd in self._commands: for cmd in self._commands:
@@ -158,10 +159,10 @@ class GuessSongPlugin(MessagePluginInterface):
command_found = True command_found = True
content = content[len(cmd):].strip() content = content[len(cmd):].strip()
break break
if not command_found: if not command_found:
return False, "不匹配的命令" return False, "不匹配的命令"
sender = message.get("sender") sender = message.get("sender")
roomid = message.get("roomid", "") roomid = message.get("roomid", "")
gbm: GroupBotManager = message.get("gbm") gbm: GroupBotManager = message.get("gbm")
@@ -182,7 +183,7 @@ class GuessSongPlugin(MessagePluginInterface):
# 简化逻辑：如果有内容，且游戏进行中，则视为答案 # 简化逻辑：如果有内容，且游戏进行中，则视为答案
if current_game and current_game.get("status") == "playing" and content: if current_game and current_game.get("status") == "playing" and content:
return await self._check_answer(bot, session_id, sender, content, current_game) return await self._check_answer(bot, session_id, sender, content, current_game)
# 否则开始新游戏（可以指定歌手或随机） # 否则开始新游戏（可以指定歌手或随机）
return await self._start_new_game(bot, session_id, sender, content if content else None) return await self._start_new_game(bot, session_id, sender, content if content else None)