be955fb6ee535a161375f0a9680403cbf2f54373
[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
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         context = super(QuestionListView, self).get_context_data(*args, **kwargs)
45         context['tags'] = Tag.objects.filter(items__question__published=True
46             ).annotate(c=models.Count('items__tag')).order_by('-c', 'slug')
47         context['tag'] = self.tag
48         return context