Integrate price optimization with undo/redo

gui/fitCommands/__init__.py

@@ -35,3 +35,4 @@ from .guiToggleDrone import GuiToggleDroneCommand
 from .guiFitRename import GuiFitRenameCommand
 from .guiChangeImplantLocation import GuiChangeImplantLocation
 from .guiImportMutatedModule import GuiImportMutatedModuleCommand
+from .guiRebaseItems import GuiRebaseItemsCommand

gui/fitCommands/calc/fitRebaseItem.py

@@ -0,0 +1,31 @@
+import wx
+
+from logbook import Logger
+
+import eos.db
+
+
+pyfalog = Logger(__name__)
+
+
+class FitRebaseItemCommand(wx.Command):
+
+    def __init__(self, fitID, containerName, position, newTypeID):
+        wx.Command.__init__(self, True, "Rebase Item")
+        self.fitID = fitID
+        self.containerName = containerName
+        self.position = position
+        self.newTypeID = newTypeID
+        self.oldTypeID = None
+
+    def Do(self):
+        fit = eos.db.getFit(self.fitID)
+        obj = getattr(fit, self.containerName)[self.position]
+        self.oldTypeID = getattr(obj.item, "ID", None)
+        newItem = eos.db.getItem(self.newTypeID)
+        obj.rebase(newItem)
+        return True
+
+    def Undo(self):
+        cmd = FitRebaseItemCommand(self.fitID, self.containerName, self.position, self.oldTypeID)
+        return cmd.Do()

gui/fitCommands/guiOptimizePrice.py

@@ -1,12 +0,0 @@
-import wx
-
-
-class GuiOptimizePriceCommand(wx.Command):
-    def __init__(self):
-        wx.Command.__init__(self, True, "Optimize Price")
-
-    def Do(self):
-        pass
-
-    def Undo(self):
-        pass

gui/fitCommands/guiRebaseItems.py

@@ -0,0 +1,46 @@
+import wx
+
+import eos.db
+import gui.mainFrame
+from gui import globalEvents as GE
+from service.fit import Fit
+from .calc.fitRebaseItem import FitRebaseItemCommand
+from .calc.fitSetCharge import FitSetChargeCommand
+
+
+class GuiRebaseItemsCommand(wx.Command):
+
+    def __init__(self, fitID, rebaseMap):
+        wx.Command.__init__(self, True, "Mass Rebase Item")
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.fitID = fitID
+        self.rebaseMap = rebaseMap
+        self.internal_history = wx.CommandProcessor()
+
+    def Do(self):
+        fit = eos.db.getFit(self.fitID)
+        for mod in fit.modules:
+            if mod.item is not None and mod.item.ID in self.rebaseMap:
+                self.internal_history.Submit(FitRebaseItemCommand(self.fitID, "modules", mod.modPosition, self.rebaseMap[mod.item.ID]))
+            if mod.charge is not None and mod.charge.ID in self.rebaseMap:
+                self.internal_history.Submit(FitSetChargeCommand(self.fitID, [mod.modPosition], self.rebaseMap[mod.charge.ID]))
+        for containerName in ("drones", "fighters", "implants", "boosters", "cargo"):
+            container = getattr(fit, containerName)
+            for obj in container:
+                if obj.item is not None and obj.item.ID in self.rebaseMap:
+                    self.internal_history.Submit(FitRebaseItemCommand(self.fitID, containerName, container.index(obj), self.rebaseMap[obj.item.ID]))
+        if self.internal_history.Commands:
+            eos.db.commit()
+            Fit.getInstance().recalc(self.fitID)
+            wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+            return True
+        else:
+            return False
+
+    def Undo(self):
+        for _ in self.internal_history.Commands:
+            self.internal_history.Undo()
+        eos.db.commit()
+        Fit.getInstance().recalc(self.fitID)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+        return True