move some files around and create a context manager for fit recalclation

gui/fitCommands/calc/fitRemoveModule.py

@@ -0,0 +1,53 @@
+import wx
+
+from gui.fitCommands.helpers import ModuleInfoCache
+import eos.db
+from logbook import Logger
+pyfalog = Logger(__name__)
+
+
+class FitRemoveModuleCommand(wx.Command):
+    """"
+    Fitting command that removes a module at a specified positions
+
+    from sFit.removeModule
+    """
+    def __init__(self, fitID: int, positions: list = None):
+        wx.Command.__init__(self, True, "Module Remove")
+        self.fitID = fitID
+        self.positions = positions
+        self.modCache = []
+        self.change = None
+
+    def Do(self):
+        fitID = self.fitID
+        pyfalog.debug("Removing module from position ({0}) for fit ID: {1}", self.positions, fitID)
+        fit = eos.db.getFit(fitID)
+
+        for x in self.positions:
+            mod = fit.modules[x]
+            if not mod.isEmpty:
+                self.modCache.append(ModuleInfoCache(mod.modPosition, mod.item.ID, mod.state, mod.charge))
+                fit.modules.toDummy(x)
+
+        # if no modules have changes, report back None
+        if not len(self.modCache) > 0:
+            return False
+
+        numSlots = len(fit.modules)
+        # todo: determine if we need to do this still
+        # self.recalc(fit)
+        # self.checkStates(fit, None)
+        fit.fill()
+        eos.db.commit()
+        self.slotsChanged = numSlots != len(fit.modules)
+        return True
+
+    def Undo(self):
+        from gui.fitCommands.calc.fitAddModule import FitAddModuleCommand  # avoids circular import
+        for mod in self.modCache:
+            cmd = FitAddModuleCommand(self.fitID, mod.itemID)
+            cmd.Do()
+            cmd.module.state = mod.state
+            cmd.module.charge = mod.charge
+        return True