From 7e620b92a4e2b6b5437d5d82ac946db8e342daa9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Okan=20A=C5=9F=C4=B1k?= Date: Mon, 4 Dec 2017 18:19:46 +0300 Subject: [PATCH] create new remove items, remove item is fixed --- src/tools/visualStates_py/gui/treemodel.py | 4 ++-- src/tools/visualStates_py/gui/visualstates.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/tools/visualStates_py/gui/treemodel.py b/src/tools/visualStates_py/gui/treemodel.py index 107e99e15..bbc7146e6 100644 --- a/src/tools/visualStates_py/gui/treemodel.py +++ b/src/tools/visualStates_py/gui/treemodel.py @@ -124,12 +124,12 @@ def removeState(self, state, parent=None): if parent is None: parent = self.rootNode childToBeRemoved = None - for s in parent.getChildren(self.rootNode): + for s in parent.getChildren(): if s.id == state.id: childToBeRemoved = s break - if childToBeRemoved != None: + if childToBeRemoved is not None: # print('remove child.id:' + str(childToBeRemoved.id)) parent.removeChild(childToBeRemoved) self.layoutChanged.emit() diff --git a/src/tools/visualStates_py/gui/visualstates.py b/src/tools/visualStates_py/gui/visualstates.py index 7f2f03465..a21b2ad5f 100644 --- a/src/tools/visualStates_py/gui/visualstates.py +++ b/src/tools/visualStates_py/gui/visualstates.py @@ -181,6 +181,8 @@ def createMenu(self): def newAction(self): self.automataScene.clearScene() + self.treeModel.removeAll() + # create new root state self.rootState = State(0, 'root', True) self.automataScene.setActiveState(self.rootState)