51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import wx
|
|
from logbook import Logger
|
|
|
|
import eos.db
|
|
from eos.exception import HandledListActionError
|
|
from service.fit import Fit
|
|
|
|
|
|
pyfalog = Logger(__name__)
|
|
|
|
|
|
class FitSwapModuleCommand(wx.Command):
|
|
|
|
def __init__(self, fitID, position1, position2):
|
|
wx.Command.__init__(self, True, 'Swap Modules')
|
|
self.fitID = fitID
|
|
self.position1 = position1
|
|
self.position2 = position2
|
|
|
|
def Do(self):
|
|
pyfalog.debug('Doing swapping between {} and {} for fit {}'.format(self.position1, self.position2, self.fitID))
|
|
self.__swap(self.fitID, self.position1, self.position2)
|
|
return True
|
|
|
|
def Undo(self):
|
|
self.__swap(self.fitID, self.position2, self.position1)
|
|
pyfalog.debug('Undoing swapping between {} and {} for fit {}'.format(self.position1, self.position2, self.fitID))
|
|
return True
|
|
|
|
def __swap(self, fitID, srcPosition, dstPosition):
|
|
fit = Fit.getInstance().getFit(fitID)
|
|
srcMod = fit.modules[srcPosition]
|
|
dstMod = fit.modules[dstPosition]
|
|
fit.modules.free(srcPosition)
|
|
fit.modules.free(dstPosition)
|
|
try:
|
|
fit.modules.replace(dstPosition, srcMod)
|
|
except HandledListActionError:
|
|
fit.modules.replace(srcPosition, srcMod)
|
|
fit.modules.replace(dstPosition, dstMod)
|
|
return False
|
|
try:
|
|
fit.modules.replace(srcPosition, dstMod)
|
|
except HandledListActionError:
|
|
fit.modules.free(dstPosition)
|
|
fit.modules.replace(srcPosition, srcMod)
|
|
fit.modules.replace(dstPosition, dstMod)
|
|
return False
|
|
eos.db.commit()
|
|
return True
|