make sure all add-ons have all_categories defined

This commit is contained in:
Jeff Balogh 2010-05-20 17:22:19 -07:00
Родитель 12e7aaeab9
Коммит a8afc15017
1 изменённых файлов: 4 добавлений и 3 удалений

Просмотреть файл

@ -590,11 +590,12 @@ class Category(amo.models.ModelBase):
@staticmethod
def transformer(addons):
addon_dict = dict((a.id, a) for a in addons)
qs = (Category.uncached.filter(addons__in=addons)
.extra(select={'addon_id': 'addons_categories.addon_id'}))
for addon_id, cats in sorted_groupby(qs, 'addon_id'):
addon_dict[addon_id].all_categories = list(cats)
cats = dict((addon_id, list(cs))
for addon_id, cs in sorted_groupby(qs, 'addon_id'))
for addon in addons:
addon.all_categories = cats.get(addon.id, [])
class CompatibilityReport(models.Model):