Working book shop.
[prawokultury.git] / questions / views.py
1 # -*- coding: utf-8 -*-
2 # This file is part of PrawoKultury, licensed under GNU Affero GPLv3 or later.
3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 #
5 from django.core.urlresolvers import reverse_lazy
6 from django.db import models
7 from django.shortcuts import get_object_or_404
8 from django.views.generic import ListView
9 from django.views.generic.edit import FormView
10 from .forms import QuestionForm
11 from .models import Question, Tag, TagCategory
12
13
14 class QuestionFormView(FormView):
15     form_class = QuestionForm
16     template_name = "questions/question_form.html"
17     success_url = reverse_lazy("questions_thanks")
18
19     def form_valid(self, form):
20         form.save()
21         return super(QuestionFormView, self).form_valid(form)
22
23
24 class QuestionListView(ListView):
25     def get(self, request, *args, **kwargs):
26         self.tag = None
27         if 'tag' in request.GET:
28             try:
29                 self.tag = Tag.objects.filter(items__question__published=True, slug=request.GET['tag'])[0]
30             except IndexError:
31                 pass
32         return super(QuestionListView, self).get(request, *args, **kwargs)
33
34     def get_queryset(self):
35         qs = Question.objects.filter(published=True
36                 ).order_by('-published_at')
37         if self.tag:
38             qs = qs.filter(tags=self.tag)
39             self.tag.click_count += 1
40             self.tag.save()
41         return qs
42
43     def get_context_data(self, *args, **kwargs):
44         def get_cloud_size(clicks, relate_to):
45             return '%.2f' % (0.7 + (float(clicks) / relate_to if relate_to != 0 else 0))
46         
47         context = super(QuestionListView, self).get_context_data(*args, **kwargs)
48         context['tag'] = self.tag
49         context['tag_categories'] = TagCategory.objects.all().annotate(click_count = models.Sum('tags__click_count'))
50         context['tag_lists'] = dict()
51         
52         annotated_categories = dict()
53         all_tags_click_count = Tag.objects.all().aggregate(models.Sum('click_count'))['click_count__sum']
54         for category in context['tag_categories']:
55             annotated_categories[category.id] = category
56             category.cloud_size =  get_cloud_size(category.click_count, all_tags_click_count)
57         
58         tags = Tag.objects.filter(items__question__published=True) \
59                   .annotate(c=models.Count('items__tag')).order_by('category__slug', '-c', 'slug')
60         uncategorized_tags_click_count = Tag.objects.filter(category=None).aggregate(models.Sum('click_count'))['click_count__sum']
61         for tag in tags:
62             if tag.category:
63                 category_click_count = annotated_categories[tag.category.id].click_count
64             else:
65                 category_click_count = uncategorized_tags_click_count
66             tag.cloud_size = get_cloud_size(tag.click_count, category_click_count)
67             context['tag_lists'].setdefault(tag.category.id if tag.category else 0, []).append(tag)
68         context['has_uncategorized_tags'] = 0 in context['tag_lists']
69         
70         return context