116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
import re
|
|
|
|
# noinspection PyPackageRequirements
|
|
import wx
|
|
|
|
import gui.mainFrame
|
|
from gui.contextMenu import ContextMenuUnconditional
|
|
from service.fit import Fit
|
|
|
|
|
|
class AddBrowsedFits(ContextMenuUnconditional):
|
|
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
def display(self, callingWindow, srcContext):
|
|
if srcContext not in ('projected', 'commandView', 'graphFitList', 'graphTgtList'):
|
|
return False
|
|
return True
|
|
|
|
def getText(self, callingWindow, itmContext):
|
|
return 'Add Fit...'
|
|
|
|
def activate(self, callingWindow, fullContext, i):
|
|
titles = {
|
|
'projected': 'Add Projected Fits',
|
|
'commandView': 'Add Command Fits',
|
|
'graphFitList': 'Add Fits to Graph',
|
|
'graphTgtList': 'Add Targets to Graph'}
|
|
dlg = FitBrowserLiteDialog(self.mainFrame, title=titles[fullContext[0]])
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
pass
|
|
|
|
|
|
AddBrowsedFits.register()
|
|
|
|
|
|
class FitBrowserLiteDialog(wx.Dialog):
|
|
|
|
def __init__(self, parent, title='Add Fits'):
|
|
wx.Dialog.__init__(self, parent, title=title, style=wx.DEFAULT_DIALOG_STYLE)
|
|
|
|
from gui.builtinViews.fitListLite import FitListView
|
|
|
|
self.sFit = Fit.getInstance()
|
|
self.allFits = sorted(self.sFit.getAllFitsLite(), key=lambda f: (f.shipName, f.name))
|
|
self.SetMinSize((400, 400))
|
|
|
|
mainSizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
searchSizer = wx.BoxSizer(wx.HORIZONTAL)
|
|
self.searchBox = wx.TextCtrl(self, wx.ID_ANY, style=wx.TE_PROCESS_ENTER)
|
|
searchSizer.Add(self.searchBox, 1, wx.EXPAND | wx.ALL, 5)
|
|
mainSizer.Add(searchSizer, 0, wx.EXPAND | wx.ALL, 0)
|
|
|
|
listSizer = wx.BoxSizer(wx.HORIZONTAL)
|
|
self.fromList = FitListView(self)
|
|
listSizer.Add(self.fromList, 1, wx.EXPAND | wx.ALL, 5)
|
|
|
|
listButtonSizer = wx.BoxSizer(wx.VERTICAL)
|
|
listButtonSizer.AddStretchSpacer()
|
|
addButton = wx.Button(self, wx.ID_ANY, '>>', wx.DefaultPosition, wx.DefaultSize, 0)
|
|
listButtonSizer.Add(addButton, 0, wx.EXPAND | wx.ALL, 5)
|
|
removeButton = wx.Button(self, wx.ID_ANY, '<<', wx.DefaultPosition, wx.DefaultSize, 0)
|
|
listButtonSizer.Add(removeButton, 0, wx.EXPAND | wx.ALL, 5)
|
|
listButtonSizer.AddStretchSpacer()
|
|
listSizer.Add(listButtonSizer, 0, wx.EXPAND | wx.ALL, 5)
|
|
|
|
toList = FitListView(self)
|
|
listSizer.Add(toList, 1, wx.EXPAND | wx.ALL, 5)
|
|
mainSizer.Add(listSizer, 1, wx.EXPAND | wx.ALL, 0)
|
|
|
|
buttonSizer = self.CreateButtonSizer(wx.OK | wx.CANCEL)
|
|
if buttonSizer:
|
|
mainSizer.Add(buttonSizer, 0, wx.EXPAND | wx.ALL, 5)
|
|
|
|
self.resetContents()
|
|
|
|
self.inputTimer = wx.Timer(self)
|
|
self.Bind(wx.EVT_TIMER, self.OnInputTimer, self.inputTimer)
|
|
self.searchBox.Bind(event=wx.EVT_TEXT, handler=self.OnSearchChanged)
|
|
|
|
self.SetSizer(mainSizer)
|
|
self.Layout()
|
|
self.SetSize(self.GetBestSize())
|
|
self.CenterOnParent()
|
|
self.searchBox.SetFocus()
|
|
|
|
def OnSearchChanged(self, event):
|
|
event.Skip()
|
|
self.inputTimer.Stop()
|
|
self.inputTimer.Start(self.sFit.serviceFittingOptions['marketSearchDelay'], True)
|
|
|
|
def OnInputTimer(self, event):
|
|
event.Skip()
|
|
searchPattern = self.searchBox.GetValue().strip()
|
|
if not searchPattern:
|
|
self.resetContents()
|
|
else:
|
|
|
|
def isMatch(fit, searchTokens):
|
|
for token in searchTokens:
|
|
if token not in fit.name.lower() and token not in fit.shipName.lower():
|
|
return False
|
|
return True
|
|
|
|
matches = []
|
|
searchTokens = [t.lower() for t in re.split('\s+', searchPattern)]
|
|
for fit in self.allFits:
|
|
if isMatch(fit, searchTokens):
|
|
matches.append(fit)
|
|
self.fromList.updateData(matches)
|
|
|
|
def resetContents(self):
|
|
self.fromList.updateData(self.allFits)
|