# -*- coding: utf-8 -*- from copy import deepcopy from tools import trace, timer from PyQt5.QtWidgets import ( QMessageBox, QUndoCommand, QUndoStack, ) from Model.LateralContribution.LateralContribution import LateralContribution class SetDataCommand(QUndoCommand): def __init__(self, data, index, column, new_value): QUndoCommand.__init__(self) self._data = data self._index = index self._column = column self._old = self._data.get_i(self._index)[self._column] self._new = new_value def undo(self): self._data._set_i_c_v(self._index, self._column, self._old) def redo(self): self._data._set_i_c_v(self._index, self._column, self._new) class AddCommand(QUndoCommand): def __init__(self, data, index): QUndoCommand.__init__(self) self._data = data self._index = index self._new = None def undo(self): self._data.delete_i([self._index]) def redo(self): if self._new is None: self._new = self._data.add(self._index) else: self._data.insert(self._index, self._new) class DelCommand(QUndoCommand): def __init__(self, data, rows): QUndoCommand.__init__(self) self._data = data self._rows = rows self._lc = [] for row in rows: self._lc.append((row, self._data.get_i(row))) self._lc.sort() def undo(self): for row, el in self._lc: self._data.insert(row, el) def redo(self): self._data.delete_i(self._rows) class SortCommand(QUndoCommand): def __init__(self, data, _reverse): QUndoCommand.__init__(self) self._data = data self._reverse = _reverse self._old = self._data.data self._indexes = None def undo(self): ll = self._data.data self._data.sort( key=lambda x: self._indexes[ll.index(x)] ) def redo(self): self._data.sort( _reverse=self._reverse, key=lambda x: x[0] ) if self._indexes is None: self._indexes = list( map( lambda p: self._old.index(p), self._data.data ) ) self._old = None class MoveCommand(QUndoCommand): def __init__(self, data, up, i): QUndoCommand.__init__(self) self._data = data self._up = up == "up" self._i = i def undo(self): if self._up: self._data.move_up(self._i) else: self._data.move_down(self._i) def redo(self): if self._up: self._data.move_up(self._i) else: self._data.move_down(self._i) class PasteCommand(QUndoCommand): def __init__(self, data, row, lcs): QUndoCommand.__init__(self) self._data = data self._row = row self._lcs = lcs self._lcs.reverse() def undo(self): self._data.delete(self._lcs) def redo(self): for bc in self._lcs: self._data.insert(self._row, bc) class DuplicateCommand(QUndoCommand): def __init__(self, data, rows, bc): QUndoCommand.__init__(self) self._data = data self._rows = rows self._lc = deepcopy(bc) self._lc.reverse() def undo(self): self._data.delete(self._lc) def redo(self): for bc in self._lcs: self._data.insert(self._rows[0], bc)