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__":