From 78412edb447b690bfe2f25f5ab30b9d2435c3a05 Mon Sep 17 00:00:00 2001 From: ccc_dw <2931589710@qq.com> Date: Thu, 25 Dec 2025 17:53:37 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E6=93=8D=E4=BD=9C=E8=AE=B0?= =?UTF-8?q?=E5=BD=95=E6=95=B0=E6=8D=AE=E5=BA=93?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/bf_bot/user_data/data_utils.py | 38 +++++++++++----------- 1 file changed, 19 insertions(+), 19 deletions(-) diff --git a/src/plugins/bf_bot/user_data/data_utils.py b/src/plugins/bf_bot/user_data/data_utils.py index a07d583..d7dda20 100644 --- a/src/plugins/bf_bot/user_data/data_utils.py +++ b/src/plugins/bf_bot/user_data/data_utils.py @@ -1,5 +1,5 @@ -import sqlite3 import os +import sqlite3 from typing import List, Dict, Any, Optional @@ -100,24 +100,24 @@ class DesignationManager(TableManager): return [dict(row) for row in cursor.fetchall()] -# class QueryRecordManager(TableManager): -# def log_query(self, user_id: str, target_id: str, status: str) -> int: -# """记录查询操作""" -# cursor = self._execute( -# """INSERT INTO query_record -# (user_id, target_id, status) -# VALUES (?, ?, ?)""", -# (user_id, target_id, status) -# ) -# return cursor.lastrowid -# -# def get_user_history(self, user_id: str) -> List[Dict[str, Any]]: -# """获取用户查询历史""" -# cursor = self._execute( -# "SELECT * FROM query_record WHERE user_id = ?", -# (user_id,) -# ) -# return [dict(row) for row in cursor.fetchall()] +class QueryRecordManager(TableManager): + def log_query(self, user_id: str, ea_player_name: str, ea_player_id: str, ea_user_id: str, status: str) -> int: + """记录查询操作""" + cursor = self._execute( + """INSERT INTO query_record + (user_id, ea_player_name, ea_player_id, ea_user_id, status) + VALUES (?, ?, ?, ?, ?)""", + (user_id, ea_player_name, ea_player_id, ea_user_id, status) + ) + return cursor.lastrowid + + def get_user_history(self, user_id: str) -> List[Dict[str, Any]]: + """获取用户查询历史""" + cursor = self._execute( + "SELECT * FROM query_record WHERE user_id = ?", + (user_id,) + ) + return [dict(row) for row in cursor.fetchall()] if __name__ == "__main__":