Add context menu to change resist modes

gui/builtinContextMenus/__init__.py

@@ -12,6 +12,7 @@ from gui.builtinContextMenus import moduleAmmoChange
 from gui.builtinContextMenus import moduleSpool
 from gui.builtinContextMenus import boosterSideEffects
 from gui.builtinContextMenus import fighterAbilities
+from gui.builtinContextMenus import resistMode
 # Item info
 from gui.builtinContextMenus import itemStats
 from gui.builtinContextMenus import itemMarketJump

gui/builtinContextMenus/resistMode.py

@@ -0,0 +1,66 @@
+from collections import OrderedDict
+
+# noinspection PyPackageRequirements
+import wx
+
+import gui.mainFrame
+from graphs.events import ResistModeChanged
+from graphs.wrapper import TargetWrapper
+from gui.contextMenu import ContextMenuCombined
+from service.const import TargetResistMode
+
+
+optionMap = OrderedDict((
+    ('Auto', TargetResistMode.auto),
+    ('Shield', TargetResistMode.shield),
+    ('Armor', TargetResistMode.armor),
+    ('Hull', TargetResistMode.hull),
+    ('Weighed Average', TargetResistMode.weighedAverage)))
+
+
+class TargetWrapperResists(ContextMenuCombined):
+
+    def __init__(self):
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+
+    def display(self, callingWindow, srcContext, mainItem, selection):
+        if srcContext != 'graphTgtList':
+            return False
+        if not isinstance(mainItem, TargetWrapper) or not mainItem.isFit:
+            return False
+        self.callingWindow = callingWindow
+        self.selection = selection
+        return True
+
+    def getText(self, callingWindow, itmContext, mainItem, selection):
+        return 'Resist Mode'
+
+    def addOption(self, menu, optionLabel):
+        id = ContextMenuCombined.nextID()
+        self.optionIds[id] = optionLabel
+        menuItem = wx.MenuItem(menu, id, optionLabel, kind=wx.ITEM_CHECK)
+        menu.Bind(wx.EVT_MENU, self.handleMode, menuItem)
+        return menuItem
+
+    def getSubMenu(self, callingWindow, context, mainItem, selection, rootMenu, i, pitem):
+        msw = True if "wxMSW" in wx.PlatformInfo else False
+        self.optionIds = {}
+        sub = wx.Menu()
+        for optionLabel, optionValue in optionMap.items():
+            menuItem = self.addOption(rootMenu if msw else sub, optionLabel)
+            sub.Append(menuItem)
+            menuItem.Check(mainItem.resistMode == optionValue)
+        return sub
+
+    def handleMode(self, event):
+        optionLabel = self.optionIds[event.Id]
+        optionValue = optionMap[optionLabel]
+        changedFitIDs = set()
+        for wrapper in self.selection:
+            if wrapper.isFit and wrapper.resistMode != optionValue:
+                wrapper.resistMode = optionValue
+                changedFitIDs.add(wrapper.item.ID)
+        wx.PostEvent(gui.mainFrame.MainFrame.getInstance(), ResistModeChanged(fitIDs=changedFitIDs))
+
+
+TargetWrapperResists.register()