diff --git a/gui/droneView.py b/gui/droneView.py index 0a383d6af..2ca138430 100644 --- a/gui/droneView.py +++ b/gui/droneView.py @@ -51,12 +51,26 @@ class DroneView(d.Display): self.original = fit.drones if fit is not None else None self.drones = stuff = fit.drones[:] if fit is not None else None + name = lambda d: d.item.name if stuff is not None: - stuff.sort(key=lambda d: d.item.name) + # Need to fix this sort parameter + stuff.sort( key=lambda d: d.item.name ) self.update(stuff) event.Skip() + DRONE_ORDER = (u'Light Scout Drones', u'Medium Scout Drones', + u'Heavy Attack Drones', u'Sentry Drones', u'Fighters', + u'Fighter Bombers', u'Combat Utility Drones', + u'Electronic Warfare Drones', u'Logistic Drones', u'Mining Drones') + def droneKey(self, drone): + item = drone.item + if item.marketGroup is None: + item = item.metaGroup.parent + + return (self.DRONE_ORDER.index(item.marketGroup.name), + drone.item.name) + def addItem(self, event): cFit = service.Fit.getInstance() fitID = self.mainFrame.getActiveFit()