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

你必须覆盖QFileIconProvider的icon()方法，并根据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_())