Fixed bugs introduced from code consistency fix

This commit is contained in:
blitzmann
2014-08-18 01:10:49 -04:00
parent 11891340af
commit 84ac71f528
2 changed files with 8 additions and 8 deletions

View File

@@ -18,10 +18,10 @@ class ItemRemove(ContextMenu):
srcContext = fullContext[0]
sFit = service.Fit.getInstance()
fitID = self.mainFrame.getActiveFit()
sFit = sFit.getFit(fitID)
fit = sFit.getFit(fitID)
idx = sFit.drones.index(selection[0])
sFit.removeDrone(fitID, idx, numDronesToRemove=sFit.drones[idx].amount)
idx = fit.drones.index(selection[0])
sFit.removeDrone(fitID, idx, numDronesToRemove=fit.drones[idx].amount)
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))

View File

@@ -19,20 +19,20 @@ class ItemRemove(ContextMenu):
srcContext = fullContext[0]
sFit = service.Fit.getInstance()
fitID = self.mainFrame.getActiveFit()
sFit = sFit.getFit(fitID)
fit = sFit.getFit(fitID)
if srcContext == "fittingModule":
for module in selection:
if module is not None:
sFit.removeModule(fitID,sFit.modules.index(module))
sFit.removeModule(fitID,fit.modules.index(module))
elif srcContext in ("fittingCharge" , "projectedCharge"):
sFit.setAmmo(fitID, None, selection)
elif srcContext == "droneItem":
sFit.removeDrone(fitID, sFit.drones.index(selection[0]))
sFit.removeDrone(fitID, fit.drones.index(selection[0]))
elif srcContext == "implantItem":
sFit.removeImplant(fitID, sFit.implants.index(selection[0]))
sFit.removeImplant(fitID, fit.implants.index(selection[0]))
elif srcContext == "boosterItem":
sFit.removeBooster(fitID, sFit.boosters.index(selection[0]))
sFit.removeBooster(fitID, fit.boosters.index(selection[0]))
else:
sFit.removeProjected(fitID, selection[0])