Флажок PyQt QAbstractTableModel не устанавливается - PullRequest
Новая
       52

Флажок PyQt QAbstractTableModel не устанавливается

0 голосов
/ 02 мая 2018

Я использую собственную табличную модель с QAbstractTableModel, где у меня есть первый столбец с флажком (checkable причиной flags Qt.ItemIsUserCheckable | Qt.ItemIsSelectable | Qt.ItemIsEnabled). У меня возникают проблемы, когда я пытаюсь использовать флажки, потому что они не подлежат проверке (невозможно выполнить проверку или снять отметку в им ) в показанной таблице.

Что я делаю не так? Я использую эти методы в классе собственной таблицы: 

def data(self, index, role):

    row = index.row()
    col = index.column()

    if role == Qt.DisplayRole:
        return '{0}'.format(self.tableData[row][col])

    if role == Qt.CheckStateRole: 
        if col == 0:
            return Qt.Unchecked
        else: 
            return None

def setData(self, index, value, role):
    if not index.isValid():
       return False

    if (role == Qt.CheckStateRole):
        if (index.data(Qt.CheckStateRole) == Qt.Checked):
            return True
        else:
            return False
    else:
        return False

1 Ответ

0 голосов
/ 02 мая 2018

Вы должны сохранить состояния для него, нам нужно иметь что-то постоянное в качестве ссылки, для этого мы используем QPersistentModelIndex, в данном случае словарь, где ключом является QPersistentModelIndex, а значением является состояние QCheckBox. 

from PyQt4.QtGui import *
from PyQt4.QtCore import *


class TableModel(QAbstractTableModel):
    def __init__(self, parent=None):
        super(TableModel, self).__init__(parent)
        self.tableData = [[1, 2, 3], [1, 2, 3], [1, 2, 3]]
        self.checks = {}

    def columnCount(self, *args):
        return 3

    def rowCount(self, *args):
        return 3

    def checkState(self, index):
        if index in self.checks.keys():
            return self.checks[index]
        else:
            return Qt.Unchecked

    def data(self, index, role=Qt.DisplayRole):
        row = index.row()
        col = index.column()
        if role == Qt.DisplayRole:
            return '{0}'.format(self.tableData[row][col])
        elif role == Qt.CheckStateRole and col == 0:
            return self.checkState(QPersistentModelIndex(index))
        return None

    def setData(self, index, value, role=Qt.EditRole):

        if not index.isValid():
            return False
        if role == Qt.CheckStateRole:
            self.checks[QPersistentModelIndex(index)] = value
            return True
        return False

    def flags(self, index):
        fl = QAbstractTableModel.flags(self, index)
        if index.column() == 0:
            fl |= Qt.ItemIsEditable | Qt.ItemIsUserCheckable
        return fl


if __name__ == "__main__":
    import sys

    app = QApplication(sys.argv)
    view = QTableView()
    model = TableModel()
    view.setModel(model)
    view.show()
    sys.exit(app.exec_())
...