publish

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)