| 10 | |
| 11 | |
| 12 | class CalcChangeShipModeCommand(wx.Command): |
| 13 | |
| 14 | def __init__(self, fitID, itemID): |
| 15 | wx.Command.__init__(self, True, 'Change Ship Mode') |
| 16 | self.fitID = fitID |
| 17 | self.itemID = itemID |
| 18 | self.savedItemID = None |
| 19 | |
| 20 | def Do(self): |
| 21 | pyfalog.debug('Doing changing ship mode to {} for fit {}'.format(self.itemID, self.fitID)) |
| 22 | fit = Fit.getInstance().getFit(self.fitID) |
| 23 | self.savedItemID = fit.mode.item.ID |
| 24 | item = Market.getInstance().getItem(self.itemID) |
| 25 | mode = Mode(item) |
| 26 | fit.mode = mode |
| 27 | return True |
| 28 | |
| 29 | def Undo(self): |
| 30 | pyfalog.debug('Undoing changing ship mode to {} for fit {}'.format(self.itemID, self.fitID)) |
| 31 | cmd = CalcChangeShipModeCommand(self.fitID, self.savedItemID) |
| 32 | return cmd.Do() |