How to set custom icons for QTreeView items relative to it's suffix

Question

i'm using QTreeView and QFileSystemModel class to display folder structure at given path.Now i need help to customize the icons for tree item, according to file extension(i.e if file type is *.png set icon pngIcon.svg or if file type is *.mp4 then set icon to mp4Icon.svg).

Here is my code and snapshot of GUI.

from PySide2 import QtCore, QtGui, QtWidgets
import sys

class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.verticalLayout = QtWidgets.QVBoxLayout(self.centralwidget)
        self.verticalLayout.setObjectName("verticalLayout")
        self.inodTreeView = QtWidgets.QTreeView(self.centralwidget)
        self.inodTreeView.setObjectName("inodTreeView")
        self.verticalLayout.addWidget(self.inodTreeView)
        MainWindow.setCentralWidget(self.centralwidget)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        MainWindow.setWindowTitle(QtWidgets.QApplication.translate("MainWindow", "MainWindow", None, -1))

class inodWindow(QtWidgets.QMainWindow,Ui_MainWindow):
    def __init__(self, parent=None):
        super(inodWindow, self).__init__(parent)
        self.setupUi(self)
        iModel = QtWidgets.QFileSystemModel()
        iModel.setRootPath(QtCore.QDir.rootPath())
        self.inodTreeView.setModel(iModel)
        self.inodTreeView.setRootIndex(iModel.index('E:/Tech/Code/inode'))


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    w = inodWindow()
    w.show()
    sys.exit(app.exec_())

Answer 1

You have to use a QFileIconProvider to the QFileSystemModel, and overriden the method icon:

import os
current_dir =os.path.dirname(os.path.realpath(__file__))

class FileIconProvider(QtWidgets.QFileIconProvider):
    def icon(self, parameter):
        if isinstance(parameter, QtCore.QFileInfo):
            info = parameter
            if info.suffix() == "png":
                return QtGui.QIcon(os.path.join(current_dir, "pngIcon.svg"))
            elif info.suffix() == "mp4":
                return QtGui.QIcon(os.path.join(current_dir, "mp4Icon.svg"))
        return super(FileIconProvider, self).icon(parameter)

iModel = QtWidgets.QFileSystemModel()
iModel.setIconProvider(FileIconProvider())
iModel.setRootPath(QtCore.QDir.rootPath())