Add .Check() to tactical mode menu items

This commit is contained in:
blitzmann
2014-11-30 16:56:43 -05:00
parent 23a71e7448
commit edbdaf5631

View File

@@ -11,13 +11,14 @@ class TacticalMode(ContextMenu):
def display(self, srcContext, selection):
sFit = service.Fit.getInstance()
fitID = self.mainFrame.getActiveFit()
ship = sFit.getFit(fitID).ship
self.modes = ship.getModes()
fit = sFit.getFit(fitID)
self.modes = fit.ship.getModes()
self.currMode = fit.mode
return srcContext == "fittingShip" and self.modes is not None
def getText(self, itmContext, selection):
return "Tactical Modes"
return "Tactical Mode"
def addMode(self, rootMenu, mode):
label = mode.item.name.rsplit()[-2]
@@ -33,9 +34,10 @@ class TacticalMode(ContextMenu):
sub = wx.Menu()
for item in self.modes:
menuItem = self.addMode(rootMenu, item)
for mode in self.modes:
menuItem = self.addMode(rootMenu, mode)
sub.AppendItem(menuItem)
menuItem.Check(self.currMode.item == mode.item)
return sub