From 08e71722e8e401367d5cc22ee0d87f799159e4da Mon Sep 17 00:00:00 2001 From: liuwei Date: Fri, 21 Feb 2025 16:32:51 +0800 Subject: [PATCH] =?UTF-8?q?feature=EF=BC=9A=E6=96=B0=E5=A2=9E=E7=BE=A4?= =?UTF-8?q?=E7=99=BE=E7=A7=91=E6=B8=B8=E6=88=8F=EF=BC=8C=E7=94=A8=E4=BA=8E?= =?UTF-8?q?=E4=BF=83=E8=BF=9B=E7=BE=A4=E6=B4=BB=E8=B7=83=E3=80=82=E5=8A=A0?= =?UTF-8?q?=E5=85=A5=E4=BA=86=E6=96=B0=E7=9A=84=E6=8C=87=E4=BB=A4=20=20?= =?UTF-8?q?=E5=8F=AF=E7=94=A8=EF=BC=9A/start,=20/tasks,=20/list,=20/answer?= =?UTF-8?q?=20[=E4=BB=BB=E5=8A=A1ID]=20[=E7=AD=94=E6=A1=88],=20/addgroup,?= =?UTF-8?q?=20/rank?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- game_task/game_chatgpt_qa.py | 67 ++++++ game_task/game_task_encyclopedia.py | 321 ++++++++++++++++++++++++++++ main.py | 3 + robot.py | 11 +- task/message_task_json.py | 7 +- 5 files changed, 404 insertions(+), 5 deletions(-) create mode 100644 game_task/game_chatgpt_qa.py create mode 100644 game_task/game_task_encyclopedia.py diff --git a/game_task/game_chatgpt_qa.py b/game_task/game_chatgpt_qa.py new file mode 100644 index 0000000..3505f4f --- /dev/null +++ b/game_task/game_chatgpt_qa.py @@ -0,0 +1,67 @@ +import requests +import json + + +# 解析JSON +def extract_content(data_string): + try: + data = json.loads(data_string) + # 提取content字段 + content = data["choices"][0]["message"].get("content", "") + return content + except json.JSONDecodeError: + print("Invalid JSON") + return None + + +def message_task_json(prompt, content): + # 设置Authorization和URL + authorization = "46a5674a-e978-491b-a810-5d54605f2c36" # 请替换为真实的Authorization token + url = 'http://127.0.0.1:8080/v1/chat/completions' + + data = { + # "stream": True, + "model": "windsurf/gpt4o", + "messages": [ + { + "role": "system", + "content": f"{prompt}" + }, + { + "role": "user", + "content": f"{content}" + } + + ] + } + + # 设置请求头 + headers = { + "Content-Type": "application/json; charset=utf-8", + "Authorization": authorization + } + + # 发送POST请求 + response = requests.post(url, headers=headers, data=json.dumps(data), ) + response.encoding = 'utf-8' + + # 输出响应内容 + print(response.status_code) + print(response.text) + return extract_content(response.text) + + +def game_question_json(question): + prompt = "你是一个益智百科问答大师，可以随时提出百科类的问题，问题需要有一定的难度，回答完毕之后用户能有所收获，并且对问题进行打分，同时根据问题难度告知答对之后给多少分（1-10）请只返回JSON格式的内容：格式要求如下：{\"question\": \"哪个国家最早将玫瑰与爱情联系起来？\", \"score\":\"1\", \"answer\": \"波斯\",\"description\":\"描述问题答案的原因\"}" + return message_task_json(prompt, question) + + +def game_answer_json(answar): + prompt = "你是一个益智百科问答大师，可以根据用户回答的答案进行判断，并且对问题(question)答案(answer)进行打分,打分时请参考最高分要求(top_score)，告知用户能获得多少分，请在description中描述打分理由，请只返回JSON格式的内容：格式要求如下：{\"question\": \"哪个国家最早将玫瑰与爱情联系起来？\", \"score\":\"1\", \"answer\": \"波斯\",\"description\":\"描述问题答案的原因\"}" + return message_task_json(prompt, answar) + + +if __name__ == '__main__': + + print(game_question_json('请出题！')) + print(game_answer_json('question:哪个国家的节日与裸体狂欢有关？,answer:古罗马,top_score:3')) diff --git a/game_task/game_task_encyclopedia.py b/game_task/game_task_encyclopedia.py new file mode 100644 index 0000000..e71d335 --- /dev/null +++ b/game_task/game_task_encyclopedia.py @@ -0,0 +1,321 @@ +import random +import schedule +import time +from datetime import datetime +import pymysql + +from game_task.game_chatgpt_qa import game_question_json, game_answer_json + +# 数据库连接配置 +db_config = { + 'host': '192.168.2.32', # 替换为你的MariaDB服务器地址 + 'user': 'root', # 替换为你的MariaDB用户名 + 'password': 'lw123456', # 替换为你的MariaDB密码 + 'database': 'message_archive', + 'charset': 'utf8mb4', + 'cursorclass': pymysql.cursors.DictCursor +} + + +# 连接数据库 +def get_db_connection(): + return pymysql.connect(**db_config) + + +# 添加群聊 +def add_group(group_id): + conn = get_db_connection() + cursor = conn.cursor() + try: + cursor.execute( + "INSERT INTO t_encyclopedia_groups (group_id) VALUES (%s)", + (group_id,) + ) + conn.commit() + return f"群 {group_id} 已成功添加！" + except pymysql.err.IntegrityError: + return f"群 {group_id} 已存在！" + finally: + cursor.close() + conn.close() + + +# 获取所有群聊ID +def get_group_ids(): + conn = get_db_connection() + cursor = conn.cursor() + try: + cursor.execute("SELECT group_id FROM t_encyclopedia_groups") + return [row['group_id'] for row in cursor.fetchall()] + finally: + cursor.close() + conn.close() + + +# 开始游戏（使用 player_id） +def start_game(group_id, player_id, player_name="未知玩家"): + conn = get_db_connection() + cursor = conn.cursor() + try: + # 确保群聊存在 + cursor.execute("SELECT group_id FROM t_encyclopedia_groups WHERE group_id = %s", (group_id,)) + if not cursor.fetchone(): + return f"群 {group_id} 未注册，请先添加群聊！" + + cursor.execute( + "INSERT INTO t_encyclopedia_players (player_id, group_id, player_name) VALUES (%s, %s, %s)", + (player_id, group_id, player_name) + ) + conn.commit() + return f"欢迎加入百科全书挑战！{player_name} (ID: {player_id}) 已加入群 {group_id}，任务将随机分配，任何人可抢答。" + except pymysql.err.IntegrityError: + cursor.execute( + "SELECT player_name FROM t_encyclopedia_players WHERE group_id = %s AND player_id = %s", + (group_id, player_id) + ) + existing_name = cursor.fetchone()['player_name'] + return f"你已在群 {group_id} 中加入游戏！玩家ID: {player_id}，名称: {existing_name}" + finally: + cursor.close() + conn.close() + + +# 随机分配任务 +def assign_random_task(group_id): + conn = get_db_connection() + cursor = conn.cursor() + try: + # 获取当前群聊的玩家列表 + cursor.execute("SELECT player_id FROM t_encyclopedia_players WHERE group_id = %s", (group_id,)) + players = [row['player_id'] for row in cursor.fetchall()] + if not players: + return f"群 {group_id} 暂无玩家参与，无法分配任务！" + + # 使用内部方法获取任务 + task = game_question_json("请出题！") + question = task["question"] + answer = task["answer"] + score = int(task["score"]) + description = task.get("description", "") + + # 分配给随机玩家 + holder_id = random.choice(players) + cursor.execute( + "INSERT INTO t_encyclopedia_active_tasks (group_id, question, answer, score, description, holder_id) VALUES (%s, %s, %s, %s, %s, %s)", + (group_id, question, answer, score, description, holder_id) + ) + conn.commit() + + # 获取分配的任务ID和持有者名称 + cursor.execute( + "SELECT active_task_id FROM t_encyclopedia_active_tasks WHERE group_id = %s AND question = %s AND holder_id = %s ORDER BY assigned_at DESC LIMIT 1", + (group_id, question, holder_id) + ) + active_task_id = cursor.fetchone()['active_task_id'] + cursor.execute( + "SELECT player_name FROM t_encyclopedia_players WHERE group_id = %s AND player_id = %s", + (group_id, holder_id) + ) + holder_name = cursor.fetchone()['player_name'] + + return f"任务分配给 {holder_name} (ID: {holder_id})：{question}（群 {group_id} 所有人可抢答！任务ID: task_{active_task_id}，积分：{score}）" + finally: + cursor.close() + conn.close() + + +# 提交答案并计分 +def submit_answer(group_id, player_id, task_id, answer): + conn = get_db_connection() + cursor = conn.cursor() + try: + # 检查玩家是否存在并获取名称 + cursor.execute("SELECT player_name FROM t_encyclopedia_players WHERE group_id = %s AND player_id = %s", + (group_id, player_id)) + player_row = cursor.fetchone() + if not player_row: + return f"请先在群 {group_id} 输入 /start 加入游戏！玩家ID: {player_id}" + player_name = player_row['player_name'] + + # 检查任务是否存在并获取任务详情 + active_task_id = int(task_id.split('_')[1]) + cursor.execute( + "SELECT question, answer, score, holder_id, status FROM t_encyclopedia_active_tasks WHERE group_id = %s AND active_task_id = %s", + (group_id, active_task_id) + ) + task_data = cursor.fetchone() + if not task_data or task_data['status'] == 'completed': + return f"群 {group_id} 中此任务不存在或已完成！" + + question, correct_answer_db, top_score, holder_id, _ = task_data['question'], task_data['answer'].lower(), \ + task_data['score'], task_data['holder_id'], task_data['status'] + + # 获取持有者名称 + cursor.execute("SELECT player_name FROM t_encyclopedia_players WHERE group_id = %s AND player_id = %s", + (group_id, holder_id)) + holder_name = cursor.fetchone()['player_name'] + + # 调用内部方法校验答案 + answer_json = {"question": question, "top_score": str(top_score), "answer": answer} + result = game_answer_json(answer_json) + user_answer = answer.strip().lower() + points = int(result["score"]) + correct_answer = result["answer"].lower() + description = result["description"] + is_correct = points > 0 + + # 记录历史 + cursor.execute( + "INSERT INTO t_encyclopedia_task_history (group_id, active_task_id, player_id, answer, is_correct, points_earned) VALUES (%s, %s, %s, %s, %s, %s)", + (group_id, active_task_id, player_id, answer, is_correct, points) + ) + + if is_correct: + # 更新玩家积分 + cursor.execute( + "UPDATE t_encyclopedia_players SET points = points + %s WHERE group_id = %s AND player_id = %s", + (points, group_id, player_id) + ) + # 标记任务完成 + cursor.execute( + "UPDATE t_encyclopedia_active_tasks SET status = 'completed' WHERE group_id = %s AND active_task_id = %s", + (group_id, active_task_id) + ) + conn.commit() + + if player_id == holder_id: + return f"{player_name} (ID: {player_id}) 回答正确！任务：{question}\n获得 {points} 分\n描述：{description}" + else: + return f"{player_name} (ID: {player_id}) 抢答成功！任务：{question}（原持有者：{holder_name} (ID: {holder_id})）\n获得 {points} 分\n描述：{description}" + else: + conn.commit() + return f"{player_name} (ID: {player_id}) 回答错误！任务：{question}\n你的答案：{answer}，正确答案：{correct_answer}\n描述：{description}" + finally: + cursor.close() + conn.close() + + +# 显示排行榜 +def show_rank(group_id): + conn = get_db_connection() + cursor = conn.cursor() + try: + cursor.execute( + "SELECT player_name, points FROM t_encyclopedia_players WHERE group_id = %s ORDER BY points DESC LIMIT 10", + (group_id,) + ) + ranks = cursor.fetchall() + if not ranks: + return f"群 {group_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 rank_text + finally: + cursor.close() + conn.close() + + +# 显示当前活跃任务 +def show_active_tasks(group_id): + conn = get_db_connection() + cursor = conn.cursor() + try: + cursor.execute(""" + SELECT a.active_task_id, a.question, p.player_name, p.player_id + FROM t_encyclopedia_active_tasks a + JOIN t_encyclopedia_players p ON a.holder_id = p.player_id AND a.group_id = p.group_id + WHERE a.group_id = %s AND a.status = 'pending' + """, (group_id,)) + tasks = cursor.fetchall() + if not tasks: + return f"群 {group_id} 当前没有活跃任务！" + task_text = f"群 {group_id} 当前活跃任务：\n" + for task in tasks: + task_text += f"任务ID: task_{task['active_task_id']} - {task['question']}（持有者：{task['player_name']} (ID: {task['player_id']})）\n" + return task_text + finally: + cursor.close() + conn.close() + + +# 列举所有未完成任务及其所属者 +def list_uncompleted_tasks(group_id): + conn = get_db_connection() + cursor = conn.cursor() + try: + cursor.execute(""" + SELECT a.active_task_id, a.question, p.player_name, p.player_id + FROM t_encyclopedia_active_tasks a + JOIN t_encyclopedia_players p ON a.holder_id = p.player_id AND a.group_id = p.group_id + WHERE a.group_id = %s AND a.status = 'pending' + """, (group_id,)) + tasks = cursor.fetchall() + if not tasks: + return f"群 {group_id} 当前没有未完成的任务！" + task_text = f"群 {group_id} 所有未完成任务列表：\n" + for task in tasks: + task_text += f"任务ID: task_{task['active_task_id']} - {task['question']}（所属：{task['player_name']} (ID: {task['player_id']})）\n" + return task_text + finally: + cursor.close() + conn.close() + + +# 定时任务：整点触发，排除23:00-08:00 +def run_random_task_assignment(group_id): + current_hour = datetime.now().hour + if current_hour >= 23 or current_hour < 9: # 排除23:00-08:00 + print(f"{datetime.now()} 群 {group_id} 当前时间 {current_hour}:00 在23:00-08:00区间，跳过任务发放") + return + result = assign_random_task(group_id) + print(f"{datetime.now()} {result}") + + +# 处理群聊消息 +def process_message(group_id, player_id, message, player_name="未知玩家"): + if message == "/start": + return start_game(group_id, player_id, player_name) + elif message == "/tasks": + return show_active_tasks(group_id) + elif message == "/list": + return list_uncompleted_tasks(group_id) + elif message.startswith("/answer"): + parts = message.split(" ", 2) + if len(parts) < 3: + return "请使用格式：/answer [任务ID] [答案]，如 /answer task_1 钒（Vanadium）" + task_id, answer = parts[1], parts[2] + return submit_answer(group_id, player_id, task_id, answer) + elif message == "/rank": + return show_rank(group_id) + elif message.startswith("/addgroup"): + return add_group(group_id) + else: + return "无效命令！可用：/start, /tasks, /list, /answer [任务ID] [答案], /addgroup, /rank" + + +# 设置定时任务 +def setup_schedule(): + group_ids = get_group_ids() + for gid in group_ids: + schedule.every().hour.at(":00").do(run_random_task_assignment, group_id=gid) + + +# 主程序 +if __name__ == "__main__": + # 初始化群聊 + print(add_group(1)) + print(add_group(2)) + + # 初始化玩家 + print(process_message(1, 1001, "/start", "玩家1")) + print(process_message(1, 1002, "/start", "玩家2")) + print(process_message(2, 2001, "/start", "玩家A")) + print(process_message(2, 2002, "/start", "玩家B")) + + # 设置调度 + setup_schedule() + + while True: + schedule.run_pending() + time.sleep(1) \ No newline at end of file diff --git a/main.py b/main.py index 6ccb0c6..e891662 100644 --- a/main.py +++ b/main.py @@ -6,6 +6,7 @@ from argparse import ArgumentParser from configuration import Config from constants import ChatType +from game_task.game_task_encyclopedia import setup_schedule from robot import Robot, __version__ from wcferry import Wcf @@ -43,6 +44,8 @@ def main(chat_type: int): # sehuatang robot.onEveryTime("15:00", robot.generateSehuatangPdf) + # 游戏的定时任务一直执行 + setup_schedule() # 让机器人一直跑 robot.keepRunningAndBlockProcess() diff --git a/robot.py b/robot.py index 8773eda..be736a2 100644 --- a/robot.py +++ b/robot.py @@ -1,5 +1,5 @@ # -*- coding: utf-8 -*- - +import json import logging import re import time @@ -24,6 +24,7 @@ from base.func_xinghuo_web import XinghuoWeb from base.func_claude import Claude from configuration import Config from constants import ChatType +from game_task.game_task_encyclopedia import game_process_message from group_auto.group_auto_invite import get_first_group_id, process_command from group_auto.group_member_change import GroupMemberChange from robot_cmd.robot_command import GroupBotManager @@ -147,8 +148,12 @@ class Robot(Job): match = re.match(pattern, q) # 所有人员都可以要求他撤回刚刚的信息 - if q == '撤回': - self.revoke_messages(msg.roomid) + if msg.from_group() and q.startswith("/"): + # 因为内容中存在空格指令，所以不能使用q + game_message = re.sub(r"@.*?[\u2005|\s]", "", msg.content) + resp = game_process_message(group_id=msg.roomid, player_id=msg.sender, message=game_message, + player_name=self.allContacts.get(msg.sender, msg.sender)) + self.sendTextMsg(resp.get("message"), msg.roomid, msg.sender) return True if q == "#今日百度新闻": self.newsBaiduReport((msg.roomid if msg.from_group() else msg.sender)) diff --git a/task/message_task_json.py b/task/message_task_json.py index e13eea9..751b222 100644 --- a/task/message_task_json.py +++ b/task/message_task_json.py @@ -19,8 +19,10 @@ def message_task_json(content): authorization = "46a5674a-e978-491b-a810-5d54605f2c36" # 请替换为真实的Authorization token url = 'http://127.0.0.1:8080/v1/chat/completions' - prompt = ('你是一个个人助理，请根据用户给的任务要求及时安排日程。需要你理解用户的自然语言，并即时反馈在什么时候加入了日程提醒。提醒时间为什么时候。请只返回json格式的内容用于提取数据。格式如下：{\"task\": \"洗澡\", \"reminder_time\":\"00 20 * * *\", \"reason\": \"用户今天晚上八点要洗澡\"}') + prompt = "你是一个益智百科问答大师，可以随时提出百科类的问题，问题需要有一定的难度，回答完毕之后用户能有所收获，并且对问题进行打分，同时根据问题难度告知答对之后给多少分（1-10）请只返回JSON格式的内容：格式要求如下：{\"question\": \"哪个国家最早将玫瑰与爱情联系起来？\", \"score\":\"1\", \"answer\": \"波斯\",\"description\":\"描述问题答案的原因\"}" # 设置请求的payload + # prompt ="你是一个益智百科问答大师，可以根据用户回答的答案进行判断，并且对问题答案进行打分,打分时请参考最高分要求，告知用户能获得多少分，请只返回JSON格式的内容：格式要求如下：{\"question\": \"哪个国家最早将玫瑰与爱情联系起来？\", \"score\":\"1\", \"answer\": \"波斯\",\"description\":\"描述问题答案的原因\"}" + data = { # "stream": True, "model": "windsurf/gpt4o", @@ -54,4 +56,5 @@ def message_task_json(content): if __name__ == '__main__': - print(message_task_json('提醒我明天晚上9点20要去游泳')) \ No newline at end of file + # print(message_task_json('question:哪个国家的节日与裸体狂欢有关？,answer:古罗马,top_score:3')) + print(message_task_json('请出题！')) \ No newline at end of file