PyQt4 本地目录视图，带有选择文件夹的选项

Question

我知道如何使用 QDirModel（或 QFileSystemModel）制作一个简单的 QTreeView() 来显示系统中的文件/文件夹，但我想在每个文件/文件夹旁边添加一个复选框，以便用户可以选择一些文件夹/文件在他的系统上。显然，我还需要知道他选择了哪些。有什么提示吗？

基本上是这样的......

以下是创建目录视图但没有复选框的示例代码。

from PyQt4 import QtGui

if __name__ == '__main__':

    import sys

    app = QtGui.QApplication(sys.argv)

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

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

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

    sys.exit(app.exec_())

Answer 1

您可以继承 QDirModel，并重新实现 data(index,role) 方法，您应该在其中检查 role 是否为 QtCore.Qt.CheckStateRole。如果是，您应该返回QtCore.Qt.Checked 或QtCore.Qt.Unchecked。此外，您还需要重新实现 setData 方法，以处理用户检查/取消检查，并 flags 返回 QtCore.Qt.ItemIsUserCheckable 标志，以启用用户检查/取消检查。即：

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
        self.emit(QtCore.SIGNAL("dataChanged(QModelIndex,QModelIndex)"), index, index)
        return True 

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

那你用这个类代替QDirModel:

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