def stage2Callback(self, data):
if self.GetActiveStage() != 2:
return
self.navpanel.ToggleRecentShips(False, False)
categoryID = self._stage2Data
ships = list(data[1])
sFit = Fit.getInstance()
ships.sort(key=self.raceNameKey)
racesList = []
subRacesFilter = {}
t_fits = 0 # total number of fits in this category
for ship in ships:
if ship.race:
if ship.race not in racesList:
racesList.append(ship.race)
for race, state in self.racesFilter.items():
if race in racesList:
subRacesFilter[race] = self.racesFilter[race]
override = True
for race, state in subRacesFilter.items():
if state:
override = False
break
for ship in ships:
fits = sFit.countFitsWithShip(ship.ID)
t_fits += fits
filter_ = subRacesFilter[ship.race] if ship.race else True
if override:
filter_ = True
if self.filterShipsWithNoFits:
if fits > 0:
if filter_:
self.lpane.AddWidget(ShipItem(self.lpane, ship))
else:
if filter_:
self.lpane.AddWidget(ShipItem(self.lpane, ship))
self.raceselect.RebuildRaces(racesList)
# refresh category cache
if t_fits == 0:
self.categoryFitCache[categoryID] = False
else:
self.categoryFitCache[categoryID] = True
self.lpane.ShowLoading(False)
self.lpane.RefreshList()
if self.showRacesFilterInStage2Only:
self.raceselect.Show(True)
self.Layout()
def searchStage(self, event):
self.lpane.ShowLoading(False)
self.navpanel.ShowNewFitButton(False)
self.navpanel.ShowSwitchEmptyGroupsButton(False)
if not event.back:
if self._activeStage != 4:
if len(self.browseHist) > 0:
self.browseHist.append((self._activeStage, self.lastdata))
else:
self.browseHist.append((1, 0))
self._lastStage = self._activeStage
self._activeStage = 4
sMkt = Market.getInstance()
sFit = Fit.getInstance()
query = event.text
self.lpane.Freeze()
self.lpane.RemoveAllChildren()
if query:
ships = sMkt.searchShips(query)
fitList = sFit.searchFits(query)
for ship in ships:
shipTrait = ship.traits.traitText if (
ship.traits
is not None) else "" # empty string if no traits
self.lpane.AddWidget(
ShipItem(self.lpane, ship.ID,
(ship.name, shipTrait,
len(sFit.getFitsWithShip(ship.ID))), ship.race))
for ID, name, shipID, shipName, booster, timestamp, notes in fitList:
ship = sMkt.getItem(shipID)
shipTrait = ship.traits.traitText if (
ship.traits
is not None) else "" # empty string if no traits
self.lpane.AddWidget(
FitItem(
self.lpane, ID,
(shipName, shipTrait, name, booster, timestamp, notes),
shipID))
if len(ships) == 0 and len(fitList) == 0:
self.lpane.AddWidget(
PFStaticText(self.lpane, label=u"No matching results."))
self.lpane.RefreshList(doFocus=False)
self.lpane.Thaw()
self.raceselect.RebuildRaces(self.RACE_ORDER)
if self.showRacesFilterInStage2Only:
self.raceselect.Show(False)
self.Layout()
