[commands] improve AddNodeCommand

use node name instead of a direct reference to the Node
This commit is contained in:
Yann Lanthony 2017-11-07 20:27:52 +01:00
parent 87a81ec3fb
commit 9bc8eaeb1b

View file

@ -74,16 +74,15 @@ class AddNodeCommand(GraphCommand):
def __init__(self, graph, nodeType, parent=None):
super(AddNodeCommand, self).__init__(graph, parent)
self.nodeType = nodeType
self.node = None
self.nodeName = None
def redoImpl(self):
self.node = self.graph.addNewNode(self.nodeType)
self.setText("Add Node {}".format(self.node.getName()))
self.nodeName = self.graph.addNewNode(self.nodeType).name
self.setText("Add Node {}".format(self.nodeName))
return True
def undoImpl(self):
self.graph.removeNode(self.node.getName())
self.node = None
self.graph.removeNode(self.nodeName)
class RemoveNodeCommand(GraphCommand):