46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
|
|
import sqlite3
|
|
import json
|
|
import random
|
|
|
|
# 根据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()
|
|
|
|
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]
|
|
|
|
mouse_data = get_data_by_distance(100)
|
|
for data in mouse_data:
|
|
print(data)
|
|
|
|
# cursor.execute("SELECT * FROM Tracker")
|
|
|
|
# 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)
|