Continue work on projections (adding/removing projected environment and fit)

gui/fitCommands/calc/fitRemoveProjectedFit.py

@@ -0,0 +1,46 @@
+import wx
+from service.fit import Fit
+
+import gui.mainFrame
+from gui import globalEvents as GE
+#from .helpers import ModuleInfoCache
+from eos.saveddata.module import Module, State
+import eos.db
+from logbook import Logger
+from eos.saveddata.module import Module
+from eos.saveddata.drone import Drone
+from eos.saveddata.fighter import Fighter
+from .fitRemoveProjectedModule import FitRemoveProjectedModuleCommand
+pyfalog = Logger(__name__)
+
+
+# this has the same exact definition that regular rpojected modules, besides the undo
+class FitRemoveProjectedFitCommand(FitRemoveProjectedModuleCommand):
+    """"
+    from sFit.project
+    """
+
+    def __init__(self, fitID, projectedFitID):
+        wx.Command.__init__(self, True)
+        self.fitID = fitID
+        self.projectedFitID = projectedFitID
+
+    def Do(self):
+        pyfalog.debug("Removing ({0}) onto: {1}", self.fitID, self.projectedFitID)
+        fit = eos.db.getFit(self.fitID)
+        projectedFit = eos.db.getFit(self.projectedFitID)
+
+        if projectedFit is None:
+            return False
+
+        del fit.projectedFitDict[projectedFit.ID]
+
+        eos.db.commit()
+        return True
+
+    def Undo(self):
+        # todo: figure out if I need to return false here if the fit doesn't return true (means it was deleted)
+        from gui.fitCommands.calc.fitAddProjectedFit import FitAddProjectedFitCommand
+        cmd = FitAddProjectedFitCommand(self.fitID, self.projectedFitID)
+        cmd.Do()
+        return True