如何在保留标头的同时使用空的2d数组初始化QAbstractTableModel

时间:2014-09-19 18:39:21

标签: python qt pyside qabstracttablemodel

我将用一个小例子来证明我的问题:

import sys
from PySide import QtCore, QtGui

class TaskModel(QtCore.QAbstractTableModel):
    def __init__(self, tasks=[[" ", " ", " ", " "]]):
    # def __init__(self, tasks=[[]]):
        super().__init__()
        self.__tasks = tasks
        self.__headers = ["Folder", "Command", "Patterns", "Active", "Recursive"]

    def rowCount(self, *args, **kwargs):
        return len(self.__tasks)

    def columnCount(self, *args, **kwargs):
        coln = len(self.__tasks[0])
        return coln

    def headerData(self, section, orientation, role):
        if role == QtCore.Qt.DisplayRole:
            if orientation == QtCore.Qt.Horizontal:
                return self.__headers[section]
            else:
                # set row names: color 0, color 1, ...
                return "%s" % str(section+1)

    def data(self, index, role):
        row = index.row()
        col = index.column()
        value = self.__tasks[row][col]

        # text content
        if role == QtCore.Qt.DisplayRole:
            return value


    def insertRows(self, position, rows, parent=QtCore.QModelIndex()):
        self.beginInsertRows(parent, position, position + rows - 1)
        row = ["a", "b", "c", "d"]
        for i in range(rows):
            self.__tasks.insert(position, row)
        self.endInsertRows()
        return True

class Mc(QtGui.QWidget):
    def __init__(self):
        super().__init__()
        self.tab = QtGui.QTableView()
        self.model = TaskModel()
        self.tab.setModel(self.model)
        self.addbtn = QtGui.QPushButton("Add")
        self.addbtn.clicked.connect(self.insert)
        layout = QtGui.QVBoxLayout()
        layout.addWidget(self.tab)
        layout.addWidget(self.addbtn)
        self.setLayout(layout)

    def insert(self):
        self.model.insertRow(0)

app = QtGui.QApplication(sys.argv)
mc = Mc()
mc.show()
sys.exit(app.exec_())

请注意,我有两行__init__函数用于TaskModel类,第一行给出[[“”,“”,“”,“”]作为__task的默认数据集,第二行一个给[[]]而不是。

第一个工作正常:

enter image description here

除了底部有一个不需要的行。

第二个__init__函数在尝试删除不需要的行时,使用[[]]作为默认数据集,但结果是灾难性的:

enter image description here

如何在仍然有效的情况下删除不需要的底行(带标题和所有内容)?

1 个答案:

答案 0 :(得分:1)

您有额外行的原因是因为您强制表模型的每个实例都有一个通过将其设置为默认参数。 (此外,使用可变对象作为默认参数几乎总是一个错误。如果这样做,同一个对象将最终在类的所有实例之间共享。)

缺少标题的问题是由您实施columnCount的方式引起的。如果列表中的第一个任务为空,则列计数将为零,因此不会显示任何标题。

以下是解决这些问题的一种方法:

class TaskModel(QtCore.QAbstractTableModel):
    def __init__(self, tasks=None):
        super().__init__()
        self.__tasks = [] if tasks is None else tasks
        self.__headers = ["Folder", "Command", "Patterns", "Active", "Recursive"]

    def rowCount(self, *args, **kwargs):
        return len(self.__tasks)

    def columnCount(self, *args, **kwargs):
        if self.__tasks:
            return len(self.__tasks[0])
        return len(self.__headers)