QListView 按数据属性过滤（不是文本）

Question

我希望视图根据数据的内部属性显示相同数据的不同子集，当然这些视图共享相同的模型。

这是一个带有屏幕截图的示例：

1 - 显示带有“名称”和“状态”的项目列表的表格视图

2 - 一个列表视图，显示按“状态”过滤的项目，但不显示状态（理想情况下，我希望此列表视图在名称旁边有状态颜色，但这可能是一个不同的问题）。

我的目标是创建两种查看数据的方式：作为全局（未过滤）表，以及作为一种看板，其中包含每个“状态”（或任何其他属性）的列表。

表格视图显示所有

显示过滤数据的列表视图

这是我的模型的代码以及我如何构建数据。

我对 Qt 很满意，但我对它的 Model-View 类还不熟悉，所以如果有任何关于如何更好地做到这一点的建议，我们非常欢迎！

import sys
from PySide2.QtWidgets import *
from PySide2 import QtCore, QtGui
from PySide2.QtCore import Qt


class TestModel(QtCore.QAbstractTableModel):
    def __init__(self, items=None, parent=None):
        super(TestModel, self).__init__(parent)
        self.__items = items or []

    def headerData(self, section, orientation, role):
        headers = ('Name', 'Status')
        if role == Qt.DisplayRole and orientation == Qt.Horizontal:
                return headers[section]

    def columnCount(self, parent):
        return 2

    def rowCount(self, parent):
        return len(self.__items)

    def data(self, index, role):
        column = index.column()
        item = self.__items[index.row()]
        # Name
        if column == 0:
            if role == Qt.DisplayRole:
                return item.name
            if role == Qt.EditRole:
                return item.name
        # Status
        elif column == 1:
            if role == Qt.DisplayRole:
                return item.status
            if role == Qt.EditRole:
                return item.status
            if role == Qt.DecorationRole:
                pixmap = QtGui.QPixmap(26, 26)
                pixmap.fill(item.status_color)
                icon = QtGui.QIcon(pixmap)
                return icon

        if role == Qt.ToolTipRole:
            return item.__repr__()


class Item(object):
    def __init__(self, name='', status=None):
        self._valid_statuses = ('active', 'hold', 'closed', 'done')
        self._color_map = {self._valid_statuses[0]: '#56FF51',  # valid
                           self._valid_statuses[1]: '#B8E5EC', # hold
                           self._valid_statuses[2]: '#F42525',
                           self._valid_statuses[3]: '#E7F64D'
                           } # closed
        self.name = name
        self._status = status or self._valid_statuses[0]

    @property
    def status(self):
        return self._status

    @status.setter
    def status(self, status):
        if status not in self._valid_statuses:
            status = self._valid_statuses[0]
        self._status = status

    @property
    def status_color(self):
        # return black if invalid
        return self._color_map.get(self._status, '#000000')


app = QApplication([])

data = [
    Item(name='one', status='active'),
    Item(name='two', status='active'),
    Item(name='three', status='hold'),
    Item(name='four', status='closed'),
    Item(name='five', status='done'),
]

model = TestModel(data)

# Table
table_view = QTableView()
table_view.show()
table_view.setModel(model)

# List with the wanted filters i.e.: by 'approved' status
list_view = QListView()
list_view.show()
list_view.setModel(model)

sys.exit(app.exec_())

Answer 1

在这种情况下，解决方案是使用 QSortFilterProxyModel 并制作自定义过滤器：

class StatusFilterModel(QtCore.QSortFilterProxyModel):
    def __init__(self, parent=None):
        super(StatusFilterModel, self).__init__(parent)
        self._status = None

    @property
    def status(self):
        return self._status

    @status.setter
    def status(self, status):
        self._status = status
        self.invalidateFilter()

    def filterAcceptsRow(self, source_row, source_parent):
        if self.status is None:
            return True
        source_index = self.sourceModel().index(source_row, 1, source_parent)
        status = source_index.data()
        return self.status == status

list_view = QListView()
list_view.show()
proxy_model = StatusFilterModel()
proxy_model.setSourceModel(model)
proxy_model.status = "active"
list_view.setModel(proxy_model)