PyQT5 QAbstractTableModel|极客教程

PyQT5 QAbstractTableModel

在PyQt5中，QAbstractTableModel是一个用于显示表格数据的抽象类。它提供了一个数据模型，使得用户可以方便地在表格中展示数据，并且支持对数据的修改、删除和插入等操作。在本文中，我们将详细介绍如何使用QAbstractTableModel来创建一个简单的表格，并展示一些常用的操作方法。

安装PyQt5

在开始之前，我们首先需要安装PyQt5。可以通过以下命令来安装PyQt5：

pip install PyQt5

安装完成后，我们就可以开始使用QAbstractTableModel来创建表格了。

创建表格模型

要使用QAbstractTableModel，首先需要创建一个继承自QAbstractTableModel的子类。下面是一个简单的示例：

from PyQt5.QtCore import QAbstractTableModel, Qt, QVariant

class MyTableModel(QAbstractTableModel):
    def __init__(self, data):
        super(MyTableModel, self).__init__()
        self.data = data

    def rowCount(self, parent):
        return len(self.data)

    def columnCount(self, parent):
        return len(self.data[0])

    def data(self, index, role):
        if not index.isValid():
            return QVariant()

        if role == Qt.DisplayRole:
            return str(self.data[index.row()][index.column()])

        return QVariant()

在这个示例中，我们创建了一个继承自QAbstractTableModel的子类MyTableModel，并实现了rowCount、columnCount和data方法。其中，rowCount方法用于返回表格的行数，columnCount方法用于返回表格的列数，data方法用于返回表格中的数据。现在我们还没有具体的数据，我们先创建一些假数据来测试：

data = [
    [1, 'geek-docs.com', 100],
    [2, 'geek-docs.com', 200],
    [3, 'geek-docs.com', 300]
]

model = MyTableModel(data)

# 输出第一行第一列的数据
index = model.index(0, 0)
print(model.data(index, Qt.DisplayRole))

上述代码中，我们创建了一个包含3行3列数据的列表data，并使用MyTableModel类创建了一个表格模型model。然后我们通过index方法获取了表格中第一行第一列的数据，并输出了该数据。运行结果应该为1。

显示表格

创建了表格模型后，我们还需要创建一个视图来显示这个表格。这里我们使用QTableView来展示我们的表格数据：

from PyQt5.QtWidgets import QApplication, QTableView
import sys

app = QApplication(sys.argv)

view = QTableView()
view.setModel(model)
view.show()

sys.exit(app.exec_())

上述代码中, 我们创建了一个QTableView视图view，并将我们之前创建的表格模型model设置为view的数据模型。然后调用show方法展示表格。运行这段代码后，应该可以看到一个包含我们测试数据的表格。

添加数据

在QAbstractTableModel中，如果我们想要添加数据，可以通过实现insertRows、insertColumns、removeRows和removeColumns来实现。下面是一个添加行数据的示例：

class MyTableModel(QAbstractTableModel):
    def insertRows(self, position, rows, parent):
        self.beginInsertRows(parent, position, position + rows - 1)
        # 在position位置插入rows行数据
        # 如：self.data.insert(position, [1, 'geek-docs.com', 100])
        self.endInsertRows()
        return True

在这个示例中，我们实现了一个insertRows方法来插入行数据。在beginInsertRows和endInsertRows之间，我们可以将新增的行数据添加到self.data中，然后展示到表格中。

删除数据

同样的，如果我们想要删除数据，可以通过实现removeRows和removeColumns来实现。下面是一个删除行数据的示例：

class MyTableModel(QAbstractTableModel):
    def removeRows(self, position, rows, parent):
        self.beginRemoveRows(parent, position, position + rows - 1)
        # remove rows data from self.data
        # 如：self.data = self.data[:position] + self.data[position+rows:]
        self.endRemoveRows()
        return True

在这个示例中，我们实现了一个removeRows方法来删除行数据。在beginRemoveRows和endRemoveRows之间，我们可以通过对self.data进行切片操作来删除指定位置的行数据。

修改数据

如果我们想要修改数据，可以通过setData方法来实现。下面是一个修改数据的示例：

class MyTableModel(QAbstractTableModel):
    def setData(self, index, value, role):
        if role == Qt.EditRole:
            self.data[index.row()][index.column()] = value
            self.dataChanged.emit(index, index)
            return True
        return False

在这个示例中，我们实现了一个setData方法来修改数据。当role为EditRole时，我们修改指定位置的数据并发出dataChanged信号来通知视图更新。我们可以通过在视图中直接编辑数据来测试这个功能。