HYTOS

# coding: utf-8

""" This is create command module """

import copy

from PyQt5.QtCore import *

from PyQt5.QtGui import *

from PyQt5.QtWidgets import *

class CreateCommand(QUndoCommand):

    def __init__(self, scene, items, parent=None):

        super(CreateCommand, self).__init__(parent)

        self._scene = scene

        self._items = items

        self._created = True

    def undo(self):

        """undo"""

        for item in self._items:

            if hasattr(item, 'transfer'):

                item.transfer.onRemoved.emit(item)

            else:

                self._scene.removeItem(item)

        self._created = False

    def redo(self):

        """redo"""

        if not self._created:

            for item in self._items:

                self._scene.addItem(item)

        self._created = True