Move more logic to base class

Author: DarkFenX

_development/helpers_fits.py

@@ -62,4 +62,4 @@ def HeronFit(DB, Gamedata, Saveddata):
     for _ in range(4):
         fit.modules.append(mod)
 
-    return fit
+    return fit

eos/saveddata/targetProfile.py

@@ -58,6 +58,7 @@ def getIdeal(cls):
                 signatureRadius=None,
                 radius=0)
             cls._idealTarget.name = 'Ideal Target'
+            cls._idealTarget.ID = -1
         return cls._idealTarget
 
     @property

gui/builtinContextMenus/itemRemove.py

@@ -159,7 +159,7 @@ def __handleCommandFit(self, callingWindow, mainItem, selection):
             fitID=fitID, commandFitIDs=commandFitIDs))
 
     def __handleGraphFit(self, callingWindow, mainItem, selection):
-        callingWindow.removeFits(selection)
+        callingWindow.removeListItems(selection)
 
 
 RemoveItem.register()

gui/graphFrame/lists.py

@@ -40,6 +40,9 @@ def __init__(self, graphFrame, parent):
         super().__init__(parent)
         self.graphFrame = graphFrame
 
+        self.Bind(wx.EVT_CHAR_HOOK, self.kbEvent)
+        self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick)
+
     def refreshExtraColumns(self, extraColSpecs):
         baseColNames = set()
         for baseColName in self.DEFAULT_COLS:
@@ -56,19 +59,38 @@ def refreshExtraColumns(self, extraColSpecs):
             self.appendColumnBySpec(colSpec)
         self.refreshView()
 
+    def kbEvent(self, event):
+        keycode = event.GetKeyCode()
+        mstate = wx.GetMouseState()
+        if keycode == 65 and mstate.GetModifiers() == wx.MOD_CONTROL:
+            self.selectAll()
+        elif keycode in (wx.WXK_DELETE, wx.WXK_NUMPAD_DELETE) and mstate.GetModifiers() == wx.MOD_NONE:
+            self.removeListItems(self.getSelectedListItems())
+        event.Skip()
+
+    def OnLeftDClick(self, event):
+        row, _ = self.HitTest(event.Position)
+        item = self.getListItem(row)
+        if item is None:
+            return
+        self.removeListItems([item])
+
     def refreshView(self):
         raise NotImplementedError
 
     def updateView(self):
         raise NotImplementedError
 
-    def getItem(self, row):
+    def getListItem(self, row):
         raise NotImplementedError
 
-    def getSelectedItems(self):
+    def removeListItems(self, items):
+        raise NotImplementedError
+
+    def getSelectedListItems(self):
         items = []
         for row in self.getSelectedRows():
-            item = self.getItem(row)
+            item = self.getListItem(row)
             if item is None:
                 continue
             items.append(item)
@@ -87,8 +109,6 @@ def __init__(self, graphFrame, parent):
         self.graphFrame.mainFrame.Bind(EVT_FIT_RENAMED, self.OnFitRenamed)
         self.graphFrame.mainFrame.Bind(GE.FIT_REMOVED, self.OnFitRemoved)
 
-        self.Bind(wx.EVT_CHAR_HOOK, self.kbEvent)
-        self.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick)
         self.Bind(wx.EVT_CONTEXT_MENU, self.spawnMenu)
         self.Bind(wx.EVT_MOTION, self.OnMouseMove)
         self.Bind(wx.EVT_LEAVE_WINDOW, self.OnLeaveWindow)
@@ -110,35 +130,16 @@ def updateView(self):
         self.update(self.fits)
 
     def spawnMenu(self, event):
-        selection = self.getSelectedItems()
+        selection = self.getSelectedListItems()
         clickedPos = self.getRowByAbs(event.Position)
-        mainItem = self.getItem(clickedPos)
+        mainItem = self.getListItem(clickedPos)
 
         sourceContext = 'graphFitList'
         itemContext = None if mainItem is None else 'Fit'
         menu = ContextMenu.getMenu(self, mainItem, selection, (sourceContext, itemContext))
         if menu:
             self.PopupMenu(menu)
 
-    def kbEvent(self, event):
-        keycode = event.GetKeyCode()
-        mstate = wx.GetMouseState()
-        if keycode == 65 and mstate.GetModifiers() == wx.MOD_CONTROL:
-            self.selectAll()
-        elif keycode in (wx.WXK_DELETE, wx.WXK_NUMPAD_DELETE) and mstate.GetModifiers() == wx.MOD_NONE:
-            self.removeFits(self.getSelectedItems())
-        event.Skip()
-
-    def OnLeftDClick(self, event):
-        row, _ = self.HitTest(event.Position)
-        if row != -1:
-            try:
-                fit = self.fits[row]
-            except IndexError:
-                pass
-            else:
-                self.removeFits([fit])
-
     def OnFitRenamed(self, event):
         event.Skip()
         self.updateView()
@@ -152,24 +153,24 @@ def OnFitRemoved(self, event):
         event.Skip()
         fit = next((f for f in self.fits if f.ID == event.fitID), None)
         if fit is not None:
-            self.removeFits([fit])
+            self.removeListItems([fit])
 
-    def getItem(self, row):
+    def getListItem(self, row):
         if row == -1:
             return None
         try:
             return self.fits[row]
         except IndexError:
             return None
 
-    def removeFits(self, fits):
-        toRemove = [f for f in fits if f in self.fits]
+    def removeListItems(self, items):
+        toRemove = [i for i in items if i in self.fits]
         if not toRemove:
             return
         for fit in toRemove:
             self.fits.remove(fit)
         self.updateView()
-        for fit in fits:
+        for fit in toRemove:
             self.graphFrame.clearCache(reason=GraphCacheCleanupReason.fitRemoved, extraData=fit.ID)
         self.graphFrame.draw()
 
@@ -231,7 +232,7 @@ def refreshView(self):
     def updateView(self):
         self.update(self.targets)
 
-    def getItem(self, row):
+    def getListItem(self, row):
         if row == -1:
             return None
 
@@ -246,6 +247,22 @@ def getItem(self, row):
         else:
             return self.profiles[row - numFits]
 
+    def removeListItems(self, items):
+        fitsToRemove = [i for i in items if i in self.fits]
+        profilesToRemove = [i for i in items if i in self.profiles]
+        if not fitsToRemove and not profilesToRemove:
+            return
+        for fit in fitsToRemove:
+            self.fits.remove(fit)
+        for profile in profilesToRemove:
+            self.profiles.remove(profile)
+        self.updateView()
+        for fit in fitsToRemove:
+            self.graphFrame.clearCache(reason=GraphCacheCleanupReason.fitRemoved, extraData=fit.ID)
+        for profile in profilesToRemove:
+            self.graphFrame.clearCache(reason=GraphCacheCleanupReason.profileRemoved, extraData=profile.ID)
+        self.graphFrame.draw()
+
     @property
     def targets(self):
         return self.fits + self.profiles

service/const.py

@@ -123,6 +123,8 @@ class GraphDpsDroneMode(IntEnum):
 class GraphCacheCleanupReason(IntEnum):
     fitChanged = auto()
     fitRemoved = auto()
+    profileChanged = auto()
+    profileRemoved = auto()
     graphSwitched = auto()
     inputChanged = auto()
     optionChanged = auto()