dave/pyfa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 12 Wiki Activity

Rework projected fighter fit commands

Browse Source
This commit is contained in:
DarkPhoenix
2019-04-13 18:26:05 +03:00
parent 7f870cf675
commit 1ca4c73a3e
12 changed files with 94 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
gui/fitCommands/calc/fitAddProjectedModule.py
View File

@@ -12,21 +12,21 @@ pyfalog = Logger(__name__)
class FitAddProjectedModuleCommand(wx.Command):
def __init__(self, fitID, newModInfo, newPosition=None):
def __init__(self, fitID, modInfo, position=None):
wx.Command.__init__(self, True)
self.fitID = fitID
self.newModInfo = newModInfo
self.newPosition = newPosition
self.newModInfo = modInfo
self.newPosition = position
self.oldModInfo = None
self.oldPosition = None
def Do(self):
pyfalog.debug('Doing addition of projected module {} onto: {}'.format(self.newModInfo, self.fitID))
fit = Fit.getInstance().getFit(self.fitID)
newMod = self.newModInfo.toModule(fallbackState=FittingModuleState.ACTIVE)
if newMod is None:
return False
fit = Fit.getInstance().getFit(self.fitID)
if not newMod.canHaveState(newMod.state, projectedOnto=fit):
newMod.state = FittingModuleState.OFFLINE
@@ -54,8 +54,8 @@ class FitAddProjectedModuleCommand(wx.Command):
if self.oldPosition is not None and self.oldModInfo is not None:
cmd = FitAddProjectedModuleCommand(
fitID=self.fitID,
newModInfo=self.oldModInfo,
newPosition=self.oldPosition)
modInfo=self.oldModInfo,
position=self.oldPosition)
return cmd.Do()
if self.newPosition is None:
return False