Search code examples
pythonqtpyqtqfiledialog

QFileDialog view folders and files but select folders only?


I'm creating my own custom file dialog using the following code:

file_dialog = QtGui.QFileDialog()
file_dialog.setFileMode(QtGui.QFileDialog.Directory)
file_dialog.setViewMode(QtGui.QFileDialog.Detail)
file_dialog.setOption(QtGui.QFileDialog.DontUseNativeDialog, True)

The behaviour that i'm interested in is for the user to be able to view both files and folders, but select folders only. (making files unselectable). Is that possible?

Note: Using the DirectoryOnly option is not good for me since it doesn't allow you to view files, just folders.

Edit (extra code that i forgot to add which responsible for being able to select multiple folders instead of just one):

file_view = file_dialog.findChild(QtGui.QListView, 'listView')
if file_view:
    file_view.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
f_tree_view = file_dialog.findChild(QtGui.QTreeView)
if f_tree_view:
    f_tree_view.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)

Solution

  • To prevent files being selected, you can install a proxy model which manipulates the flags for items in the file-view:

    dialog = QFileDialog()
    dialog.setFileMode(QFileDialog.Directory)
    dialog.setOption(QFileDialog.DontUseNativeDialog, True)
    
    class ProxyModel(QIdentityProxyModel):
        def flags(self, index):
            flags = super(ProxyModel, self).flags(index)
            if not self.sourceModel().isDir(index):
                flags &= ~Qt.ItemIsSelectable
                # or disable all files
                # flags &= ~Qt.ItemIsEnabled
            return flags
    
    proxy = ProxyModel(dialog)
    dialog.setProxyModel(proxy)
    
    dialog.exec()