Do not crash when user cancels setting of projection range

This commit is contained in:
DarkPhoenix
2019-10-09 13:30:01 +03:00
parent 8d81db0a3a
commit 2aa96fc819

View File

@@ -52,19 +52,18 @@ class ChangeItemProjectionRange(ContextMenuCombined):
newRange = cleanInputFloat * 1000 newRange = cleanInputFloat * 1000
else: else:
newRange = None newRange = None
fitID = self.mainFrame.getActiveFit()
fitID = self.mainFrame.getActiveFit() items = selection
items = selection if wx.GetMouseState().GetModifiers() in (wx.MOD_ALT, wx.MOD_CONTROL):
if wx.GetMouseState().GetModifiers() in (wx.MOD_ALT, wx.MOD_CONTROL): if isinstance(mainItem, EosModule):
if isinstance(mainItem, EosModule): fit = Fit.getInstance().getFit(fitID)
fit = Fit.getInstance().getFit(fitID) positions = getSimilarModPositions(fit.projectedModules, mainItem)
positions = getSimilarModPositions(fit.projectedModules, mainItem) items = [fit.projectedModules[p] for p in positions]
items = [fit.projectedModules[p] for p in positions] elif isinstance(mainItem, EosFighter):
elif isinstance(mainItem, EosFighter): fit = Fit.getInstance().getFit(fitID)
fit = Fit.getInstance().getFit(fitID) items = getSimilarFighters(fit.projectedFighters, mainItem)
items = getSimilarFighters(fit.projectedFighters, mainItem) self.mainFrame.command.Submit(cmd.GuiChangeProjectedItemsProjectionRangeCommand(
self.mainFrame.command.Submit(cmd.GuiChangeProjectedItemsProjectionRangeCommand( fitID=fitID, items=items, projectionRange=newRange))
fitID=fitID, items=items, projectionRange=newRange))
ChangeItemProjectionRange.register() ChangeItemProjectionRange.register()