From 1eba77936d7046c33c067c0e429ecee5fb403c33 Mon Sep 17 00:00:00 2001 From: cncfanatics Date: Thu, 28 Oct 2010 18:06:01 +0200 Subject: [PATCH] Actualy add the new ammoPattern file --- gui/builtinContextMenus/ammoPattern.py | 45 ++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100755 gui/builtinContextMenus/ammoPattern.py diff --git a/gui/builtinContextMenus/ammoPattern.py b/gui/builtinContextMenus/ammoPattern.py new file mode 100755 index 000000000..fcbcabf4b --- /dev/null +++ b/gui/builtinContextMenus/ammoPattern.py @@ -0,0 +1,45 @@ +from gui.contextMenu import ContextMenu +import gui.mainFrame +import service +import gui.fittingView +import wx +import gui.fittingView +from gui import bitmapLoader + +class AmmoPattern(ContextMenu): + def __init__(self): + self.mainFrame = gui.mainFrame.MainFrame.getInstance() + + + def display(self, context, selection): + if context not in ("item", "itemSearch") or self.mainFrame.getActiveFit() is None: + return False + + item = selection[0] + if context == "itemSearch": + item = service.Market.getInstance().getItem(item.ID) + + for attr in ("emDamage", "thermalDamage", "explosiveDamage", "kineticDamage"): + if item.getAttribute(attr) is not None: + return True + + return False + + def getText(self, context, selection): + return "Set as damage pattern" + + def activate(self, context, selection, i): + item = selection[0] + if context == "itemSearch": + item = service.Market.getInstance().getItem(item.ID) + + fit = self.mainFrame.getActiveFit() + sFit = service.Fit.getInstance() + sFit.setAsPattern(fit, item) + wx.PostEvent(self.mainFrame, gui.fittingView.FitChanged(fitID=fit)) + + def getBitmap(self, context, selection): + return None + + +AmmoPattern.register()