Fixed #397 , take in account forced market groups

This commit is contained in:
HomeWorld
2011-01-19 11:40:39 +02:00
parent 0ad1132f21
commit e9a4faee45
2 changed files with 20 additions and 6 deletions

View File

@@ -236,6 +236,22 @@ class Market():
def getMarketGroup(self, marketGroupId):
return eos.db.getMarketGroup(marketGroupId)
def getForcedMarketGroupID(self,itemID):
if itemID in self.FORCED_ITEM_MKTGRPS:
grpID,meta = self.FORCED_ITEM_MKTGRPS[itemID]
return grpID
return None
def getMarketGroupName(self, item):
if item.marketGroup == None:
marketGroupID = self.getForcedMarketGroupID(item.ID)
marketGroup = self.getMarketGroup(marketGroupID)
groupName = marketGroup.name
else:
groupName = item.marketGroup.name
return groupName
MARKET_GROUPS = (9, #Modules
1111, #Rigs