135 lines
5.1 KiB
Python
135 lines
5.1 KiB
Python
import math
|
|
|
|
import wx
|
|
|
|
import gui.fitCommands as cmd
|
|
import gui.mainFrame
|
|
from gui.contextMenu import ContextMenuCombined
|
|
from gui.fitCommands.helpers import getSimilarModPositions, getSimilarFighters
|
|
from service.fit import Fit
|
|
from service.settings import ContextMenuSettings
|
|
|
|
|
|
class RemoveItem(ContextMenuCombined):
|
|
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
self.settings = ContextMenuSettings.getInstance()
|
|
|
|
def display(self, srcContext, mainItem, selection):
|
|
if srcContext not in (
|
|
"fittingModule", "droneItem",
|
|
"implantItem", "boosterItem",
|
|
"projectedModule", "cargoItem",
|
|
"projectedFit", "projectedDrone",
|
|
"fighterItem", "projectedFighter",
|
|
"commandFit"
|
|
):
|
|
return False
|
|
|
|
if mainItem is None or getattr(mainItem, "isEmpty", False):
|
|
return False
|
|
|
|
return True
|
|
|
|
def getText(self, itmContext, mainItem, selection):
|
|
return 'Remove {}{}'.format(
|
|
itmContext if itmContext is not None else 'Item',
|
|
' Stack' if itmContext in ('Drone', 'Fit') else '')
|
|
|
|
def activate(self, fullContext, mainItem, selection, i):
|
|
handlerMap = {
|
|
'fittingModule': self.__handleModule,
|
|
'droneItem': self.__handleDrone,
|
|
'fighterItem': self.__handleFighter,
|
|
'implantItem': self.__handleImplant,
|
|
'boosterItem': self.__handleBooster,
|
|
'cargoItem': self.__handleCargo,
|
|
'projectedFit': self.__handleProjectedItem,
|
|
'projectedModule': self.__handleProjectedItem,
|
|
'projectedDrone': self.__handleProjectedItem,
|
|
'projectedFighter': self.__handleProjectedItem,
|
|
'commandFit': self.__handleCommandFit}
|
|
srcContext = fullContext[0]
|
|
handler = handlerMap.get(srcContext)
|
|
if handler is None:
|
|
return
|
|
handler(mainItem, selection)
|
|
|
|
def __handleModule(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
fit = Fit.getInstance().getFit(fitID)
|
|
if wx.GetMouseState().GetModifiers() == wx.MOD_ALT:
|
|
positions = getSimilarModPositions(fit.modules, mainItem)
|
|
else:
|
|
positions = []
|
|
for mod in selection:
|
|
if mod in fit.modules:
|
|
positions.append(fit.modules.index(mod))
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveLocalModuleCommand(
|
|
fitID=fitID, positions=positions))
|
|
|
|
def __handleDrone(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
fit = Fit.getInstance().getFit(fitID)
|
|
positions = []
|
|
for drone in selection:
|
|
if drone in fit.drones:
|
|
positions.append(fit.drones.index(drone))
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveLocalDronesCommand(
|
|
fitID=fitID, positions=positions, amount=math.inf))
|
|
|
|
def __handleFighter(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
fit = Fit.getInstance().getFit(fitID)
|
|
if wx.GetMouseState().GetModifiers() == wx.MOD_ALT:
|
|
fighters = getSimilarFighters(fit.fighters, mainItem)
|
|
else:
|
|
fighters = selection
|
|
positions = []
|
|
for fighter in fighters:
|
|
if fighter in fit.fighters:
|
|
positions.append(fit.fighters.index(fighter))
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveLocalFightersCommand(
|
|
fitID=fitID, positions=positions))
|
|
|
|
def __handleImplant(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
fit = Fit.getInstance().getFit(fitID)
|
|
positions = []
|
|
for implant in selection:
|
|
if implant in fit.implants:
|
|
positions.append(fit.implants.index(implant))
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveImplantsCommand(
|
|
fitID=fitID, positions=positions))
|
|
|
|
def __handleBooster(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
fit = Fit.getInstance().getFit(fitID)
|
|
positions = []
|
|
for booster in selection:
|
|
if booster in fit.boosters:
|
|
positions.append(fit.boosters.index(booster))
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveBoostersCommand(
|
|
fitID=fitID, positions=positions))
|
|
|
|
def __handleCargo(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
itemIDs = [c.itemID for c in selection]
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveCargosCommand(
|
|
fitID=fitID, itemIDs=itemIDs))
|
|
|
|
def __handleProjectedItem(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveProjectedItemsCommand(
|
|
fitID=fitID, items=selection, amount=math.inf))
|
|
|
|
def __handleCommandFit(self, mainItem, selection):
|
|
fitID = self.mainFrame.getActiveFit()
|
|
commandFitIDs = [cf.ID for cf in selection]
|
|
self.mainFrame.command.Submit(cmd.GuiRemoveCommandFitsCommand(
|
|
fitID=fitID, commandFitIDs=commandFitIDs))
|
|
|
|
|
|
RemoveItem.register()
|