Флажок PyQt QAbstractTableModel не устанавливается
Я использую собственную табличную модель с QAbstractTableModel, где у меня есть первый столбец с флажком (checkable причиной флаги 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 ответ
Решение
Вы должны хранить состояния для этого, нам нужно иметь что-то постоянное в качестве ссылки, для этого мы используем 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_())