Filtering UI

[prawokultury.git] / questions / views.py

diff --git a/questions/views.py b/questions/views.py
index d493dd9..9b4b035 100644 (file)
--- a/questions/views.py
+++ b/questions/views.py
@@ -42,20 +42,21 @@ class QuestionListView(ListView):
 
     def get_context_data(self, *args, **kwargs):
         context = super(QuestionListView, self).get_context_data(*args, **kwargs)
-        context['tags'] = Tag.objects.filter(items__question__published=True
-            ).annotate(c=models.Count('items__tag')).order_by('-c', 'slug')
         context['tag'] = self.tag
         context['tag_categories'] = TagCategory.objects.all().annotate(click_count = models.Sum('tags__click_count'))
+        context['tag_lists'] = dict()
         
-        # Calculating factors for category and tag clouds
+        tags = Tag.objects.filter(items__question__published=True
+            ).annotate(c=models.Count('items__tag')).order_by('category__slug', '-c', 'slug')
         all_tag_clicks_count = Tag.objects.all().aggregate(models.Sum('click_count'))['click_count__sum']
         annotated_categories = dict()
         minimum_factor = 0.7
         for category in context['tag_categories']:
             annotated_categories[category.id] = category
             category.factor =  '%.2f' % (minimum_factor + (float(category.click_count) / all_tag_clicks_count))
-        for tag in [t for t in context['tags'] if t.category]:
-            category = annotated_categories[tag.category.id]
-            tag.factor = '%.2f' % (minimum_factor + ((float(tag.click_count) / category.click_count) if category.click_count else 0))
-        
+        for tag in tags:
+            if tag.category:
+                category = annotated_categories[tag.category.id]
+                tag.factor = '%.2f' % (minimum_factor + ((float(tag.click_count) / category.click_count) if category.click_count else 0))
+            context['tag_lists'].setdefault(tag.category.id if tag.category else 0, []).append(tag)
         return context