Python数据库客户添加功能报错求助：判断记录是否添加成功

一、先解决报错问题 #

你遇到的这个错误，是因为Python在submit()函数里默认会把generatedID当作局部变量处理，但你并没有在函数内部给它赋值就直接使用了。虽然你在全局作用域定义了generatedID，但函数内部需要明确声明要使用全局变量，才能正确访问它。

只需要在submit()函数开头添加一行全局变量声明即可：

def submit():
    global generatedID  # 声明使用全局作用域的generatedID
    cursor.execute('SELECT * FROM tblCustomer WHERE customerID = ?', (generatedID,))
    record = cursor.fetchone()
    if record:
        addwindow.destroy()
        messagebox.showinfo("Customer", "The Record Was Successfully Added To The Database")
    else:
        messagebox.showerror("Error", "The Record Was Not Added To The Database")

二、再修正你的功能逻辑 #

现在你的代码存在一个核心问题：submit()函数只做了查询，完全没有执行插入客户信息的操作。你想要实现“添加客户后提示是否成功”，但当前代码根本没有把客户数据写入数据库，自然无法正确验证添加结果。

正确的流程应该是：

1. 生成唯一的generatedID（你已经用randomString()实现了这一步，建议加个循环确保ID不重复）
2. 在submit()中先执行INSERT语句，把客户的完整信息插入数据库
3. 通过查询或数据库反馈验证插入是否成功

下面是修正后的完整示例代码（假设你有获取客户姓名、邮箱等信息的GUI控件）：

import string
import random
import tkinter as tk
from tkinter import messagebox
import sqlite3

# 初始化数据库连接和游标（根据你的实际数据库类型调整）
conn = sqlite3.connect('customer_db.db')
cursor = conn.cursor()

def submit():
    global generatedID
    try:
        # 第一步：执行插入操作（替换成你实际的表字段）
        cursor.execute('''
            INSERT INTO tblCustomer (customerID, name, email)
            VALUES (?, ?, ?)
        ''', (generatedID, customer_name.get(), customer_email.get()))
        conn.commit()  # 必须提交事务，否则数据不会保存到数据库
        
        # 第二步：验证插入结果
        cursor.execute('SELECT * FROM tblCustomer WHERE customerID = ?', (generatedID,))
        record = cursor.fetchone()
        if record:
            addwindow.destroy()
            messagebox.showinfo("Customer", "The Record Was Successfully Added To The Database")
        else:
            messagebox.showerror("Error", "The Record Was Not Added To The Database")
    except sqlite3.IntegrityError:
        # 处理ID重复的异常（随机生成有极小概率重复）
        conn.rollback()
        messagebox.showerror("Error", "This Customer ID already exists! Generating a new one...")
        global generatedID
        generatedID = randomString()  # 重新生成唯一ID
    except Exception as e:
        conn.rollback()
        messagebox.showerror("Database Error", f"Failed to add record: {str(e)}")

def randomString(size = 6):
    char = string.ascii_uppercase + string.digits
    return ''.join(random.choice(char) for i in range(size))

# 生成初始客户ID
generatedID = randomString()

# 示例GUI代码（根据你的实际界面调整）
root = tk.Tk()
addwindow = tk.Toplevel(root)
customer_name = tk.Entry(addwindow, placeholder="Enter customer name")
customer_name.pack(pady=5)
customer_email = tk.Entry(addwindow, placeholder="Enter customer email")
customer_email.pack(pady=5)
submit_btn = tk.Button(addwindow, text="Submit", command=submit)
submit_btn.pack(pady=10)

root.mainloop()

# 程序结束时关闭数据库连接
conn.close()

几个关键注意点 #

• 必须调用conn.commit()：数据库插入/修改操作需要提交事务才能生效
• 异常处理：要考虑ID重复、数据库连接失败等异常情况，避免程序崩溃
• 全局变量优化：如果你的代码规模较大，建议用类封装数据库操作和变量，代替全局变量，代码会更易维护

内容的提问来源于stack exchange，提问作者JERRY SANTHOSH