从 QAbstractTableModel 类中访问视图和代理模型？

Posted 2023-02-16

【中文标题】从 QAbstractTableModel 类中访问视图和代理模型？

【英文标题】：Access view and proxy model from within QAbstractTableModel class?

【发布时间】：2018-02-01 12:46:47

【问题描述】：

使用QTableView 和QAbstractTableModel，我希望能够选择表格中的多个单元格并进行编辑影响所有选定的单元格。这就是我目前的做法，包括将view (QTableView) 和proxy_model (QSortFilterProxyModel) 传递到类中，以便我可以访问它们（以获取适当的行和列） ：

import sys
from pprint import pprint

try:
    from PySide2 import QtWidgets, QtCore
except ImportError:
    from PyQt5 import QtWidgets, QtCore


class MyTableModel(QtCore.QAbstractTableModel):
    def __init__(self,
                view,
                proxy_model,
                table_data,
                parent=None):
        QtCore.QAbstractTableModel.__init__(self, parent)

        self.table_data = table_data
        self.view = view
        self.proxy_model = proxy_model

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

    def columnCount(self, parent):
        return len(self.table_data[0])

    def flags(self, index):
        # Original, inherited flags:
        original_flags = super(MyTableModel, self).flags(index)

        return original_flags | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable

    def data(self, index, role):
        if role == QtCore.Qt.DisplayRole:
            row = index.row()
            column = index.column()
            item = index.internalPointer()

            if item is not None:
                print(item)
            value = self.table_data[row][column]

            return value

        return None

    def setData(self, index, value, role=QtCore.Qt.EditRole):
        if role == QtCore.Qt.EditRole:
            row = index.row()
            column = index.column()

            selection_model = self.view.selectionModel()
            selected_indexes = selection_model.selectedIndexes()

            for selected_index in selected_indexes:

                mapped_index = self.proxy_model.mapToSource(selected_index)
                selected_row = mapped_index.row()
                selected_column = mapped_index.column()

                self.table_data[selected_row][selected_column] = value
                pprint(self.table_data)

                self.dataChanged.emit(index, selected_index)  # from, to

            return True

        return False


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)

    table_data = [['one', 'two', 'three'], ['four', 'five', 'six']]

    view = QtWidgets.QTableView()
    proxy_model = QtCore.QSortFilterProxyModel()
    model = MyTableModel(view=view, proxy_model=proxy_model, table_data=table_data)

    proxy_model.setSourceModel(model)
    proxy_model.setDynamicSortFilter(True)

    view.setModel(proxy_model)
    view.setSortingEnabled(True)  # requires proxy model
    view.sortByColumn(0, QtCore.Qt.AscendingOrder)
    view.horizontalHeader().setStretchLastSection(True)
    view.horizontalHeader().setSectionsMovable(True)

    view.show()
    app.exec_()

我怀疑我不必将view 和proxy_model 传递给类，并且我可以通过其他方式访问这些对象。这可能吗？如果可以，怎么做？

---

我知道我的示例是特定于 Python 的，但我的问题实际上是一个与绑定无关的问题，所以我还用 qt 标记了我的问题。

【问题讨论】：

QAbstractTableModel 类应该不知道代理，只是代理的目的不是修改基础模型，而是使用代理作为中介。我想你不明白代理的使用。

你可以解释你使用代理和你的表的目标是什么。我认为您有设计问题而不是访问问题。

我的目标是能够选择多个单元格，输入一个新值并将该值填充到所有选定的单元格中。如果没有视图（检测“多选”）和代理模型（适当地映射单元格，以防它们被排序/过滤），我无法实现这一点。 @eyllanesc 我意识到数据模型不应该知道视图或代理，因此我的问题。

您不必在模型代码中执行的任务。你如何输入数据？你有表格吗？你必须创建一个函数来获取选定的单元格并使用函数 setData()，你不应该覆盖 setData() 方法。您有设计问题，如果您希望我们帮助您，您必须提供您的背景问题的minimal reproducible example。我强调，您不应该修改 setData 方法来执行该任务。

@eyllanesc 好的，我更新了代码。如果你运行它，你可以选择多个单元格，输入一个新值，然后按 Enter。该值被填充到您选择的每个单元格中，并且基础table_data 列表被更新。保持排序和过滤。这是我需要的行为。但我不想将视图和代理模型传递给我的表模型...

【参考方案1】：

基础模型不应该知道视图或代理，所以你应该有类似下面的东西：

class MyTableModel(QtCore.QAbstractTableModel):
    def __init__(self, table_data, parent=None):
        QtCore.QAbstractTableModel.__init__(self, parent)
        self.table_data = table_data

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

    def columnCount(self, parent):
        return len(self.table_data[0])

    def flags(self, index):
        original_flags = super(MyTableModel, self).flags(index)
        return original_flags | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role == QtCore.Qt.DisplayRole:
            row = index.row()
            column = index.column()
            item = index.internalPointer()
            if item is not None:
                print(item)
            value = self.table_data[row][column]
            return value

    def setData(self, index, value, role=QtCore.Qt.EditRole):
        if role == QtCore.Qt.EditRole:
            row = index.row()
            column = index.column()
            self.table_data[row][column] = value
            self.dataChanged.emit(index, index)
            return True
        return QtCore.QAbstractTableModel.setData(self, index, value, role)

代理具有将更改传递给原始模型的方法，因此通常不需要访问基本模型来进行更改，而是使用相同的代理，为了有一个订单我实现了一个小部件，我使用了@987654322 @ 方法，这可能导致无限循环，因此我们必须阻止其他 dataChanged() 信号，我们使用 blockSignals()。

class Widget(QtWidgets.QWidget):
    def __init__(self, *args, **kwargs):
        QtWidgets.QWidget.__init__(self, *args, **kwargs)
        self.view = QtWidgets.QTableView()
        self.setLayout(QtWidgets.QVBoxLayout())
        self.layout().addWidget(self.view)

        table_data = [['one', 'two', 'three'], ['four', 'five', 'six']]
        proxy_model = QtCore.QSortFilterProxyModel()
        model = MyTableModel(table_data=table_data)
        proxy_model.setSourceModel(model)
        proxy_model.setDynamicSortFilter(True)
        self.view.setModel(proxy_model)
        proxy_model.dataChanged.connect(self.on_data_changed)

        self.view.setSortingEnabled(True)  # requires proxy model
        self.view.sortByColumn(0, QtCore.Qt.AscendingOrder)
        self.view.horizontalHeader().setStretchLastSection(True)
        self.view.horizontalHeader().setSectionsMovable(True)

    def on_data_changed(self, _from, _to):
        model = _from.model() # proxy model
        model.blockSignals(True)
        for index in self.view.selectionModel().selectedIndexes():
            model.setData(index, _from.data())
        model.blockSignals(False)

if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    w = Widget()
    w.show()
    sys.exit(app.exec_())

更新：

PySide2 似乎有一个错误，因为它不返回选定的元素而是一个空列表，因此它没有进入 for 循环，在我们为其创建该列表之前，我们使用 @ 的 selectionChanged 信号987654328@，这在 PyQt5 中不会发生。：

class Widget(QtWidgets.QWidget):
    def __init__(self, *args, **kwargs):
        [...]
        self.view.horizontalHeader().setSectionsMovable(True)

        self.view.selectionModel().selectionChanged.connect(self.on_selectionChanged)
        self.currentSelected = []

    def on_selectionChanged(self, selected, deselected):
        for ix in selected.indexes():
           if ix not in self.currentSelected:
            self.currentSelected.append(ix)
        for ix in deselected.indexes():
            if ix in self.currentSelected:
                self.currentSelected.remove(ix)

    def on_data_changed(self, _from, _to):
        model = _from.model() # proxy model
        model.blockSignals(True)
        pindexes = [QtCore.QPersistentModelIndex(ix) for ix in self.currentSelected]
        for index in pindexes:
            model.setData(index, _from.data())
        model.blockSignals(False)

【讨论】：

哇，我希望我能给你的不仅仅是一个支持。最后一个问题，在on_data_changed 方法中，如何访问模型的table_data？我希望阅读它，以便在完成表格编辑后将其传递给我的程序的其余部分。

一般来说你不应该直接访问table_data，但是如果你想这样做你应该执行以下操作：model.sourceModel().table_data

我想通了。我将Widget.model 重命名为self.model，并且能够通过调用self.model.table_data 从on_data_changed 中访问它。再次非常感谢！

啊，model.sourceModel().table_data 更好。 “不应该直接访问”是什么意思？想象一下，例如此表下方的“提交”按钮，以便可以将表的内容提交到其他功能。这就是我想做的。

合适的做法是为模型创建一个返回副本的方法，因为如果直接修改table_data，模型将无法实现。