TrialCommend/TR/TrialRecommend/dis_collect.py

import csv
import mysql.connector
from tkinter import messagebox

def dis_collec(user,data_index):
    id=''
    with open('collections/{}.csv'.format(user), 'r', encoding='utf-8', newline='') as file:
        reader = csv.reader(file)
        index=0
        for row in reader:
            if index == data_index:
                id=row[0]
                break
            index=index+1
    db = mysql.connector.connect(
        host="localhost",
        user="root",
        passwd="A121023020309a",
        database="trial"
    )

    sql='DELETE FROM collections WHERE user_id = %s AND essay_id = %s'
    cursor=db.cursor()
    success='true'
    try:
        cursor.execute(sql, (user, id))
        db.commit()
    except db.Error as e:
        print(f"An error occurred: {e}")
        db.rollback()
        messagebox.showinfo("Error", "取消收藏失败！")
        success='false'
    finally:
        # 确保cursor关闭，释放资源
        cursor.close()
    if success == 'true':
        messagebox.showinfo('提示','取消收藏成功！')