Allow removal of several implants at once via context menu

This commit is contained in:
DarkPhoenix
2019-04-25 17:15:01 +03:00
parent 5777103d21
commit e09fce8411
2 changed files with 22 additions and 28 deletions

View File

@@ -96,10 +96,12 @@ class RemoveItem(ContextMenuCombined):
def __handleImplant(self, mainItem, selection):
fitID = self.mainFrame.getActiveFit()
fit = Fit.getInstance().getFit(fitID)
if mainItem in fit.implants:
position = fit.implants.index(mainItem)
self.mainFrame.command.Submit(cmd.GuiRemoveImplantsCommand(
fitID=fitID, positions=[position]))
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()