From 1fa04cec95f4aba570bdfd6c2bb22003a4c78bc6 Mon Sep 17 00:00:00 2001 From: j <0x006A@0x2620.org> Date: Fri, 30 Sep 2011 02:46:58 +0200 Subject: [PATCH] fix icon for dynamic lists --- pandora/itemlist/models.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/pandora/itemlist/models.py b/pandora/itemlist/models.py index adc5b628..bf6f5884 100644 --- a/pandora/itemlist/models.py +++ b/pandora/itemlist/models.py @@ -57,12 +57,15 @@ class List(models.Model): self.items_sum = self.get_items_sum(self.user) super(List, self).save(*args, **kwargs) - def get_items_sum(self, user=None): + def get_items(self, user=None): if self.query.get('static', False): - return self.items.count() - else: - from item.models import Item - return Item.objects.find({'query': self.query}, user).count() + return self.items + from item.models import Item + return Item.objects.find({'query': self.query}, user) + + + def get_items_sum(self, user=None): + return self.get_items(user).count() def add(self, item): q = self.items.filter(id=item.id) @@ -119,7 +122,8 @@ class List(models.Model): def update_icon(self): frames = [] for i in self.poster_frames: - qs = self.items.filter(itemId=i['item']) + from item.models import Item + qs = Item.objects.filter(itemId=i['item']) if qs.count() > 0: frame = qs[0].frame(i['position']) if frame: