fb20847920
Added the ability to remove charges and projected items using the itemRemove built in context menu. In this commit I fixed a minor bug in the projected view that caused the right clicked item not to be SHOWN as selected until after the context menu had been spawned and closed. I also removed all parent references from the ContextMenu class and sub classes. I am instead using the fit service to do all the work of removing the items in the itemRemove class.
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
from gui.contextMenu import ContextMenu
|
|
from gui.itemStats import ItemStatsDialog
|
|
import gui.mainFrame
|
|
import gui.globalEvents as GE
|
|
import service
|
|
import wx
|
|
|
|
class DroneSplit(ContextMenu):
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
def display(self, srcContext, selection):
|
|
return srcContext in ("droneItem", "projectedDrone") and selection[0].amount > 1
|
|
|
|
def getText(self, itmContext, selection):
|
|
return "Split {0} Stack".format(itmContext)
|
|
|
|
def activate(self, fullContext, selection, i):
|
|
srcContext = fullContext[0]
|
|
dlg = DroneSpinner(self.mainFrame, selection[0], srcContext)
|
|
dlg.ShowModal()
|
|
dlg.Destroy()
|
|
|
|
DroneSplit.register()
|
|
|
|
|
|
class DroneSpinner(wx.Dialog):
|
|
|
|
def __init__(self, drone, context):
|
|
wx.Dialog.__init__(self, title="Select Amount", size=wx.Size(220, 60))
|
|
self.drone = drone
|
|
self.context = context
|
|
|
|
bSizer1 = wx.BoxSizer(wx.HORIZONTAL)
|
|
|
|
self.spinner = wx.SpinCtrl(self)
|
|
self.spinner.SetRange(1, drone.amount - 1)
|
|
self.spinner.SetValue(1)
|
|
|
|
bSizer1.Add(self.spinner, 0, wx.ALL, 5)
|
|
|
|
self.button = wx.Button(self, wx.ID_OK, u"Split")
|
|
bSizer1.Add(self.button, 0, wx.ALL, 5)
|
|
|
|
self.SetSizer(bSizer1)
|
|
self.Layout()
|
|
self.Centre(wx.BOTH)
|
|
self.button.Bind(wx.EVT_BUTTON, self.split)
|
|
|
|
def split(self, event):
|
|
sFit = service.Fit.getInstance()
|
|
mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
fitID = mainFrame.getActiveFit()
|
|
if self.context == "droneItem":
|
|
sFit.splitDroneStack(fitID, self.drone, self.spinner.GetValue())
|
|
else:
|
|
sFit.splitProjectedDroneStack(fitID, self.drone, self.spinner.GetValue())
|
|
wx.PostEvent(mainFrame, GE.FitChanged(fitID=fitID))
|
|
event.Skip()
|