Issue #2073: Added functionality to add all implants in fit implant tab to implant set list

2019-12-10 18:14:13 +02:00
parent d92e11893a
commit 9c7fa37a72
3 changed files with 28 additions and 17 deletions
--- a/gui/builtinViews/entityEditor.py
+++ b/gui/builtinViews/entityEditor.py
@@ -178,10 +178,17 @@ class EntityEditor(wx.Panel):
        return True

    def checkEntitiesExist(self):
-        if len(self.choices) == 0:
-            self.Parent.Hide()
-            if self.OnNew(None) is False:
-                return False
-            self.Parent.Show()
+        if len(self.choices) > 0:
+            return True
+        else:
+            return self.enterNewEntity()
+
+    def enterNewEntity(self):
+        self.Parent.Hide()
+        if self.OnNew(None) is False:
+            return False
+        self.Parent.Show()

        return True
+
+