如何在编辑QTableView单元格后更改背景颜色？

Question

我有一个带有自定义模型和代理的 QTableView，如何在编辑后更改单元格的背景颜色？

我应该在代理的 setModelData() 中完成这个操作吗？

index.model.setData(index, QVariant(True),Qt.UserRole) 

并且在模型的 data（）中，它自己是如何调用的？

if role == Qt.BackgroundColorRole:
    if index.model().data(index,Qt.UserRole).toBool():
        return QVariant(QColor(Qt.darkBlue))

在模型的 setData() 中，我没有任何类似以下代码的内容：

if role==Qt.UserRole:
    ....

这该怎么正确地做呢？

编辑： 这是我在自定义模型中的整个 setData() 方法。

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

if index.isValid() and 0 <= index.row() < len(self.particles):
    particle = self.particles[index.row()]
    column = index.column()
    if column == ID:
        value,ok= value.toInt()
        if ok:
            particle.id =value                 
    elif column == CYCLEIDANDNAME:
        cycleId,cycleName= value.toString().split(' ')
        particle.cycleId =cycleId
#                also need to set cycleName
        for name in self.cycleNames:
            if name.endsWith(cycleName):
                particle.cycleFrameNormalized=particle.cycleName = name
                break

    elif column == CYCLEFRAME:
        value,ok= value.toInt()
        if ok:
            print 'set new val to :',value
            particle.cycleFrame =value 
#                    self.setData(index,QVariant(QColor(Qt.red)),Qt.BackgroundRole)

    elif column == CLASSID:
        value,ok= value.toInt()
        if ok:
            particle.classId =value                 
    elif column == VARIATIONID:
        value,ok= value.toInt()
        if ok:
            particle.variationId =value                 

    self.emit(SIGNAL("dataChanged(QModelIndex,QModelIndex)"),
              index, index)

    return True

return False

抱歉，我仍然不清楚，我将从《快速GUI开发书籍》示例中粘贴完整代码： 我在这里发布了我的代码 http://pastebin.com/ShgRRMcY 如何更改代码以使单元格背景颜色在编辑后更改？

- Shuman

---

你使用的是哪个模型？ - user1006989

@X.Jacobs 我正在使用一个继承自QAbstractTableModel的自定义模型，并且我已经实现了自己的setData()，但它只处理Qt.EditRole。 - Shuman

它如何处理 Qt.EditRole？请发布您的代码的可管理版本。 - user1006989

@Shuman：在setData中调用setData有点毫无意义。它会调用自身。您需要将该颜色存储在数据(particle)中的某个位置，在data方法中检查并返回BackgroundRole。顺便说一句，您在setData中没有以任何方式使用role。 - Avaris

Answer 1

你需要以某种方式跟踪模型中编辑的项目。你不需要使用 UserRole。你可以在内部保留这个信息，但是如果你想向外界公开这个信息，UserRole非常适合。

以下是一个简单的示例，你可以根据自己的代码进行调整：

import sys
from PyQt4 import QtGui, QtCore

class Model(QtCore.QAbstractTableModel):
    def __init__(self, parent=None):
        super(Model, self).__init__(parent)

        # list of lists containing [data for cell, changed]
        self._data = [[['%d - %d' % (i, j), False] for j in range(10)] for i in range(10)]

    def rowCount(self, parent):
        return len(self._data)

    def columnCount(self, parent):
        return len(self._data[0])

    def flags(self, index):
        return QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsEditable

    def data(self, index, role):
        if index.isValid():
            data, changed = self._data[index.row()][index.column()]

            if role in [QtCore.Qt.DisplayRole, QtCore.Qt.EditRole]:
                return data

            if role == QtCore.Qt.BackgroundRole and changed:
                return QtGui.QBrush(QtCore.Qt.darkBlue)

    def setData(self, index, value, role):
        if role == QtCore.Qt.EditRole:
            # set the new value with True `changed` status
            self._data[index.row()][index.column()] = [value.toString(), True]
            self.dataChanged.emit(index, index)
            return True
        return False

if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)

    t = QtGui.QTableView()
    m = Model(t)
    t.setModel(m)
    t.show()

    sys.exit(app.exec_())

- Avaris

---

谢谢，我尝试过了，虽然我正在使用tableView。在我的模型中的setData()中，我使用了self.setData(index,QVariant(QColor(Qt.red)),Qt.BackgroundRole)，但它没有起作用。 - Shuman

@Shuman：setData本身并不会做任何事情。你应该将传递给它的数据存储起来。如果你展示一下你的模型可能会更好。 - Avaris

@Shuman 再试一次，看看当你使用 self.setData(index,"foo-bar",Qt.EditRole) 时会发生什么。 - user1006989

@Avaris @X.Jacobs，我已经将我的模型粘贴到pastebin上，我能否只将userRole数据存储在单元格本身中？我的意思是如果我使用默认的表格模型，它会存储它，但现在我正在使用自定义模型和自定义的setData()，我该如何将值保存在单元格本身中？（实际上，即使是默认模型也不会将值保存在单元格本身中，它会将值存储在模型的某个实例变量中）。 - Shuman

Answer 2

QTableWidget有一个信号itemChanged，需要连接到一个slot。一旦连接到一个slot，信号将传递已更改的QTableWidgetItem。从那里，您可以使用QTableWidgetItem的方法，例如setBackgroundColor来更改背景。

这是一个例子

#! /usr/bin/env python2.7

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


class Main(QTableWidget):

    def __init__(self):

        super(Main, self).__init__(2, 5)
        layout = QHBoxLayout(self)

        self.itemChanged.connect(self.changeBG)

    def changeBG(self, cell):

        cell.setBackgroundColor(QColor(225, 0, 225))

if __name__ == "__main__":
    app = QApplication(sys.argv)
    main = Main()
    main.show()
    app.exec_()

- Jeff

---

很遗憾，OP正在使用QTableView而不是QTableWidget。 - ekhumoro