Do not get variations by "group" unless the item is part of a set of hard-coded categories. #1232

eos/db/gamedata/queries.py

@@ -317,8 +317,7 @@ def getVariations(itemids, groupIDs=None, where=None, eager=None):
         vars = gamedata_session.query(Item).options(*processEager(eager)).join((groups_table, joinon)).filter(
                 filter).all()
 
-        return vars
-
+    return vars
 
 @cachedQuery(1, "attr")
 def getAttributeInfo(attr, eager=None):

service/market.py

@@ -542,6 +542,10 @@ class Market(object):
         # Set-container for variables
         variations = set()
         variations_limiter = set()
+
+        # if item belongs to these categories, use their group to find "variations"
+        categories = ['Drone', 'Fighter', 'Implant']
+
         for item in items:
             if item.category.ID == 20:  # Implants and Boosters
                 implant_remove_list = set()
@@ -592,7 +596,7 @@ class Market(object):
         variations.update(parents)
         # Add all variations of parents to the set
         parentids = tuple(item.ID for item in parents)
-        groupids = tuple(item.group.ID for item in parents)
+        groupids = tuple(item.group.ID for item in parents if item.category.name in categories)
         variations_list = eos.db.getVariations(parentids, groupids)
 
         if variations_limiter: