PyQt QFileIconProvider类自定义图标

Question

我有一个带有QFileSystemModel的QTreeView来创建文件夹目录树。由于我在Windows上使用可怕的默认图标，我想创建一些我自己的（文件夹，网络驱动器等）。从阅读文档，我看到我需要创建自己的QFileIconProvider类并覆盖默认图标。我的问题是我在世界上如何在python中做到这一点？我发现这篇帖子here提供了一个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_())