[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): def __init__(self, graph, nodeType, parent=None):
super(AddNodeCommand, self).__init__(graph, parent) super(AddNodeCommand, self).__init__(graph, parent)
self.nodeType = nodeType self.nodeType = nodeType
self.node = None self.nodeName = None
def redoImpl(self): def redoImpl(self):
self.node = self.graph.addNewNode(self.nodeType) self.nodeName = self.graph.addNewNode(self.nodeType).name
self.setText("Add Node {}".format(self.node.getName())) self.setText("Add Node {}".format(self.nodeName))
return True return True
def undoImpl(self): def undoImpl(self):
self.graph.removeNode(self.node.getName()) self.graph.removeNode(self.nodeName)
self.node = None
class RemoveNodeCommand(GraphCommand): class RemoveNodeCommand(GraphCommand):