仅在QFileDialog中显示某些文件

Question

这是我编写的示例代码：

from PyQt5.QtWidgets import (QApplication, QMainWindow, QFileDialog)
import sys


class Window(QMainWindow):

    def __init__(self):
        QMainWindow.__init__(self)
        dialog = QFileDialog()
        dialog.exec()

app = QApplication(sys.argv)
window = Window()
window.show()
app.exec_()

运行它时，我得到以下信息：

这些都是我目录中的所有文件。 我想过滤掉标题中没有单词“ spam”的文件，以便在我运行该文件时，仅显示“ spam.txt”，“ spam_eggs_and_spam.txt”和“ spam_eggs_tomato_and_spam.txt”。

Answer 1

您可以简单地添加一个过滤器，如下所示：

dialog = QFileDialog()
dialog.setNameFilter("Text Spam Files (*spam*.txt)")
dialog.exec()

但是，如果用户在文件名字段中输入*。*，则可以覆盖它。

更好的方法是实现自己的QSortFilterProxyModel ，这是我的尝试：

from PyQt5.QtWidgets import (QApplication, QMainWindow, QFileDialog)
from PyQt5.QtCore import QSortFilterProxyModel, QModelIndex
import sys


class FileFilterProxyModel(QSortFilterProxyModel):
    def __init__(self, parent=None):
        super(QSortFilterProxyModel, self).__init__(parent)

    def filterAcceptsRow(self, source_row: int, source_parent: QModelIndex) -> bool:
        source_model = self.sourceModel()
        index0 = source_model.index(source_row, 0, source_parent)
        if source_model.isDir(index0):
            return True
        return 'spam' in source_model.fileName(index0).lower()


class Window(QMainWindow):

    def __init__(self):
        QMainWindow.__init__(self)
        dialog = QFileDialog()
        dialog.setOption(QFileDialog.DontUseNativeDialog)
        dialog.setProxyModel(FileFilterProxyModel())
        dialog.setNameFilter("Text Files (*.txt)")
        dialog.exec()


app = QApplication(sys.argv)
window = Window()
window.show()
app.exec_()

Answer 2

要仅显示其中带有“垃圾邮件”字样的文件，您可以添加：

dialog.setNameFilters(["*spam*"])