71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
import wx
|
|
from service.fit import Fit
|
|
|
|
import gui.mainFrame
|
|
from gui import globalEvents as GE
|
|
from eos.saveddata.module import Module
|
|
from .calc.fitAddProjectedModule import FitAddProjectedModuleCommand
|
|
from .calc.fitAddProjectedEnv import FitAddProjectedEnvCommand
|
|
from .calc.fitAddProjectedFit import FitAddProjectedFitCommand
|
|
from logbook import Logger
|
|
import eos.db
|
|
pyfalog = Logger(__name__)
|
|
|
|
|
|
class GuiAddProjectedCommand(wx.Command):
|
|
def __init__(self, fitID, id, type='item'):
|
|
wx.Command.__init__(self, True, "Projected Add")
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
self.sFit = Fit.getInstance()
|
|
self.internal_history = wx.CommandProcessor()
|
|
self.fitID = fitID
|
|
self.id = id
|
|
self.type = type
|
|
|
|
def Do(self):
|
|
result = False
|
|
# since we can project various types, we need to switch of the fit command. We can't do this switch easily in
|
|
# the fit command since each type might have a different kind of undo, easier to split it out
|
|
if self.type == 'item':
|
|
item = eos.db.getItem(self.id, eager=("attributes", "group.category"))
|
|
|
|
if item.category.name == "Drone":
|
|
# @todo: this may need to be reworked once we visit drone commands
|
|
pyfalog.warn("DRONE PROJECTION NOT IMPLEMENTED")
|
|
# drone = None
|
|
# for d in fit.projectedDrones.find(item):
|
|
# if d is None or d.amountActive == d.amount or d.amount >= 5:
|
|
# drone = d
|
|
# break
|
|
#
|
|
# if drone is None:
|
|
# drone = Drone(item)
|
|
# fit.projectedDrones.append(drone)
|
|
#
|
|
# drone.amount += 1
|
|
elif item.category.name == "Fighter":
|
|
pyfalog.warn("FIGHTER PROJECTION NOT IMPLEMENTED")
|
|
# fighter = Fighter(item)
|
|
# fit.projectedFighters.append(fighter)
|
|
elif item.group.name in Module.SYSTEM_GROUPS:
|
|
result = self.internal_history.Submit(FitAddProjectedEnvCommand(self.fitID, self.id))
|
|
else:
|
|
# attempt a regular module projection
|
|
result = self.internal_history.Submit(FitAddProjectedModuleCommand(self.fitID, self.id))
|
|
elif self.type == 'fit':
|
|
result = self.internal_history.Submit(FitAddProjectedFitCommand(self.fitID, self.id))
|
|
|
|
if result:
|
|
self.sFit.recalc(self.fitID)
|
|
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
|
|
return True
|
|
return False
|
|
|
|
def Undo(self):
|
|
for _ in self.internal_history.Commands:
|
|
self.internal_history.Undo()
|
|
self.sFit.recalc(self.fitID)
|
|
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
|
|
return True
|
|
|