Undoing removal of drones now restores their proper state

2019-04-11 18:20:56 +03:00
parent e90e9a5ca1
commit 5579929f83
6 changed files with 65 additions and 42 deletions
--- a/gui/fitCommands/calc/fitAddProjectedModule.py
+++ b/gui/fitCommands/calc/fitAddProjectedModule.py
@@ -43,9 +43,11 @@ class FitAddProjectedModuleCommand(wx.Command):
            self.oldModuleInfo = ModuleInfoCache(oldPosition, oldItemID, oldState, oldChargeID, oldBaseItemID, oldMutaplasmidID, oldMutations)

        if self.newPosition is not None:
-            fit.projectedModules.insert(self.newPosition, mod)
+            if not fit.projectedModules.insert(self.newPosition, mod):
+                return False
        else:
-            fit.projectedModules.append(mod)
+            if not fit.projectedModules.append(mod):
+                return False
            self.newPosition = fit.projectedModules.index(mod)

        eos.db.commit()