Изменение цвета переднего плана элементов на основе самого текста элемента

У меня есть QTreeView с 5 столбцами, и я настроил делегата QComboBox для 2 из них (столбцы 1 и 2). Оба они имеют одного и того же делегата, и у них должен быть один и тот же раскрывающийся список. Допустим, список всегда такой: ["Далее", "Стоп"]. Пока все хорошо: делегат работает отлично.

Теперь проблема: я хотел, чтобы элементы столбцов 1 и 2 отображали текст разными цветами в зависимости от самого текста. Например: если текст «Далее», цвет текста должен быть зеленым, а если текст «Стоп», цвет должен быть красным.

После некоторых поисков я решил использовать делегат для установки цвета. Я нашел разные решения, но единственное, что почти работало для меня, было следующим (см. функцию paint()):

class ComboBoxDelegate(QtWidgets.QStyledItemDelegate):
    def paint(self, painter, option, index):
        painter.save()
        text = index.data(Qt.DisplayRole)
        if text == 'Next':
            color = GREEN
        elif text == 'Stop':
            color = RED
        else:
            color = BLACK
        painter.setPen(QPen(color))
        painter.drawText(option.rect, Qt.AlignLeft | Qt.AlignVCenter, text)
        painter.restore()


    def createEditor(self, parent, option, index):
        editor = QtWidgets.QComboBox(parent)
        editor.currentIndexChanged.connect(self.commitEditor)
        return editor

    # @QtCore.Slot
    def commitEditor(self):
        editor = self.sender()
        self.commitData.emit(editor)
        if isinstance(self.parent(), QtWidgets.QAbstractItemView):
            self.parent().updateEditorGeometries()
        self.closeEditor.emit(editor)

    def setEditorData(self, editor, index):
        try:
            values = index.data(QtCore.Qt.UserRole + 100)
            val = index.data(QtCore.Qt.UserRole + 101).strip('<>')
            editor.clear()
            for i, x in enumerate(values):
                editor.addItem(x, x)
                if val == x:
                    editor.setCurrentIndex(i)
        except (TypeError, AttributeError):
            LOG.warning(f"No values in drop-down list for item at row: {index.row()} and column: {index.column()}")

    def setModelData(self, editor, model, index):
        values = index.data(QtCore.Qt.UserRole + 100)
        if values:
            ix = editor.currentIndex()
            model.setData(index, values[ix], QtCore.Qt.UserRole + 101)
            model.setData(index, values[ix], QtCore.Qt.DisplayRole)

    def updateEditorGeometry(self, editor, option, index):
        editor.setGeometry(option.rect)

Как вы можете видеть на изображении ниже, выделение и наведение не работают должным образом:

Что я делаю не так?

Минимальный воспроизводимый пример (не забудьте импортировать ComboBoxDelegate):

import sys

from PySide2.QtGui import QStandardItemModel, QStandardItem
from PySide2.QtWidgets import QTreeView, QApplication, QMainWindow

from combo_box_delegate import ComboBoxDelegate

COLUMN_HEADER_LIST = ["0", "1", "2", "3", "4"]


class MyTreeView(QTreeView):
    def __init__(self):
        super(MyTreeView, self).__init__()
        self.model = QStandardItemModel()
        self.root = self.model.invisibleRootItem()
        self.setModel(self.model)
        delegate = ComboBoxDelegate(self)
        self.setItemDelegateForColumn(1, delegate)
        self.setItemDelegateForColumn(2, delegate)
        self.data = {
            "a": {
                "b": {
                    "c": ["Next", "Stop", "1", "Hello World!"],
                    "d": ["Next", "Stop", "2", "Hello World!"],
                    "e": ["Next", "Stop", "3", "Hello World!"],
                    "f": ["Next", "Stop", "4", "Hello World!"]
                }
            }
        }
        self.populate_tree(self.root, self.data)
        self._format_columns()

    def populate_tree(self, parent, data):
        for key, value in data.items():
            if isinstance(value, dict):
                child = QStandardItem(key)
                parent.appendRow(child)
                self.populate_tree(child, data[key])
            elif isinstance(value, list):
                items = [QStandardItem(key)] + [QStandardItem(str(val)) for val in value]
                parent.appendRow(items)

    def _format_columns(self):
        self.model.setHorizontalHeaderLabels(COLUMN_HEADER_LIST)
        self.expandAll()


class Main(QMainWindow):
    def __init__(self):
        super(Main, self).__init__()
        tree = MyTreeView()
        self.setCentralWidget(tree)
        self.setMinimumWidth(600)
        self.setMinimumHeight(400)


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