Fix for #1090 - using variations on implants is not recalcing fit correctly. (#1094)

This commit is contained in:
Ryan Holmes
2017-04-02 01:04:34 -05:00
committed by GitHub
parent 112ab91e34
commit 216dac068d
2 changed files with 5 additions and 4 deletions

View File

@@ -179,8 +179,8 @@ class MetaSwap(ContextMenu):
elif isinstance(selected_item, Implant):
for idx, implant_stack in enumerate(fit.implants):
if implant_stack is selected_item:
sFit.removeImplant(fitID, idx)
sFit.addImplant(fitID, item.ID, False)
sFit.removeImplant(fitID, idx, False)
sFit.addImplant(fitID, item.ID, True)
break
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))

View File

@@ -254,14 +254,15 @@ class Fit(object):
self.recalc(fit)
return True
def removeImplant(self, fitID, position):
def removeImplant(self, fitID, position, recalc=True):
if fitID is None:
return False
fit = eos.db.getFit(fitID)
implant = fit.implants[position]
fit.implants.remove(implant)
self.recalc(fit)
if recalc:
self.recalc(fit)
return True
def addBooster(self, fitID, itemID):