在QSqlRelationalTableModel中插入行的Pyside6

from PySide6.QtSql import QSqlDatabase, QSqlRelationalTableModel, QSqlRelation
from PySide6.QtCore import Qt

# 初始化数据库连接
db = QSqlDatabase.addDatabase('QSQLITE')
db.setDatabaseName('example.db')

if not db.open():
    print("无法打开数据库")
    exit(1)

# 创建表（如果表不存在）
db.exec_("CREATE TABLE IF NOT EXISTS employees ("
         "id INTEGER PRIMARY KEY AUTOINCREMENT,"
         "name VARCHAR(255),"
         "department_id INTEGER,"
         "FOREIGN KEY (department_id) REFERENCES departments(id))")

# 创建模型
model = QSqlRelationalTableModel(db=db)
model.setTable('employees')
model.setEditStrategy(QSqlRelationalTableModel.OnFieldChange)

# 设置关系（如果有外键）
# model.setRelation(department_id_column_index, QSqlRelation('departments', 'id', 'name'))

# 插入新行
new_row = [None, 'John Doe', 1]  # 假设 department_id 为 1
model.insertRow(-1, new_row)

# 提交更改到数据库
if model.submitAll():
    print("行插入成功")
else:
    print("行插入失败:", model.lastError().text())

# 关闭数据库连接
db.close()