การเปลี่ยนสีแถวเมื่อคลิก PyQt

วิธีแก้ไขง่ายๆ คือการสร้างเขตข้อมูลในตารางฐานข้อมูลและใช้เป็นแฟล็กเพื่อเปลี่ยนสีพื้นหลัง แต่ถ้าคุณไม่สามารถทำการเปลี่ยนแปลงในฐานข้อมูลได้ คุณจะต้องสร้างบทบาทใหม่ จากนั้นเราจะใช้บทบาทนั้นเพื่อทำการเปลี่ยนแปลงที่จำเป็น เพื่อสิ่งนี้เราจะบันทึกแฟล็กในพจนานุกรม คีย์ของพจนานุกรมจะเป็น id เนื่องจากสิ่งเหล่านี้ไม่มีการเปลี่ยนแปลง

class SqlTableModel(QtSql.QSqlTableModel):
    ExecuteRole = QtCore.Qt.UserRole + 1

    def __init__(self, parent=None, db = QtSql.QSqlDatabase()):
        QtSql.QSqlTableModel.__init__(self, parent, db)
        self.d = {}

    def data(self, index, role):
        if role == self.ExecuteRole:
            _id = self.getId(index)
            if _id in self.d.keys():
                return self.d[_id]
            return False

        if role == QtCore.Qt.BackgroundRole:
            if self.data(index, self.ExecuteRole):
                return QtGui.QBrush(QtCore.Qt.red)
            if 'MILESTONE' in QtSql.QSqlTableModel.data(self, self.index(index.row(), 2), QtCore.Qt.DisplayRole):
                return QtGui.QBrush(QtCore.Qt.yellow)

        return QtSql.QSqlTableModel.data(self, index, role)

    def getId(self, index):
        ix = self.fieldIndex("id")
        return self.data(self.index(index.row(), ix), QtCore.Qt.DisplayRole)

    def setData(self, index, value, role):
        if role == self.ExecuteRole:
            self.d[self.getId(index)] = value
            return True
        return QtSql.QSqlTableModel.setData(self, index, value, role)

    def roleNames(self):
        rn = QtSql.QSqlTableModel.roleNames(self)
        rn[self.SelectRole] = QtCore.QByteArray(b'execute')
        return rn

การเปลี่ยนแปลงข้างต้นเปลี่ยนเฉพาะรายการที่เลือกจนกว่าจะรีเฟรชด้วยตนเอง แต่ลักษณะการทำงานนี้ไม่ใช่ลักษณะที่ต้องการทำ ดังนั้นจึงต้องออกสัญญาณโมเดล dataChanged()

class TableView(QtWidgets.QTableView):
    def contextMenuEvent(self, event):
        pos = event.pos()
        menu = QtWidgets.QMenu()
        stepx = menu.addAction('Mark Step as Executed')
        stepdx = menu.addAction('Clear Step Execution')

        action = menu.exec_(self.mapToGlobal(pos))
        if action == stepx:
            if self.model():
                index = self.model().index(self.rowAt(pos.y()), 2)
                self.model().setData(index, True, SqlTableModel.ExecuteRole)
                self.model().dataChanged.emit(self.model().index(index.row(), 0),
                                              self.model().index(index.row(), self.model().columnCount()-1),
                                              [QtCore.Qt.BackgroundRole])

การใช้งานปัจจุบันมีข้อดีคือยกเลิกการเลือกหากเราเปลี่ยนสถานะจากจริงเป็นเท็จ

self.model().setData(index, False, SqlTableModel.ExecuteRole)