Как правильно переопределить метод QSqlTableModel.data() для декорирования выводимой информации

from PyQt5 import QtCore, QtWidgets, QtSql, QtGui
from PyQt5.QtCore import Qt
import sys
from PyQt5.QtSql import QSqlTableModel
 
 
class TableModel(QSqlTableModel):
    def __init__(self):
        super().__init__()
 
    def data(self, index, role):
        if role == Qt.DecorationRole:
            if index.column() == 4:
                return QtGui.QIcon("calendar.png")
        elif role == Qt.ForegroundRole:
            if index.column() == 2:
                return QtGui.QColor("red")
 
        return super().data(index, role)
 
 
def addRecord():
    stm.insertRow(stm.rowCount())
    stm.submit()
 
 
def delRecord():
    current_rov = tv.currentIndex().row()
    stm.removeRow(current_rov)
    stm.select()
 
 
 
def update_filter(self):
    filter_str = 'date BETWEEN "2023-04-01" AND "2023-06-01"'
    stm.setFilter(filter_str)
 
 
app = QtWidgets.QApplication(sys.argv)
window = QtWidgets.QWidget()
con = QtSql.QSqlDatabase.addDatabase("QSQLITE")
con.setDatabaseName("saper.db")
con.open()
 
stm = TableModel()
stm.setTable("users")
 
stm.select()
 
vbox = QtWidgets.QVBoxLayout()
tv = QtWidgets.QTableView()
tv.setModel(stm)
tv.resizeRowsToContents()
 
vbox.addWidget(tv)
btnAdd = QtWidgets.QPushButton("Add record")
btnDel = QtWidgets.QPushButton("Del record")
btnFilter = QtWidgets.QPushButton("Filter")
btnFilter.clicked.connect(update_filter)
btnAdd.clicked.connect(addRecord)
vbox.addWidget(btnAdd)
btnDel.clicked.connect(delRecord)
vbox.addWidget(btnDel)
vbox.addWidget(btnFilter)
window.setLayout(vbox)
window.resize(450, 400)
window.show()
sys.exit(app.exec_())

import datetime
import sys
 
import qdarktheme
from PyQt5 import QtWidgets, QtSql, QtGui
from PyQt5.QtCore import Qt, pyqtSlot
from PyQt5.QtSql import QSqlTableModel
from PyQt5.QtWidgets import QItemDelegate
 
from ui_app import Ui_Form
 
 
class TableModel(QSqlTableModel):
    def __init__(self):
        super().__init__()
        self.setEditStrategy(QSqlTableModel.OnFieldChange)
 
    def set_header(self):
        self.setHeaderData(1, Qt.Horizontal, "Имя")
        self.setHeaderData(2, Qt.Horizontal, "Пол")
        self.setHeaderData(3, Qt.Horizontal, "Возраст")
        self.setHeaderData(4, Qt.Horizontal, "Счет")
        self.setHeaderData(5, Qt.Horizontal, "Дата")
 
    def data(self, idx, role=None):
        value = super().data(idx, Qt.DisplayRole)
        if role == Qt.DecorationRole:
            if idx.column() == 5:
                return QtGui.QIcon("assets/calendar.png")
            if idx.column() == 2:
                if value == "мужской":
                    return QtGui.QIcon("assets/ok.png")
                if value == "женский":
                    return QtGui.QIcon("assets/false.png")
 
        elif role == Qt.ForegroundRole:
            if idx.column() == 5:
                return QtGui.QColor("green")
 
        elif role == Qt.TextAlignmentRole:
            if idx.column() == 3 or idx.column() == 4:
                return Qt.AlignCenter
            elif idx.column() == 5:
                return Qt.AlignLeft
        return super().data(idx, role)
 
 
class ItemDelegate(QItemDelegate):
    def __init__(self, parent):
        super(ItemDelegate, self).__init__(parent)
 
    def createEditor(self, parent, option, index):
        column = index.column()
        if column == 0:
            return None
        if column == 1:
            editor = QtWidgets.QLineEdit(parent)
            return editor
        elif column == 2:
            editor = QtWidgets.QComboBox(parent)
            editor.addItems(["мужской", "женский"])
            return editor
        elif column == 3:
            editor = QtWidgets.QSpinBox(parent)
            editor.setMinimum(16)
            editor.setMaximum(100)
            return editor
        elif column == 4:
            editor = QtWidgets.QSpinBox(parent)
            editor.setMaximum(100000)
            return editor
        elif column == 5:
            editor = QtWidgets.QDateEdit(parent=parent, calendarPopup=True)
            editor.setDate(datetime.datetime.now())
            return editor
 
class WindowApp(QtWidgets.QWidget, Ui_Form):
    def __init__(self):
        super().__init__()
        self.setupUi(self)
        self.setWindowTitle("Приложение")
        self.setFixedSize(550, 500)
        self.con = QtSql.QSqlDatabase.addDatabase("QSQLITE")
        self.con.setDatabaseName("app.db")
        self.con.open()
        self.model = TableModel()
        self.model.setTable("users")
        self.model.select()
        self.model.set_header()
        self.tableView.setItemDelegate(ItemDelegate(self.tableView))
        self.tableView.setModel(self.model)
        self.tableView.resizeRowsToContents()
        self.tableView.hideColumn(0)
        self.tableView.horizontalHeader().setStretchLastSection(True)
 
        self.btnAdd.clicked.connect(self.add_record)
        self.btnDel.clicked.connect(self.del_record)
 
    @pyqtSlot()
    def add_record(self):
        self.model.insertRow(self.model.rowCount())
        self.model.submit()
 
    @pyqtSlot()
    def del_record(self):
        current_idx = self.tableView.currentIndex()
        if self.tableView.selectionModel().isSelected(current_idx):
            current_row = current_idx.row()
            id_row = self.model.record(current_row).value("id")
            query = QtSql.QSqlQuery()
            query.prepare(f"DELETE FROM users WHERE id = ?")
            query.addBindValue(id_row)
            query.exec_()
            self.model.select()
 
 
if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    qss = """QDateEdit::drop-down{image:url('assets/calendar.png');}"""
    qdarktheme.setup_theme(custom_colors={"primary": "#00FF7F", "background": "#696969"}, additional_qss=qss)
    window = WindowApp()
    window.show()
    sys.exit(app.exec_())