Support new directAttrRequest format in pyfa

This commit is contained in:
DarkPhoenix
2011-06-24 13:01:55 +04:00
parent 1b53dd9019
commit 2c1ec5ae93

View File

@@ -595,11 +595,18 @@ class Market():
"""Find items according to given text pattern"""
self.searchWorkerThread.scheduleSearch(name, callback)
def directAttrRequest(self, items, attrID):
itemIDs = tuple(map(lambda i: i.ID, items))
def directAttrRequest(self, items, attribs):
try:
itemIDs = tuple(map(lambda i: i.ID, items))
except TypeError:
itemIDs = (items.ID,)
try:
attrIDs = tuple(map(lambda i: i.ID, attribs))
except TypeError:
attrIDs = (attribs.ID,)
info = {}
for ID, val in eos.db.directAttributeRequest(itemIDs, attrID):
info[ID] = val
for itemID, typeID, val in eos.db.directAttributeRequest(itemIDs, attrIDs):
info[itemID] = val
return info