7

为了创建一个可检查的目录视图，我编写了以下代码。但是在 CheckableDirModel 中，每次检查一个文件夹时，它都必须遍历所有子文件夹来检查它们，这非常慢。我希望有人可以帮助我解决这个问题。

这就是这个现在的样子。但它很慢，特别是如果单击一个大文件夹。

在此处输入图像描述

代码可执行...

from PyQt4 import QtGui, QtCore


class CheckableDirModel(QtGui.QDirModel):
    def __init__(self, parent=None):
        QtGui.QDirModel.__init__(self, None)
        self.checks = {}

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.CheckStateRole:
            return QtGui.QDirModel.data(self, index, role)
        else:
            if index.column() == 0:
                return self.checkState(index)

    def flags(self, index):
        return QtGui.QDirModel.flags(self, index) | QtCore.Qt.ItemIsUserCheckable

    def checkState(self, index):
        if index in self.checks:
            return self.checks[index]
        else:
            return QtCore.Qt.Unchecked

    def setData(self, index, value, role):
        if (role == QtCore.Qt.CheckStateRole and index.column() == 0):
            self.checks[index] = value
            for i in range(self.rowCount(index)):
                self.setData(index.child(i,0),value,role)
            return True 

        return QtGui.QDirModel.setData(self, index, value, role)

    def exportChecked(self, acceptedSuffix=['jpg', 'png', 'bmp']):
        selection=[]
        for c in self.checks.keys():
            if self.checks[c]==QtCore.Qt.Checked and self.fileInfo(c).completeSuffix().toLower() in acceptedSuffix:
                try:

                    selection.append(self.filePath(c).toUtf8())
                except:
                    pass
        return selection   

if __name__ == '__main__':
    import sys

    app = QtGui.QApplication(sys.argv)

    model = QtGui.QDirModel()
    tree = QtGui.QTreeView()
    tree.setModel(CheckableDirModel())

    tree.setAnimated(False)
    tree.setIndentation(20)
    tree.setSortingEnabled(True)

    tree.setWindowTitle("Dir View")
    tree.resize(640, 480)
    tree.show()

    sys.exit(app.exec_())
4

1 回答 1

8

您不能单独存储每个文件的复选框状态。它们可能太多了。我建议您执行以下操作：

您保留用户实际单击的索引的复选框值列表。当用户单击某些内容时，您将一个条目添加到列表中（或者如果它已经存在则更新它），然后删除列表中存在的子索引的所有条目。您需要发出有关父索引数据的信号，并且所有子索引都已更改。

当请求复选框值时（通过调用data()模型），您在列表中搜索请求的索引并返回其值。如果列表中不存在索引，则搜索最近的父索引并返回其值。

请注意，除了执行缓慢之外，您的代码中还有另一个问题。当文件树的级别太多时，会发生“超出最大递归深度”异常。在实施我的建议时，不要以这种方式使用递归。文件树深度几乎是无限的。

这是实现：

from collections import deque

def are_parent_and_child(parent, child):
    while child.isValid():
        if child == parent:
            return True
        child = child.parent()
    return False


class CheckableDirModel(QtGui.QDirModel):
    def __init__(self, parent=None):
        QtGui.QDirModel.__init__(self, None)
        self.checks = {}

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role == QtCore.Qt.CheckStateRole and index.column() == 0:
            return self.checkState(index)
        return QtGui.QDirModel.data(self, index, role)

    def flags(self, index):
        return QtGui.QDirModel.flags(self, index) | QtCore.Qt.ItemIsUserCheckable

    def checkState(self, index):
        while index.isValid():
            if index in self.checks:
                return self.checks[index]
            index = index.parent()
        return QtCore.Qt.Unchecked

    def setData(self, index, value, role):
        if role == QtCore.Qt.CheckStateRole and index.column() == 0:
            self.layoutAboutToBeChanged.emit()
            for i, v in self.checks.items():
                if are_parent_and_child(index, i):
                    self.checks.pop(i)
            self.checks[index] = value
            self.layoutChanged.emit()
            return True 

        return QtGui.QDirModel.setData(self, index, value, role)

    def exportChecked(self, acceptedSuffix=['jpg', 'png', 'bmp']):
        selection=set()
        for index in self.checks.keys():
            if self.checks[index] == QtCore.Qt.Checked:
                for path, dirs, files in os.walk(unicode(self.filePath(index))):
                    for filename in files:
                        if QtCore.QFileInfo(filename).completeSuffix().toLower() in acceptedSuffix:
                            if self.checkState(self.index(os.path.join(path, filename))) == QtCore.Qt.Checked:
                                try:
                                    selection.add(os.path.join(path, filename))
                                except:
                                    pass
    return selection

我没有找到使用dataChanged信号通知视图所有子索引的数据已更改的方法。我们不知道当前显示了哪些索引，并且我们无法通知每个子索引，因为它可能很慢。所以我用 layoutAboutToBeChangedandlayoutChanged来强制视图更新所有数据。看来这个方法够快了。

exportChecked有点复杂。它没有经过优化，有时索引会被处理多次。我习惯于set()过滤重复项。如果它工作得太慢，也许可以以某种方式进行优化。但是，如果用户检查了包含许多文件和子目录的大型目录，则此功能的任何实现都会很慢。所以优化没有意义，尽量不要经常调用这个函数。

于 2013-06-12T10:02:53.030 回答