fb20847920
Added the ability to remove charges and projected items using the itemRemove built in context menu. In this commit I fixed a minor bug in the projected view that caused the right clicked item not to be SHOWN as selected until after the context menu had been spawned and closed. I also removed all parent references from the ContextMenu class and sub classes. I am instead using the fit service to do all the work of removing the items in the itemRemove class.
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from gui.contextMenu import ContextMenu
|
|
import gui.mainFrame
|
|
import service
|
|
import wx
|
|
import gui.globalEvents as GE
|
|
|
|
class AmmoPattern(ContextMenu):
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
|
|
def display(self, srcContext, selection):
|
|
if srcContext not in ("marketItemGroup", "marketItemMisc") or self.mainFrame.getActiveFit() is None:
|
|
return False
|
|
|
|
item = selection[0]
|
|
for attr in ("emDamage", "thermalDamage", "explosiveDamage", "kineticDamage"):
|
|
if item.getAttribute(attr) is not None:
|
|
return True
|
|
|
|
return False
|
|
|
|
def getText(self, itmContext, selection):
|
|
return "Set {0} as Damage Pattern".format(itmContext if itmContext is not None else "Item")
|
|
|
|
def activate(self, fullContext, selection, i):
|
|
item = selection[0]
|
|
fit = self.mainFrame.getActiveFit()
|
|
sFit = service.Fit.getInstance()
|
|
sFit.setAsPattern(fit, item)
|
|
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fit))
|
|
|
|
def getBitmap(self, context, selection):
|
|
return None
|
|
|
|
|
|
AmmoPattern.register()
|