107 lines
3.5 KiB
Python
107 lines
3.5 KiB
Python
import re
|
|
|
|
# noinspection PyPackageRequirements
|
|
import wx
|
|
|
|
import gui.display as d
|
|
from service.fit import Fit
|
|
|
|
|
|
class FitBrowserLiteDialog(wx.Dialog):
|
|
|
|
def __init__(self, parent, title='Add Fits', excludedFitIDs=()):
|
|
wx.Dialog.__init__(self, parent, title=title, style=wx.DEFAULT_DIALOG_STYLE)
|
|
|
|
self.sFit = Fit.getInstance()
|
|
self.allFits = sorted(
|
|
(f for f in self.sFit.getAllFitsLite() if f.ID not in excludedFitIDs),
|
|
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)
|
|
|
|
|
|
class FitListView(d.Display):
|
|
|
|
DEFAULT_COLS = ['Base Name']
|
|
|
|
def __init__(self, parent):
|
|
super().__init__(parent, style=wx.BORDER_NONE)
|
|
self.fits = []
|
|
|
|
def updateView(self):
|
|
self.update(self.fits)
|
|
|
|
def refreshView(self):
|
|
self.refresh(self.fits)
|
|
|
|
def updateData(self, fits):
|
|
self.fits = fits
|
|
self.updateView()
|