时间戳

首页 > 解决方案 > 如何在运行时使用 PYQT 向 QAbstractTable 模型插入新列

python - 如何在运行时使用 PYQT 向 QAbstractTable 模型插入新列

问题描述

我创建了一个 QAbstractTableModel 类型的模型类，向其中添加了许多方法，如下所示：

class resultsModel(QAbstractTableModel):
    def __init__(self, parent, headerData, arraydata, *args):
        QAbstractTableModel.__init__(self, parent, *args)
        self.arraydata = arraydata
        self.headerdata = headerData #['Timestamp', 'Force (N)', 'Diplacement (mm)']
    
    def rowCount(self, parent):
        return len(self.arraydata)

    def columnCount(self, parent):
        if len(self.arraydata) > 0: 
            return len(self.arraydata[0]) 
        return 0
        
    def headerData(self, col, orientation, role):
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return self.headerdata[col]
        return None
    
    #Make table cells non-editable
    def flags(self, index):
        return Qt.ItemIsEnabled | Qt.ItemIsSelectable   

    def data(self, index, role):
        if not index.isValid():
            return None
        value = self.arraydata[index.row()][index.column()]    
        if role == Qt.EditRole:
            return value
        elif role == Qt.DisplayRole:
            return value

    def setData(self, index, value, role):
        if not index.isValid():
            return None
        row = index.row()
        column = index.column()
        self.arraydata[row][column] = value
        self.dataChanged.emit(index, index)
        return True
        
    def setHeaderData(self, col, orientation, value, role):
        if role != Qt.DisplayRole or orientation != Qt.Horizontal:
            return False

        self.headerdata[col] = value
        result = self.headerdata[col]
        if result:
            self.headerDataChanged.emit(orientation, col, col)
        
        return result

当应用程序启动时，我使用 QTableview 和上面的模型：

resultsHeaders = ['Timestamp', 'Force (N)', 'Diplacement (mm)']
resultsData = [['','','']]
self.resultsTableModel = resultsModel(self, resultsHeaders, resultsData)
self.resultsTable = QTableView()
self.resultsTable.setModel(self.resultsTableModel)

在运行时，如果连接了串行设备，我想向模型添加一些额外的列，但在添加新的标头值之前，我很难为模型实现“insertColumns”方法。

#insert columns
???
#update header values
for i in range(len(resultsHeaders)):            
    self.resultsTable.model().setHeaderData(i, Qt.Horizontal, str(resultsHeaders[int(i)]), Qt.DisplayRole)

标签： pythonpyqtqtableviewqabstractitemmodelqabstracttablemodel

解决方案

必须实现适当的“插入列”功能，因为应该正确更新模型以反映实际列数，以便新标题也正确显示。

为了正确添加列，beginInsertColumn()（在添加新列数据之前）和 endInsertColumn()（在操作结束时）都应该正确调用。
通常，添加更多列需要实现insertColumns()，但考虑到情况以及它们无论如何都会调用这两种方法的事实，没有必要这样做，自定义函数就可以了。

以下仅插入单列，如果您想一次添加更多列，只需确保的第三个参数beginInsertColumns正确反映 ( newColumn + len(columnsToAdd)) 并且数组的标题数据和新的空值对应于新的列数。

class ResultsModel(QAbstractTableModel):
    # ...
    def addColumn(self, name):
        newColumn = self.columnCount()
        self.beginInsertColumns(QModelIndex(), newColumn, newColumn)
        self.headerdata.append(name)
        for row in self.arraydata:
            row.append('')
        self.endInsertColumns()

# ...
class Whatever(QWidget):
    # ...
    def addColumn(self):
        name = 'New column {}'.format(self.resultsTableModel.columnCount())
        self.resultsTableModel.addColumn(name)

注意：

  1. 两者都rowCount()应该像基本实现一样columnCount()有一个默认的关键字parent参数（公认的做法包括parent=QModelIndex()）；
  2. 类名应始终大写，因此您应将模型类重命名为R esultsModel；阅读有关Python 代码官方样式指南主题的更多信息；

推荐阅读