38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import wx
|
|
from logbook import Logger
|
|
|
|
import eos.db
|
|
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, src, dst):
|
|
fit = Fit.getInstance().getFit(fitID)
|
|
srcMod = fit.modules[src]
|
|
dstMod = fit.modules[dst]
|
|
fit.modules.remove(srcMod)
|
|
fit.modules.insert(dst, srcMod)
|
|
fit.modules.remove(dstMod)
|
|
fit.modules.insert(src, dstMod)
|
|
eos.db.commit()
|