如何使用QFileSystemModel过滤多个目录并在QTreeView上显示它们?

时间:2017-11-13 14:30:50

标签: python qt pyqt4 pyside

如何使用QFileSystemModel过滤PyQt4中的多个目录并在QTreeView上显示它们?

我试着

  1. 子类QFileSystemModel,但我不知道如何返回rowCount

  2. 使用QSortFilterProxyModel -> filterAcceptsRow(),但很难 返回

  3. QFileSystemWatcher不好。
  4. 也许我没做正确的事。

    我应该通过win32监控目录并创建自己的模型和节点吗?

    例如1 enter image description here

    例如2 enter image description here

    #!/usr/bin/python
    
    import sys
    from functools import partial
    from PyQt4.QtGui import *
    from PyQt4.QtCore import *
    from PyQt4 import QtGui
    from PyQt4 import QtCore
    
    
    disk = 'C:/'
    dir_1 = 'C:/Python27'
    dir_1_1 = 'C:/Python27/Lib'
    dir_1_1_1 = 'C:/Python27/Lib/bsddb'
    
    
    class FilterProxyModel(QSortFilterProxyModel):
        def __init__(self):
            super(FilterProxyModel, self).__init__()
            self.count = 0
    
        def filterAcceptsRow(self, src_row, src_parent):
    
            src_model = self.sourceModel()
            src_index = src_model.index(src_row, 0, src_parent)
    
            item_data = src_model.itemData(src_parent)
            # print 'item_data: ', item_data
            item_var = src_index.data(Qt.DisplayRole)
            # print 'item_var: ', item_var
    
            file_path = src_model.filePath(src_index)
            file_path = str(file_path)
            if disk in file_path:
                # print 'file_path: ', file_path
                if file_path == disk:
                    # print 'file_path: ', file_path
                    return True
    
                elif dir_1 == file_path:
                    # print 'file_path: ', file_path
                    return True
    
                elif dir_1_1 == file_path:
                    # print 'file_path: ', file_path
                    return True
    
                elif dir_1_1_1 == file_path:
                    # print 'file_path: ', file_path
                    return True
    
                elif file_path.endswith('.py'):
                    print 'file_path: ', file_path
                    return True
    
            return False
    
    
    
    class MyQFileSystemModel(QFileSystemModel):
        """docstring for MyQFileSystemModel"""
        def __init__(self, parent=None):
            super(MyQFileSystemModel, self).__init__(parent)
    
        def columnCount(self, parent):
            return 1
    
    
    class MyWindow(QWidget):
        def __init__(self, parent=None):
            super(MyWindow, self).__init__(parent)
    
            self.pathRoot = QDir.rootPath()
    
            # self.model = QFileSystemModel(self)
            self.model = MyQFileSystemModel(self)
            self.model.setRootPath(self.pathRoot)
            self.model.setNameFilterDisables(0)
    
    
            filter = ['*.py', '*mll']
            self.model.setNameFilters(filter)
            self.model.setNameFilterDisables(0)
            # self.model.removeColumn(2)
    
            self.proxy = FilterProxyModel()
            self.proxy.setSourceModel(self.model)
            self.proxy.setDynamicSortFilter(True)
    
            self.treeView = QTreeView(self)
            self.treeView.setModel(self.proxy)
            # self.treeView.setRootIndex(self.indexRoot)
            # self.listView = QListView(self)
            # self.listView.setModel(self.proxy)
    
    
            self.layout = QHBoxLayout(self)
            self.layout.addWidget(self.treeView)
            # self.layout.addWidget(self.listView)
    
    
    
    if __name__ == "__main__":
        app = QApplication(sys.argv)
        app.setApplicationName('MyWindow')
    
        main = MyWindow()
        main.resize(666, 333)
    
        main.show()
    
        sys.exit(app.exec_())
    

    我就是这样做的。 最后,我放弃了,我用另一种方法。

1 个答案:

答案 0 :(得分:2)

我完全符合你的情况,并找到了一些东西。诀窍是获取要隐藏的路径列表,并使用setHiddenRow方法将其传递给treeView。这是我在addWidget方法

之前写的
# hide all no desired folders.
foldersToShow = []
for currentDir, dirnames, filenames in os.walk(rootPath): 
    if currentDir in "/folder/I/want/to/show" : #whatever condition you prefer, this is your main filter for what you want to KEEP
        foldersToShow.append(currentDir)
    else:
        index = self.model.index(currentDir)
        self.treeView.setRowHidden(index.row(), index.parent(), True) # what doesnt meet your requirements get to be hidden

从这里开始,有一个问题。由于我没有得到的原因,您不想显示要保留的文件夹中的文件。但有一种方法可以显示它们。从每个目录中,将false变为每个文件的rowHidden参数

# display all files from foldersToShow
for folder in foldersToShow:
    for currentDir, dirnames, filenames in os.walk(folder):
        for filename in filenames:
            filenamePath = os.path.join(currentDir.replace(r"/",'\\'), filename)
            fileId = self.model.index(filenamePath)
            self.treeView.setRowHidden(fileId.row(), fileId.parent(), False)

layout.addWidget(self.treeView)