Have projectedName handle projected fits instead of passing them on and

having the slave die
This commit is contained in:
cncfanatics
2010-11-05 18:05:58 +01:00
parent 653a3108d6
commit 91309299f5
2 changed files with 7 additions and 2 deletions

View File

@@ -20,7 +20,6 @@
from gui import builtinViewColumns from gui import builtinViewColumns
from gui.viewColumn import ViewColumn from gui.viewColumn import ViewColumn
from gui import bitmapLoader from gui import bitmapLoader
from eos.types import Slot
import wx import wx
class StuffName(ViewColumn): class StuffName(ViewColumn):

View File

@@ -20,8 +20,9 @@
from gui.viewColumn import ViewColumn from gui.viewColumn import ViewColumn
import gui.builtinViewColumns.name import gui.builtinViewColumns.name
import gui.builtinViewColumns.droneNameAmount import gui.builtinViewColumns.droneNameAmount
from eos.types import Drone from eos.types import Drone, Fit
import wx import wx
from gui import bitmapLoader
class ProjectedName(ViewColumn): class ProjectedName(ViewColumn):
name = "Projected Name" name = "Projected Name"
@@ -29,18 +30,23 @@ class ProjectedName(ViewColumn):
ViewColumn.__init__(self, fittingView) ViewColumn.__init__(self, fittingView)
self.columnText = "Name" self.columnText = "Name"
self.mask = wx.LIST_MASK_TEXT self.mask = wx.LIST_MASK_TEXT
self.shipImage = fittingView.imageList.Add(bitmapLoader.getBitmap("ship_small", "icons"))
self.slave = gui.builtinViewColumns.name.StuffName(fittingView, params) self.slave = gui.builtinViewColumns.name.StuffName(fittingView, params)
self.droneSlave = gui.builtinViewColumns.droneNameAmount.DroneNameAmount(fittingView, params) self.droneSlave = gui.builtinViewColumns.droneNameAmount.DroneNameAmount(fittingView, params)
def getText(self, stuff): def getText(self, stuff):
if isinstance(stuff, Drone): if isinstance(stuff, Drone):
return self.droneSlave.getText(stuff) return self.droneSlave.getText(stuff)
elif isinstance(stuff, Fit):
return "%s (%s)" % (stuff.name, stuff.ship.item.name)
else: else:
return self.slave.getText(stuff) return self.slave.getText(stuff)
def getImageId(self, thing): def getImageId(self, thing):
if isinstance(thing, Drone): if isinstance(thing, Drone):
return self.droneSlave.getImageId(thing) return self.droneSlave.getImageId(thing)
elif isinstance(thing, Fit):
return self.shipImage
else: else:
return self.slave.getImageId(thing) return self.slave.getImageId(thing)