为了创建一个可检查的目录视图,我编写了以下代码。但是在每次检查文件夹时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_())
答案 0 :(得分: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
信号来通知视图所有子索引的数据已被更改的方法。我们不知道当前显示哪些索引,并且我们无法通知每个子索引,因为它可能很慢。因此,我使用layoutAboutToBeChanged
和layoutChanged
强制查看更新所有数据。看来这种方法足够快。
exportChecked
有点复杂。它没有经过优化,有时会多次处理索引。我使用set()
来过滤重复项。也许它可以以某种方式进行优化,如果它工作得太慢。但是,如果用户已经检查了一些包含许多文件和子目录的大型目录,则此函数的任何实现都将很慢。因此优化没有意义,只是尝试不经常调用此函数。