429 lines
16 KiB
Python
429 lines
16 KiB
Python
import random
|
||
from datetime import datetime
|
||
from typing import Dict, List, Optional
|
||
|
||
from game_task.game_chatgpt_qa import game_question_json, game_answer_json
|
||
from db.connection import DBConnectionManager
|
||
from db.encyclopedia import EncyclopediaDB
|
||
|
||
# 获取数据库连接管理器的单例
|
||
db_manager = DBConnectionManager()
|
||
encyclopedia_db = EncyclopediaDB(db_manager)
|
||
|
||
|
||
# 添加群聊
|
||
def add_group(group_id, player_id):
|
||
try:
|
||
result = encyclopedia_db.add_group(group_id)
|
||
if result:
|
||
message = f"🎉 群 {group_id} 已就位，准备开燥！"
|
||
else:
|
||
message = f"🌟 群 {group_id} 早就蓄势待发啦！"
|
||
return {"message": message, "player_id": player_id}
|
||
except Exception as e:
|
||
print(f"添加群聊出错: {e}")
|
||
message = f"🌟 群 {group_id} 早就蓄势待发啦！"
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
|
||
# 获取所有群聊ID
|
||
def get_group_ids():
|
||
try:
|
||
return encyclopedia_db.get_all_groups()
|
||
except Exception as e:
|
||
print(f"获取群聊ID出错: {e}")
|
||
return []
|
||
|
||
|
||
# 确保游戏启动（自动初始化群聊和玩家）
|
||
def ensure_game_started(group_id, player_id, player_name="未知玩家"):
|
||
try:
|
||
# 检查并添加群聊
|
||
if not encyclopedia_db.check_group_exists(group_id):
|
||
add_group(group_id, player_id)
|
||
|
||
# 检查并添加玩家
|
||
player = encyclopedia_db.get_player(player_id, group_id)
|
||
if not player:
|
||
encyclopedia_db.add_player(player_id, group_id, player_name)
|
||
message = (
|
||
f"🎉 哇塞，{player_name} 你来啦！\n"
|
||
f"🌟 群 {group_id} 瞬间燃爆！\n"
|
||
f"🎈 快来接任务，秀翻全场！指令： /t"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
return {"message": None, "player_id": player_id}
|
||
except Exception as e:
|
||
print(f"确保游戏启动出错: {e}")
|
||
return {"message": None, "player_id": player_id}
|
||
|
||
|
||
# 随机分配任务
|
||
def assign_random_task(group_id, player_id=None):
|
||
try:
|
||
# 获取群内所有玩家
|
||
players = encyclopedia_db.get_all_players_in_group(group_id)
|
||
|
||
if not players:
|
||
message = (
|
||
f"😔 哎呀，群 {group_id} 静悄悄\n"
|
||
f"🌟 快拉小伙伴来嗨吧！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
if player_id:
|
||
player_dict = {p['player_id']: p['player_name'] for p in players}
|
||
if player_id not in player_dict:
|
||
message = (
|
||
f"😅 嘿，{player_id} 你谁啊？\n"
|
||
f"🌟 先用 /s 报名，不然没法玩哦！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
holder_id = player_id
|
||
holder_name = player_dict[player_id]
|
||
else:
|
||
holder = random.choice(players)
|
||
holder_id = holder['player_id']
|
||
holder_name = holder['player_name']
|
||
|
||
task = game_question_json("请出题！")
|
||
category = task["category"]
|
||
question = task["question"]
|
||
answer = task["answer"]
|
||
score = int(task["score"])
|
||
description = task.get("description", "")
|
||
|
||
# 创建活跃任务
|
||
active_task_id = encyclopedia_db.create_active_task(
|
||
group_id, question, answer, score, description, holder_id
|
||
)
|
||
|
||
if not active_task_id:
|
||
message = f"😔 任务创建失败，请稍后再试！"
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
if player_id:
|
||
message = (
|
||
f"🎁 {holder_name}，你的专属任务闪亮登场！\n"
|
||
f"🎀 任务ID: {active_task_id}\n"
|
||
f"🎈 问题：[{category}]{question}\n"
|
||
f"🌼 积分：{score}\n"
|
||
f"🌈 快上答案：/a {active_task_id} 答案"
|
||
)
|
||
else:
|
||
message = (
|
||
f"🎁 新任务来袭，够不够刺激？\n"
|
||
f"🎀 任务ID: {active_task_id}\n"
|
||
f"🌟 幸运鹅：{holder_name}\n"
|
||
f"🎈 问题：[{category}]{question}\n"
|
||
f"🌼 积分：{score}\n"
|
||
f"🌈 抢答格式：/a {active_task_id} 答案"
|
||
)
|
||
return {"message": message, "player_id": holder_id}
|
||
except Exception as e:
|
||
print(f"分配任务出错: {e}")
|
||
message = f"😔 任务分配出错，请稍后再试！"
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
|
||
# 提交答案并计分
|
||
def submit_answer(group_id, player_id, task_id, answer):
|
||
try:
|
||
# 获取玩家信息
|
||
player = encyclopedia_db.get_player(player_id, group_id)
|
||
if not player:
|
||
message = (
|
||
f"😅 嘿，{player_id} 你是路人甲吗？\n"
|
||
f"🌟 用 /s 先加入群 {group_id} 吧！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
player_name = player['player_name']
|
||
|
||
if not task_id.isdigit():
|
||
message = (
|
||
f"😅 喂，任务ID得是数字好吗？\n"
|
||
f"🌟 比如：1\n"
|
||
f"🎈 别瞎搞，重新来！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
active_task_id = int(task_id)
|
||
|
||
# 获取任务信息
|
||
task_data = encyclopedia_db.get_task_by_id(group_id, active_task_id)
|
||
|
||
if not task_data:
|
||
message = (
|
||
f"😔 哎哟，任务 task_{active_task_id} 不翼而飞啦！\n"
|
||
f"🌼 可能被别人抢先一步咯！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
if task_data['status'] == 'completed':
|
||
message = (
|
||
f"😄 哈哈，你慢了一步！\n"
|
||
f"🌟 任务 task_{active_task_id} 已经完结\n"
|
||
f"🎈 快去抢新任务吧！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
question = task_data['question']
|
||
correct_answer_db = task_data['answer'].lower()
|
||
top_score = task_data['score']
|
||
holder_id = task_data['holder_id']
|
||
|
||
# 获取任务持有者信息
|
||
holder = encyclopedia_db.get_task_holder(group_id, holder_id)
|
||
holder_name = holder['player_name'] if holder else "未知玩家"
|
||
|
||
answer_json = {"question": question, "top_score": str(top_score), "answer": answer}
|
||
result = game_answer_json(answer_json)
|
||
points = int(result["score"])
|
||
description = result["description"]
|
||
is_correct = points > 0
|
||
|
||
if is_correct:
|
||
# 更新玩家积分
|
||
encyclopedia_db.update_player_points(player_id, group_id, points)
|
||
|
||
# 完成任务
|
||
encyclopedia_db.complete_task(active_task_id)
|
||
|
||
if player_id == holder_id:
|
||
message = (
|
||
f"🎉 {player_name} 你是天才吗？\n"
|
||
f"🌟 任务：{question}\n"
|
||
f"🎈 答对啦，简直无敌！\n"
|
||
f"🌈 奖励：{points} 分\n"
|
||
f"🎀 彩蛋：{description}"
|
||
)
|
||
else:
|
||
message = (
|
||
f"🎉 {player_name} 抢答王上线！\n"
|
||
f"🌟 任务：{question}\n"
|
||
f"🎈 原主：{holder_name} 被你截胡啦！\n"
|
||
f"🌈 狂揽 {points} 分，太骚了！\n"
|
||
f"🎀 彩蛋：{description}"
|
||
)
|
||
else:
|
||
# 扣除积分
|
||
encyclopedia_db.update_player_points(player_id, group_id, -1)
|
||
points = -1
|
||
message = (
|
||
f"😅 {player_name} 你这是要笑死我吗？\n"
|
||
f"🌼 任务：{question}\n"
|
||
f"🎈 你答：{answer}\n"
|
||
f"🌟 正确答案：{correct_answer_db}\n"
|
||
f"🌈 扣 1 分，别哭哦！\n"
|
||
f"🎀 提示：{description}\n"
|
||
f"🌟 任务ID: {active_task_id} 还能抢救一下！"
|
||
)
|
||
|
||
# 添加任务历史记录
|
||
encyclopedia_db.add_task_history(
|
||
group_id, active_task_id, player_id, answer, is_correct, points
|
||
)
|
||
|
||
return {"message": message, "player_id": player_id}
|
||
except ValueError as e:
|
||
print(f"submit_answer:{e}")
|
||
message = (
|
||
f"😅 喂，任务ID得是数字懂吗？\n"
|
||
f"🌟 比如：1\n"
|
||
f"🎈 重来，别皮！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
except Exception as e:
|
||
print(f"submit_answer:{e}")
|
||
message = (
|
||
f"😅 出错了，任务ID得是数字哦！\n"
|
||
f"🌟 比如：1\n"
|
||
f"🎈 再试一次吧！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
|
||
# 显示排行榜
|
||
def show_rank(group_id, player_id):
|
||
try:
|
||
# 获取排行榜
|
||
ranks = encyclopedia_db.get_player_ranking(group_id, 10)
|
||
|
||
if not ranks:
|
||
message = (
|
||
f"😔 群 {group_id} 冷冷清清\n"
|
||
f"🌟 快来一起燥起来吧！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
rank_text = f"🎉 群 {group_id} 排行榜（Top 10）来啦！\n"
|
||
for i, row in enumerate(ranks, 1):
|
||
rank_text += f"🐓 {i}. {row['player_name']}: {row['points']} 分\n"
|
||
return {"message": rank_text, "player_id": player_id}
|
||
except Exception as e:
|
||
print(f"显示排行榜出错: {e}")
|
||
message = f"😔 获取排行榜出错，请稍后再试！"
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
|
||
# 显示当前活跃任务
|
||
def show_active_tasks(group_id, player_id):
|
||
try:
|
||
# 获取活跃任务
|
||
tasks = encyclopedia_db.get_active_tasks_in_group(group_id)
|
||
|
||
if not tasks:
|
||
message = (
|
||
f"😄 群 {group_id} 现在一片祥和\n"
|
||
f"🌟 没任务？快用 /t 搞一个！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
task_text = f"🎉 群 {group_id} 活跃任务速递：\n"
|
||
for task in tasks:
|
||
task_text += (
|
||
f"🌈 任务ID: task_{task['active_task_id']}\n"
|
||
f"🎀 问题：{task['question']}\n"
|
||
f"🌼 大佬：{task['player_name']}\n"
|
||
)
|
||
return {"message": task_text, "player_id": player_id}
|
||
except Exception as e:
|
||
print(f"显示活跃任务出错: {e}")
|
||
message = f"😔 获取活跃任务出错，请稍后再试！"
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
|
||
# 列举所有未完成任务
|
||
def list_uncompleted_tasks(group_id, player_id):
|
||
try:
|
||
# 获取未完成任务
|
||
tasks = encyclopedia_db.get_active_tasks_in_group(group_id)
|
||
|
||
if not tasks:
|
||
message = (
|
||
f"😄 群 {group_id} 全员开挂？\n"
|
||
f"🌟 没未完成任务，快用 /t 再战！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
task_text = f"🎉 群 {group_id} 未完成任务大曝光：\n"
|
||
for task in tasks:
|
||
task_text += (
|
||
f"🌈 任务ID: task_{task['active_task_id']}\n"
|
||
f"🎀 问题：{task['question']}\n"
|
||
f"🌼 主人：{task['player_name']}\n"
|
||
)
|
||
return {"message": task_text, "player_id": player_id}
|
||
except Exception as e:
|
||
print(f"列举未完成任务出错: {e}")
|
||
message = f"😔 获取未完成任务出错，请稍后再试！"
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
|
||
# 定时任务：整点触发，排除23:00-08:00
|
||
def run_random_task_assignment(group_id):
|
||
current_hour = datetime.now().hour
|
||
if current_hour >= 23 or current_hour < 9:
|
||
message = (
|
||
f"😴 现在 {current_hour}:00，小伙伴们都睡啦\n"
|
||
f"🌙 9点到22点再来嗨吧！"
|
||
)
|
||
print(f"{datetime.now()} 群 {group_id} 当前时间 {current_hour}:00 在23:00-08:00区间，跳过任务发放")
|
||
return {"message": message, "player_id": None}
|
||
result = assign_random_task(group_id)
|
||
print(f"{datetime.now()} {result['message']}")
|
||
return result
|
||
|
||
|
||
# 处理群聊消息
|
||
def game_process_message(group_id, player_id, message, player_name="未知玩家"):
|
||
init_result = ensure_game_started(group_id, player_id, player_name)
|
||
init_message = init_result["message"]
|
||
player_id = init_result["player_id"]
|
||
|
||
if message == "/s": # 可选：显式加入游戏
|
||
if init_message:
|
||
return {"message": init_message, "player_id": player_id}
|
||
message = (
|
||
f"😄 嘿，{player_name} 你来啦！\n"
|
||
f"🌼 已加入群 {group_id} 的狂欢\n"
|
||
f"🎶 快用 /t 开干吧！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
elif message == "/ts": # 查看活跃任务
|
||
if init_message:
|
||
message = init_message + "\n🌟 用 /ts 看看当前任务吧！"
|
||
return {"message": message, "player_id": player_id}
|
||
return show_active_tasks(group_id, player_id)
|
||
elif message == "/l": # 列出未完成任务
|
||
if init_message:
|
||
message = init_message + "\n🌟 用 /l 查未完成任务哦！"
|
||
return {"message": message, "player_id": player_id}
|
||
return list_uncompleted_tasks(group_id, player_id)
|
||
elif message.startswith("/a"): # 提交答案
|
||
if init_message:
|
||
message = init_message + "\n🌟 用 /a [任务ID] [答案] 放大招吧！"
|
||
return {"message": message, "player_id": player_id}
|
||
parts = message.split(" ", 2)
|
||
if len(parts) < 3:
|
||
message = (
|
||
f"😅 喂，格式不对啊！\n"
|
||
f"🌟 正确姿势：/a [任务ID] [答案]\n"
|
||
f"🎈 比如：/a 1 钒"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
task_id, answer = parts[1], parts[2]
|
||
return submit_answer(group_id, player_id, task_id, answer)
|
||
elif message == "/r": # 查看排行榜
|
||
if init_message:
|
||
message = init_message + "\n🌟 用 /r 看看谁是大佬！"
|
||
return {"message": message, "player_id": player_id}
|
||
return show_rank(group_id, player_id)
|
||
elif message == "/ag": # 可选：手动添加群聊
|
||
if init_message:
|
||
return {"message": init_message, "player_id": player_id}
|
||
return add_group(group_id, player_id)
|
||
elif message == "/t": # 获取任务
|
||
if init_message:
|
||
message = init_message + "\n🌟 已为你准备好任务，快接招！"
|
||
return {"message": message, "player_id": player_id}
|
||
current_hour = datetime.now().hour
|
||
if current_hour >= 23 or current_hour < 9:
|
||
message = (
|
||
f"😴 现在 {current_hour}:00\n"
|
||
f"🌙 太晚啦，大家都睡了！\n"
|
||
f"🌞 9点到22点再来嗨吧！"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
return assign_random_task(group_id, player_id)
|
||
else:
|
||
message = (
|
||
f"😄 嘿，二货！\n"
|
||
f"🌟 指令玩错啦！\n"
|
||
f"🎈 快试试这些：\n"
|
||
f"🌈 /s - 加入狂欢（可选）\n"
|
||
f"🎀 /ts - 看活跃任务\n"
|
||
f"🌼 /l - 未完成任务\n"
|
||
f"🌟 /a [任务ID] [答案] - 提交答案\n"
|
||
f"🎈 /ag - 加群（可选）\n"
|
||
f"🌈 /t - 抢任务\n"
|
||
f"🎀 /r - 谁是大佬"
|
||
)
|
||
return {"message": message, "player_id": player_id}
|
||
|
||
|
||
# 设置定时任务
|
||
def setup_schedule():
|
||
group_ids = get_group_ids()
|
||
for gid in group_ids:
|
||
run_random_task_assignment(group_id=gid)
|
||
|
||
|
||
# 主程序
|
||
if __name__ == "__main__":
|
||
# 测试用例
|
||
print(game_process_message("45317011307@chatroom", "Jyunere", "/t")) # 新用户获取任务
|
||
print(game_process_message("45317011307@chatroom", "Jyunere", "/a 18 罗马斗兽场")) # 提交答案
|
||
print(game_process_message("45317011307@chatroom", "Jyunere", "/r")) # 查看排行榜
|