localRemove.py

import wx
from logbook import Logger

from gui.fitCommands.helpers import DroneInfo
from service.fit import Fit


pyfalog = Logger(__name__)


class CalcRemoveLocalDroneCommand(wx.Command):

    def __init__(self, fitID, position, amount, deactivate=False):
        wx.Command.__init__(self, True, 'Remove Local Drone')
        self.fitID = fitID
        self.position = position
        self.amountToRemove = amount
        self.savedDroneInfo = None
        self.removedStack = None
        self.deactivate = deactivate

    def Do(self):
        pyfalog.debug('Doing removal of {} local drones at position {} from fit {}'.format(self.amountToRemove, self.position, self.fitID))
        fit = Fit.getInstance().getFit(self.fitID)
        drone = fit.drones[self.position]
        self.savedDroneInfo = DroneInfo.fromDrone(drone)

        drone.amount = max(drone.amount - self.amountToRemove, 0)
        if drone.amountActive > 0:
            if self.deactivate:
                drone.amountActive = 0
            else:
                drone.amountActive = min(drone.amountActive, drone.amount)

        if drone.amount == 0:
            fit.drones.remove(drone)
            self.removedStack = True
        else:
            self.removedStack = False

        return True

    def Undo(self):
        pyfalog.debug('Undoing removal of {} local drones at position {} from fit {}'.format(self.amountToRemove, self.position, self.fitID))
        fit = Fit.getInstance().getFit(self.fitID)
        if self.removedStack:
            drone = self.savedDroneInfo.toDrone()
            if drone is None:
                return False
            fit.drones.insert(self.position, drone)
            if drone not in fit.drones:
                pyfalog.warning('Failed to insert to list')
                return False
        else:
            drone = fit.drones[self.position]
            drone.amount = self.savedDroneInfo.amount
            drone.amountActive = self.savedDroneInfo.amountActive
        return True