More work on the spoolup gui, including support for commands and some other nicities (recalc fit when changing, load the input with previous)

gui/fitCommands/__init__.py

@@ -34,3 +34,4 @@ from .guiToggleDrone import GuiToggleDroneCommand
 from .guiFitRename import GuiFitRenameCommand
 from .guiChangeImplantLocation import GuiChangeImplantLocation
 from .guiImportMutatedModule import GuiImportMutatedModuleCommand
+from .guiSetSpoolup import GuiSetSpoolup

gui/fitCommands/calc/fitSetSpoolup.py

@@ -0,0 +1,37 @@
+import wx
+import eos.db
+from logbook import Logger
+from eos.saveddata.booster import Booster
+pyfalog = Logger(__name__)
+
+
+class FitSetSpoolupCommand(wx.Command):
+    def __init__(self, fitID, position, spoolType, spoolAmount):
+        wx.Command.__init__(self, True)
+        self.fitID = fitID
+        self.position = position
+        self.spoolType = spoolType
+        self.spoolAmount = spoolAmount
+        self.projected = False # todo: get this to work with projected modules? Is that a thing?
+        self.cache = None
+
+    def Do(self):
+        return self.__set(self.spoolType, self.spoolAmount)
+
+    def Undo(self):
+        if self.cache:
+            self.__set(*self.cache)
+        return True
+
+    def __set(self, type, amount):
+        fit = eos.db.getFit(self.fitID)
+        source = fit.modules if not self.projected else fit.projectedModules
+
+        mod = source[self.position]
+        self.cache = mod.spoolType, mod.spoolAmount
+
+        mod.spoolType = type
+        mod.spoolAmount = amount
+
+        eos.db.commit()
+        return True

gui/fitCommands/guiSetSpoolup.py

@@ -0,0 +1,32 @@
+import wx
+from service.fit import Fit
+
+import gui.mainFrame
+from gui import globalEvents as GE
+from .calc.fitSetSpoolup import FitSetSpoolupCommand
+
+
+class GuiSetSpoolup(wx.Command):
+    def __init__(self, fitID, module, spoolupType, spoolupAmount):
+        wx.Command.__init__(self, True, "Booster Add")
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.sFit = Fit.getInstance()
+        self.internal_history = wx.CommandProcessor()
+        self.fitID = fitID
+        self.position = module.modPosition
+        self.spoolType = spoolupType
+        self.spoolupAmount = spoolupAmount
+
+    def Do(self):
+        if self.internal_history.Submit(FitSetSpoolupCommand(self.fitID, self.position, self.spoolType, self.spoolupAmount)):
+            self.sFit.recalc(self.fitID)
+            wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+            return True
+        return False
+
+    def Undo(self):
+        for _ in self.internal_history.Commands:
+            self.internal_history.Undo()
+        self.sFit.recalc(self.fitID)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+        return True