diff --git a/database.db b/database.db index fefb04e..65080f2 100644 Binary files a/database.db and b/database.db differ diff --git a/get_mouse_tracker.py b/get_mouse_tracker.py index 8958ba4..b3837ae 100644 --- a/get_mouse_tracker.py +++ b/get_mouse_tracker.py @@ -1,17 +1,45 @@ import sqlite3 import json +import random -conn = sqlite3.connect('database.db') +# 根据distance查询数据 +def get_data_by_distance(distance, one=True): + conn = sqlite3.connect('database.db') + cursor = conn.cursor() + cursor.execute(f"SELECT * FROM Tracker WHERE distance = {distance}") + rows = cursor.fetchall() -cursor = conn.cursor() -cursor.execute("SELECT * FROM Tracker") + if one: + random_index = random.randint(0, len(rows) - 1) + return json.loads(rows[random_index][2]) + return [json.loads(row[2]) for row in rows] -rows = cursor.fetchall() +mouse_data = get_data_by_distance(100) +for data in mouse_data: + print(data) -for index, distance, mouse_tracker in rows: - print(f"Index: {index}, Distance: {distance}") - print(f"Mouse tracker: {json.loads(mouse_tracker)}") +# cursor.execute("SELECT * FROM Tracker") -# 关闭连接 -conn.close() +# rows = cursor.fetchall() + +# index_list = [] + +# for index, distance, mouse_tracker in rows: +# if not json.loads(mouse_tracker): +# index_list.append(index) +# print(f"Index: {index}, Distance: {distance}") +# # print(f"Index: {index}, Distance: {distance}") +# # print(f"Mouse tracker: {json.loads(mouse_tracker)}") + +# # 根据 index 删除数据 +# # for index in index_list: +# # cursor.execute(f"DELETE FROM Tracker WHERE id = {index}") +# # conn.commit() +# # # 关闭连接 +# conn.close() + + + + +# print(index_list)