PyQt QFileIconProvider class custom icons

Question

I have a QTreeView with a QFileSystemModel to create a folder directory tree. As I am on Windows with the terrible default icons I want to create a few of my own (folder, network drive, etc). From reading the docs I see I need to create my own QFileIconProvider class and override the default icons. My question is how in the world do I do this in python? I found this post here which gives a C++ solution but I do not know enough to translate it.

Here's my code thus far:

class IconProvider(QtGui.QFileIconProvder):
    def __init__(self):
        super(IconProvider, self).__init__()

        folder_icon = QtGui.QIcon("folder.png") # what the heck do I do with this?

self.file_model = QtGui.QFileSystemModel()
self.file_model.setIconProvider(IconProvider())
self.browse_tree = QtGui.QTreeView()
self.browse_tree.setModel(self.file_model)

Answer 1

You have to overwrite the icon() method of QFileIconProvider and return the icon according to the QFileInfo.

import sys
from PyQt4 import QtGui, QtCore

class IconProvider(QtGui.QFileIconProvider):
    def icon(self, fileInfo):
        if fileInfo.isDir():
            return QtGui.QIcon("folder.png") 
        return QtGui.QFileIconProvider.icon(self, fileInfo)

if __name__=="__main__":
    app = QtGui.QApplication(sys.argv)
    file_model = QtGui.QFileSystemModel()
    file_model.setIconProvider(IconProvider())
    file_model.setRootPath(QtCore.QDir.currentPath());
    browse_tree = QtGui.QTreeView()
    browse_tree.setModel(file_model)
    browse_tree.show()
    sys.exit(app.exec_())