python - PyQt4本地目录 View ，带有选择文件夹的选项

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

基本上是这样的...

下面是使目录 View 但没有复选框的示例代码。

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_())

最佳答案

您可以将QDirModel子类化，并重新实现data(index,role)方法，并在其中检查role是否为QtCore.Qt.CheckStateRole。如果是，则应返回QtCore.Qt.Checked或QtCore.Qt.Unchecked。同样，您还需要重新实现setData方法来处理用户检查/取消选中操作，并且需要flags返回QtCore.Qt.ItemIsUserCheckable标志，从而启用用户检查/取消选中操作。 IE。:

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)