Implement drone mutation support with some exceptions
This commit is contained in:
DarkPhoenix
@@ -28,6 +28,8 @@ from .gui.localDrone.changeAmount import GuiChangeLocalDroneAmountCommand
|
||||
from .gui.localDrone.changeMetas import GuiChangeLocalDroneMetasCommand
|
||||
from .gui.localDrone.clone import GuiCloneLocalDroneCommand
|
||||
from .gui.localDrone.imprt import GuiImportLocalDronesCommand
|
||||
from .gui.localDrone.mutatedConvert import GuiConvertMutatedLocalDroneCommand
|
||||
from .gui.localDrone.mutatedRevert import GuiRevertMutatedLocalDroneCommand
|
||||
from .gui.localDrone.remove import GuiRemoveLocalDronesCommand
|
||||
from .gui.localDrone.stackSplit import GuiSplitLocalDroneStackCommand
|
||||
from .gui.localDrone.stacksMerge import GuiMergeLocalDroneStacksCommand
|
||||
|
||||
65
gui/fitCommands/gui/localDrone/mutatedConvert.py
Normal file
65
gui/fitCommands/gui/localDrone/mutatedConvert.py
Normal file
@@ -0,0 +1,65 @@
|
||||
import math
|
||||
|
||||
import wx
|
||||
|
||||
import eos.db
|
||||
import gui.mainFrame
|
||||
from gui import globalEvents as GE
|
||||
from gui.fitCommands.calc.drone.localAdd import CalcAddLocalDroneCommand
|
||||
from gui.fitCommands.calc.drone.localRemove import CalcRemoveLocalDroneCommand
|
||||
from gui.fitCommands.helpers import DroneInfo, InternalCommandHistory
|
||||
from service.fit import Fit
|
||||
|
||||
|
||||
class GuiConvertMutatedLocalDroneCommand(wx.Command):
|
||||
|
||||
def __init__(self, fitID, position, mutaplasmid):
|
||||
wx.Command.__init__(self, True, 'Convert Local Drone to Mutated')
|
||||
self.internalHistory = InternalCommandHistory()
|
||||
self.fitID = fitID
|
||||
self.position = position
|
||||
self.itemID = mutaplasmid.resultingItem.ID
|
||||
self.mutaplasmidID = mutaplasmid.ID
|
||||
|
||||
def Do(self):
|
||||
sFit = Fit.getInstance()
|
||||
fit = sFit.getFit(self.fitID)
|
||||
try:
|
||||
drone = fit.drones[self.position]
|
||||
except IndexError:
|
||||
return False
|
||||
if drone.isMutated:
|
||||
return False
|
||||
info = DroneInfo(
|
||||
amount=drone.amount,
|
||||
amountActive=drone.amountActive,
|
||||
itemID=self.itemID,
|
||||
baseItemID=drone.item.ID,
|
||||
mutaplasmidID=self.mutaplasmidID,
|
||||
mutations={})
|
||||
cmdRemove = CalcRemoveLocalDroneCommand(
|
||||
fitID=self.fitID,
|
||||
position=self.position,
|
||||
amount=math.inf)
|
||||
cmdAdd = CalcAddLocalDroneCommand(
|
||||
fitID=self.fitID,
|
||||
droneInfo=info,
|
||||
forceNewStack=True,
|
||||
ignoreRestrictions=True)
|
||||
success = self.internalHistory.submitBatch(cmdRemove, cmdAdd)
|
||||
eos.db.flush()
|
||||
sFit.recalc(self.fitID)
|
||||
sFit.fill(self.fitID)
|
||||
eos.db.commit()
|
||||
wx.PostEvent(gui.mainFrame.MainFrame.getInstance(), GE.FitChanged(fitIDs=(self.fitID,)))
|
||||
return success
|
||||
|
||||
def Undo(self):
|
||||
success = self.internalHistory.undoAll()
|
||||
eos.db.flush()
|
||||
sFit = Fit.getInstance()
|
||||
sFit.recalc(self.fitID)
|
||||
sFit.fill(self.fitID)
|
||||
eos.db.commit()
|
||||
wx.PostEvent(gui.mainFrame.MainFrame.getInstance(), GE.FitChanged(fitIDs=(self.fitID,)))
|
||||
return success
|
||||
60
gui/fitCommands/gui/localDrone/mutatedRevert.py
Normal file
60
gui/fitCommands/gui/localDrone/mutatedRevert.py
Normal file
@@ -0,0 +1,60 @@
|
||||
import math
|
||||
|
||||
import wx
|
||||
|
||||
import eos.db
|
||||
import gui.mainFrame
|
||||
from gui import globalEvents as GE
|
||||
from gui.fitCommands.calc.drone.localAdd import CalcAddLocalDroneCommand
|
||||
from gui.fitCommands.calc.drone.localRemove import CalcRemoveLocalDroneCommand
|
||||
from gui.fitCommands.helpers import DroneInfo, InternalCommandHistory
|
||||
from service.fit import Fit
|
||||
|
||||
|
||||
class GuiRevertMutatedLocalDroneCommand(wx.Command):
|
||||
|
||||
def __init__(self, fitID, position):
|
||||
wx.Command.__init__(self, True, 'Revert Local Drone from Mutated')
|
||||
self.internalHistory = InternalCommandHistory()
|
||||
self.fitID = fitID
|
||||
self.position = position
|
||||
|
||||
def Do(self):
|
||||
sFit = Fit.getInstance()
|
||||
fit = sFit.getFit(self.fitID)
|
||||
try:
|
||||
drone = fit.drones[self.position]
|
||||
except IndexError:
|
||||
return False
|
||||
if not drone.isMutated:
|
||||
return False
|
||||
info = DroneInfo(
|
||||
amount=drone.amount,
|
||||
amountActive=drone.amountActive,
|
||||
itemID=drone.baseItemID)
|
||||
cmdRemove = CalcRemoveLocalDroneCommand(
|
||||
fitID=self.fitID,
|
||||
position=self.position,
|
||||
amount=math.inf)
|
||||
cmdAdd = CalcAddLocalDroneCommand(
|
||||
fitID=self.fitID,
|
||||
droneInfo=info,
|
||||
forceNewStack=True,
|
||||
ignoreRestrictions=True)
|
||||
success = self.internalHistory.submitBatch(cmdRemove, cmdAdd)
|
||||
eos.db.flush()
|
||||
sFit.recalc(self.fitID)
|
||||
sFit.fill(self.fitID)
|
||||
eos.db.commit()
|
||||
wx.PostEvent(gui.mainFrame.MainFrame.getInstance(), GE.FitChanged(fitIDs=(self.fitID,)))
|
||||
return success
|
||||
|
||||
def Undo(self):
|
||||
success = self.internalHistory.undoAll()
|
||||
eos.db.flush()
|
||||
sFit = Fit.getInstance()
|
||||
sFit.recalc(self.fitID)
|
||||
sFit.fill(self.fitID)
|
||||
eos.db.commit()
|
||||
wx.PostEvent(gui.mainFrame.MainFrame.getInstance(), GE.FitChanged(fitIDs=(self.fitID,)))
|
||||
return success
|
||||
@@ -21,7 +21,10 @@ class GuiConvertMutatedLocalModuleCommand(wx.Command):
|
||||
def Do(self):
|
||||
sFit = Fit.getInstance()
|
||||
fit = sFit.getFit(self.fitID)
|
||||
mod = fit.modules[self.position]
|
||||
try:
|
||||
mod = fit.modules[self.position]
|
||||
except IndexError:
|
||||
return False
|
||||
if mod.isEmpty:
|
||||
return False
|
||||
if mod.isMutated:
|
||||
|
||||
@@ -19,7 +19,10 @@ class GuiRevertMutatedLocalModuleCommand(wx.Command):
|
||||
def Do(self):
|
||||
sFit = Fit.getInstance()
|
||||
fit = sFit.getFit(self.fitID)
|
||||
mod = fit.modules[self.position]
|
||||
try:
|
||||
mod = fit.modules[self.position]
|
||||
except IndexError:
|
||||
return False
|
||||
if mod.isEmpty:
|
||||
return False
|
||||
if not mod.isMutated:
|
||||
|
||||
@@ -158,8 +158,11 @@ class ModuleInfo:
|
||||
|
||||
class DroneInfo:
|
||||
|
||||
def __init__(self, itemID, amount, amountActive):
|
||||
def __init__(self, amount, amountActive, itemID, baseItemID=None, mutaplasmidID=None, mutations=None):
|
||||
self.itemID = itemID
|
||||
self.baseItemID = baseItemID
|
||||
self.mutaplasmidID = mutaplasmidID
|
||||
self.mutations = mutations
|
||||
self.amount = amount
|
||||
self.amountActive = amountActive
|
||||
|
||||
@@ -170,22 +173,40 @@ class DroneInfo:
|
||||
info = cls(
|
||||
itemID=drone.itemID,
|
||||
amount=drone.amount,
|
||||
amountActive=drone.amountActive)
|
||||
amountActive=drone.amountActive,
|
||||
baseItemID=drone.baseItemID,
|
||||
mutaplasmidID=drone.mutaplasmidID,
|
||||
mutations={m.attrID: m.value for m in drone.mutators.values()})
|
||||
return info
|
||||
|
||||
def toDrone(self):
|
||||
item = Market.getInstance().getItem(self.itemID, eager=('attributes', 'group.category'))
|
||||
mkt = Market.getInstance()
|
||||
item = mkt.getItem(self.itemID, eager=('attributes', 'group.category'))
|
||||
if self.baseItemID and self.mutaplasmidID:
|
||||
baseItem = mkt.getItem(self.baseItemID, eager=('attributes', 'group.category'))
|
||||
mutaplasmid = eos.db.getDynamicItem(self.mutaplasmidID)
|
||||
else:
|
||||
baseItem = None
|
||||
mutaplasmid = None
|
||||
try:
|
||||
drone = Drone(item)
|
||||
drone = Drone(item, baseItem=baseItem, mutaplasmid=mutaplasmid)
|
||||
except ValueError:
|
||||
pyfalog.warning('Invalid item: {}'.format(self.itemID))
|
||||
return None
|
||||
|
||||
if self.mutations is not None:
|
||||
for attrID, mutator in drone.mutators.items():
|
||||
if attrID in self.mutations:
|
||||
mutator.value = self.mutations[attrID]
|
||||
|
||||
drone.amount = self.amount
|
||||
drone.amountActive = self.amountActive
|
||||
return drone
|
||||
|
||||
def __repr__(self):
|
||||
return makeReprStr(self, ['itemID', 'amount', 'amountActive'])
|
||||
return makeReprStr(self, [
|
||||
'itemID', 'amount', 'amountActive',
|
||||
'baseItemID', 'mutaplasmidID', 'mutations'])
|
||||
|
||||
|
||||
class FighterInfo:
|
||||
|
||||
Reference in New Issue
Block a user