如何撤消 QAbstractTableModel 中的更改？

Question

我有一个简单的例子：我的 QAbstractTableModel 最后一列中的值等于第 1 列中的值乘以 2。因此，每次对第 1 列中的值进行更改时，都会导致第 2 列中的值发生变化。

当最后一列中的值发生变化时 - 将显示一个消息框，询问用户是否希望确认操作。

假设用户更改了第 1 列中的值并看到此消息：如果单击取消，我希望我的应用撤消更改（return 第 1 列和第 2 列中的旧值），你能帮我吗那？我需要定义我的 'go_back()' 函数：

class Mainwindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()

        self.table = QtWidgets.QTableView()
        self.setCentralWidget(self.table)

        self.data = [
            [1, 0.18, 0.36],
            [2, 0.25, 0.50],
            [3, 0.43, 0.86],
            ]

        self.model = MyModel(self.data)
        self.table.setModel(self.model)

        self.table.setSelectionBehavior(self.table.SelectRows)
        self.table.setSelectionMode(self.table.SingleSelection)

        self.model.dataChanged.connect(lambda index: self.count_last_column(index))
        self.model.dataChanged.connect(lambda index: self.if_last_column_changed(index))

    def calculations(self, position):
        value = self.model.list_data[position][1] * 2
        return value

    def count_last_column(self, index):
        if index.column() == 1:
            position = index.row()
            self.model.setData(self.model.index(position,2), self.calculations(position))

    def if_last_column_changed(self, index):
        if index.column() == 2:
            message_box, message_box_button = self.show_message_box()

            if message_box_button == 'Ok':
                pass
            elif message_box_button == 'Cancel':
                self.go_back()

    def show_message_box(self):
        self.message_box = QtWidgets.QMessageBox(QtWidgets.QMessageBox.Warning, 'Action', 'Value in column 3 has changed, confirm action?')

        self.message_box.Ok = self.message_box.addButton(QtWidgets.QMessageBox.Ok)
        self.message_box.Cancel = self.message_box.addButton(QtWidgets.QMessageBox.Cancel)

        self.message_box.exec()

        if self.message_box.clickedButton() == self.message_box.Ok:
            return (self.message_box, 'Ok')
        elif self.message_box.clickedButton() == self.message_box.Cancel:
            return (self.message_box, 'Cancel')

    def go_back(self):
        pass #################



class MyModel(QtCore.QAbstractTableModel):

    def __init__(self, list_data = [[]], parent = None):
        super(MyModel, self).__init__()
        self.list_data = list_data

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

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

    def data(self, index, role):

        if role == QtCore.Qt.DisplayRole:
            row = index.row()
            column = index.column()
            value = self.list_data[row][column]
            return value

        if role == QtCore.Qt.EditRole:
            row = index.row()
            column = index.column()
            value = self.list_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.list_data[row][column] = value
            self.dataChanged.emit(index, index)
            return True

        return False

    def flags(self, index):
        return QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsUserCheckable


if __name__ == '__main__':

    app = QtWidgets.QApplication([])
    application = Mainwindow()
    application.show()


    sys.exit(app.exec())

Answer 1

一种选择是将 QUndoStack 与项目模型一起使用，并将 QUndoCommand 对象推入 setData 中的堆栈。这种方法的好处是，如果您愿意，可以轻松实现更多 undo/redo 控件。

在MyModel中，只需在构造函数中创建一个堆栈，并在修改列表数据之前添加一行以将命令压入堆栈（因此可以将先前的值存储在命令中）。 class的其余部分不变。

class MyModel(QtCore.QAbstractTableModel):

    def __init__(self, list_data = [[]], parent = None):
        super(MyModel, self).__init__()
        self.list_data = list_data
        self.stack = QtWidgets.QUndoStack()

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

        if role == QtCore.Qt.EditRole:
            row = index.row()
            column = index.column()
            
            self.stack.push(CellEdit(index, value, self))
            
            self.list_data[row][column] = value
            self.dataChanged.emit(index, index)
            return True

        return False

使用传递给构造函数的索引、值和模型创建 QUndoCommand，以便可以通过调用撤消或重做来修改所需的单元格。

class CellEdit(QtWidgets.QUndoCommand):

    def __init__(self, index, value, model, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.index = index
        self.value = value
        self.prev = model.list_data[index.row()][index.column()]
        self.model = model

    def undo(self):
        self.model.list_data[self.index.row()][self.index.column()] = self.prev

    def redo(self):
        self.model.list_data[self.index.row()][self.index.column()] = self.value

现在 go_back 中需要做的就是为两个已修改的单元格调用两次撤消方法。

def go_back(self):
    self.model.stack.undo()
    self.model.stack.undo()

如何撤消 QAbstractTableModel 中的更改？

How to undo a change in QAbstractTableModel?

python

pyqt

pyqt5

undo

qabstracttablemodel