dave/pyfa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

Add recalcs to Add and remove module

Browse Source
This commit is contained in:
blitzmann
2018-07-30 00:05:00 -04:00
parent 984978a80d
commit cfb7a70da5
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
gui/fitCommands/guiAddModule.py
View File

@@ -1,5 +1,4 @@
import wx
from service.fit import Fit
import gui.mainFrame
from gui import globalEvents as GE
@@ -22,7 +21,7 @@ class GuiModuleAddCommand(wx.Command):
success = False
# if we have a position set, try to apply the module to that position
# todo: check to see if item is a charge. if it is, dont try to add module, but instead set amm
# todo: check to see if item is a charge. if it is, dont try to add module, but instead set ammo
if self.new_position:
success = self.internal_history.Submit(FitReplaceModuleCommand(self.fitID, self.new_position, self.itemID))
if not success:
@@ -34,6 +33,7 @@ class GuiModuleAddCommand(wx.Command):
success = self.internal_history.Submit(FitAddModuleCommand(self.fitID, self.itemID))
if success:
self.sFit.recalc(self.fitID)
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID, action="modadd", typeID=self.itemID))
return True
return False
@@ -48,5 +48,6 @@ class GuiModuleAddCommand(wx.Command):
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, action="moddel", typeID=self.itemID))
return True

2
gui/fitCommands/guiRemoveModule.py
View File

@@ -24,11 +24,13 @@ class GuiModuleRemoveCommand(wx.Command):
if success is not None:
# self.slotsChanged() # todo: fix
self.sFit.recalc(self.fitID)
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID, action="moddel", typeID=set([mod.itemID for mod in self.modCache])))
return True
def Undo(self):
for x in self.internal_history.Commands:
self.internal_history.Undo()
self.sFit.recalc(self.fitID)
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID, action="modadd", typeID=set([mod.itemID for mod in self.modCache])))
return True