from flask import Flask, request
from flask_cors import CORS
import sqlite3
from threading import Lock

app = Flask(__name__)
CORS(app)
conn = sqlite3.connect('test.db', check_same_thread=False)
cursor = conn.cursor()
lock = Lock()  # 添加线程锁

@app.route('/add_task', methods=['POST'])
def add_task():
    task_name = request.form['taskName']

    # 使用线程锁确保在同一线程中对数据库进行操作
    with lock:
        cursor.execute("INSERT INTO GroundMagProfileParams (Key) VALUES (?)", (task_name,))
        conn.commit()

    return 'Task added successfully'

@app.route('/get_tasks', methods=['GET'])
def get_tasks():
    with lock:
        cursor.execute("SELECT Key,ID FROM GroundMagProfileParams")
        tasks = cursor.fetchall()
    return {'tasks': tasks}

@app.route('/update_task_name', methods=['POST'])
def update_task_name():
    task_id = request.form['taskId']
    new_task_name = request.form['newTaskName']

    with lock:
        cursor.execute("UPDATE GroundMagProfileParams SET Key = ? WHERE ID = ?", (new_task_name, task_id))
        conn.commit()

    return 'Task name updated successfully'

if __name__ == '__main__':
    app.run()