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

问题描述

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

class resultsModel(QAbstractTableModel):
    def __init__(self,parent,headerData,arraydata,*args):
        QAbstractTableModel.__init__(self,*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,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,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)
        
        return result

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

resultsHeaders = ['Timestamp','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)

解决方法

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

为了正确添加列，应该正确调用 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)

注意：

rowCount() 和 columnCount() 都应该像基本实现一样具有默认关键字 parent 参数（公认的做法涉及 parent=QModelIndex()）；
类名应该总是大写，因此您应该将模型类重命名为ResultsModel；在官方 Style Guide for Python Code 上阅读有关该主题的更多信息；

pyqt python qabstractitemmodel qabstracttablemodel qtableview