Пользовательские значки класса PyQt QFileIconProvider

Question

У меня есть QTreeView с QFileSystemModel для создания дерева каталогов папок. Поскольку я нахожусь в Windows с ужасными значками по умолчанию, я хочу создать несколько своих собственных (папка, сетевой диск и т. Д.). Читая документы, я вижу, что мне нужно создать свой собственный класс QFileIconProvider и переопределить значки по умолчанию. Мой вопрос, как в мире я делаю это на питоне? Я нашел этот пост здесь , который дает решение C ++, но я не знаю достаточно, чтобы перевести его.

Вот мой код:

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

Необходимо перезаписать метод icon() для QFileIconProvider и вернуть значок в соответствии с 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_())