在 PyQt 中，`QAbstractTableModel` 是一个抽象类，用于定义自定义的表格模型。通过继承这个类，我们可以创建一个与数据库交互的表格模型，并实现增加、删除数据的功能，同时将这些更改提交到数据库。 以下是一个完整的 Python 示例，展示如何使用 `QAbstractTableModel` 增加、删除数据并提交到 SQLite 数据库： ### 示例代码 ```python import sys from PyQt5.QtCore import QAbstractTableModel, Qt, QModelIndex from PyQt5.QtWidgets import QApplication, QTableView, QPushButton, QVBoxLayout, QWidget, QMessageBox import sqlite3 # 创建数据库连接和表 def create_connection(): conn = sqlite3.connect(':memory:') # 使用内存数据库进行演示 cursor = conn.cursor() cursor.execute(''' CREATE TABLE IF NOT EXISTS employees ( id INTEGER PRIMARY KEY, name TEXT NOT NULL, age INTEGER NOT NULL ) ''') conn.commit() return conn class EmployeeModel(QAbstractTableModel): def __init__(self, conn): super().__init__() self.conn = conn self.cursor = conn.cursor() self.load_data() def load_data(self): self.cursor.execute("SELECT * FROM employees") self._data = self.cursor.fetchall() self.layoutChanged.emit() # 通知视图数据已更改 def rowCount(self, parent=QModelIndex()): return len(self._data) def columnCount(self, parent=QModelIndex()): return 3 # id, name, age def data(self, index, role=Qt.DisplayRole): if role == Qt.DisplayRole: row = self._data[index.row()] return str(row[index.column()]) return None def headerData(self, section, orientation, role=Qt.DisplayRole): if role == Qt.DisplayRole and orientation == Qt.Horizontal: return ['ID', 'Name', 'Age'][section] return None def add_employee(self, name, age): try: self.cursor.execute("INSERT INTO employees (name, age) VALUES (?, ?)", (name, age)) self.conn.commit() self.load_data() # 刷新数据 except Exception as e: QMessageBox.critical(None, "Error", str(e)) def delete_employee(self, row): try: employee_id = self._data[row][0] # 获取ID self.cursor.execute("DELETE FROM employees WHERE id=?", (employee_id,)) self.conn.commit() self.load_data() # 刷新数据 except Exception as e: QMessageBox.critical(None, "Error", str(e)) class MainWindow(QWidget): def __init__(self): super().__init__() self.conn = create_connection() self.model = EmployeeModel(self.conn) self.table_view = QTableView() self.table_view.setModel(self.model) self.add_button = QPushButton("Add Employee") self.add_button.clicked.connect(self.add_employee) self.delete_button = QPushButton("Delete Selected Employee") self.delete_button.clicked.connect(self.delete_employee) layout = QVBoxLayout() layout.addWidget(self.table_view) layout.addWidget(self.add_button) layout.addWidget(self.delete_button) self.setLayout(layout) def add_employee(self): name, ok = QInputDialog.getText(self, "Input Dialog", "Enter Name:") if ok and name: age, ok = QInputDialog.getInt(self, "Input Dialog", "Enter Age:", min=1) if ok: self.model.add_employee(name, age) def delete_employee(self): selected = self.table_view.selectionModel().selectedRows() if selected: row = selected[0].row() self.model.delete_employee(row) else: QMessageBox.warning(self, "Warning", "Please select an employee to delete.") if __name__ == '__main__': app = QApplication(sys.argv) window = MainWindow() window.resize(400, 300) window.show() sys.exit(app.exec_()) ``` --- ### 代码解释 1. **数据库初始化 (`create_connection`)**: - 使用 SQLite 内存数据库（`:memory:`）进行演示。 - 创建了一个名为 `employees` 的表，包含 `id`, `name`, 和 `age` 三个字段。 2. **模型类 (`EmployeeModel`)**: - 继承自 `QAbstractTableModel`，实现了必要的接口方法： - `rowCount` 和 `columnCount`：返回行数和列数。 - `data`：返回指定索引的数据。 - `headerData`：返回表头信息。 - 提供了 `add_employee` 和 `delete_employee` 方法，用于向数据库添加和删除记录。 - 每次操作后调用 `load_data` 方法刷新模型数据。 3. **主窗口类 (`MainWindow`)**: - 包含一个 `QTableView` 用于显示数据。 - 提供两个按钮：`Add Employee` 和 `Delete Selected Employee`，分别用于添加和删除员工。 - 添加员工时，使用 `QInputDialog` 获取用户输入。 - 删除员工时，检查是否有选中的行。 4. **数据库提交**: - 每次增删操作后，都会调用 `conn.commit()` 将更改提交到数据库。 --- ###