frequently.views: 127 total statements, 100.0% covered

Generated: Mon 2013-04-29 09:50 CEST

Source file: /home/tobi/Projects/django-frequently/src/frequently/views.py

Stats: 118 executed, 0 missed, 9 excluded, 95 ignored

  1. """
  2. Views for the ``django-frequently`` application.
  3. """
  4. from math import fsum
  5. from django.contrib.auth.models import User
  6. from django.core.urlresolvers import reverse
  7. from django.http import HttpResponse, HttpResponseNotFound
  8. from django.template.response import TemplateResponse
  9. from django.utils import timezone
  10. from django.views.generic import CreateView, DetailView, ListView
  11. from frequently.forms import EntryForm
  12. from frequently.models import Entry, EntryCategory, Feedback
  13. class EntryMixin(object):
  14. """
  15. Mixin to handle and arrange the entry list.
  16. """
  17. def get_ordered_entries(self, queryset=False):
  18. """
  19. Custom ordering. First we get the average views and rating for
  20. the categories's entries. Second we created a rank by multiplying
  21. both. Last, we sort categories by this rank from top to bottom.
  22. Example:
  23. - Cat_1
  24. - Entry_1 (500 Views, Rating 2)
  25. - Entry_2 (200 Views, Rating -4)
  26. - Entry_3 (100 Views, Rating 3)
  27. - Cat_2
  28. - Entry_1 (200 Views, Rating 7)
  29. - Entry_2 (50 Views, Rating 2)
  30. Result:
  31. Cat_1 has a rank by: 88.88 (avg. views: 266.66, avg. rating: 0.33)
  32. Cat_2 has a rank by: 562.5 (avg. views: 125, avg. rating: 4.5)
  33. Cat_2 will be displayed at the top. The algorithm is quality-oriented,
  34. as you can see.
  35. """
  36. if queryset:
  37. self.queryset = queryset
  38. else:
  39. self.queryset = EntryCategory.objects.all()
  40. if self.queryset:
  41. for category in self.queryset:
  42. entries = category.get_entries()
  43. if entries:
  44. amount_list = [e.amount_of_views for e in entries]
  45. rating_list = [e.rating() for e in entries]
  46. views_per_entry = fsum(amount_list) / len(amount_list)
  47. rating_per_entry = fsum(rating_list) / len(rating_list)
  48. category.last_rank = views_per_entry * rating_per_entry
  49. category.save()
  50. else:
  51. self.queryset = self.queryset.exclude(pk=category.pk)
  52. self.queryset = sorted(self.queryset, key=lambda c: c.last_rank,
  53. reverse=True)
  54. return self.queryset
  55. def post(self, request, *args, **kwargs):
  56. if "get_answer" in request.POST.keys():
  57. entry = Entry.objects.get(pk=request.POST['get_answer'])
  58. entry.last_view_date = timezone.now()
  59. entry.amount_of_views += 1
  60. entry.save()
  61. return TemplateResponse(
  62. request,
  63. 'frequently/partials/answer.html',
  64. {
  65. 'entry': entry,
  66. 'rated_entries': self.request.session.get(
  67. 'rated_entries', False),
  68. },
  69. )
  70. self.feedback = Feedback()
  71. if "user_id" in request.POST.keys():
  72. try:
  73. user_id = int(request.POST.get('user_id'))
  74. try:
  75. self.feedback.user = User.objects.get(pk=user_id)
  76. except User.DoesNotExist:
  77. pass
  78. except ValueError:
  79. pass
  80. if 'ratingID' in request.POST.keys() and request.is_ajax():
  81. try:
  82. entry_id = int(request.POST.get('ratingID').replace(
  83. 'ratingID', ''))
  84. try:
  85. entry = Entry.objects.get(pk=entry_id)
  86. return HttpResponse(entry.rating())
  87. except Entry.DoesNotExist:
  88. return HttpResponseNotFound()
  89. except ValueError:
  90. return HttpResponseNotFound()
  91. for key in request.POST.keys():
  92. if key.startswith('up') or key.startswith('down'):
  93. try:
  94. entry_id = int(key.replace('up', '').replace('down', ''))
  95. try:
  96. entry = Entry.objects.get(pk=entry_id)
  97. except Entry.DoesNotExist:
  98. return HttpResponseNotFound()
  99. except ValueError:
  100. return HttpResponseNotFound()
  101. if not request.session.get('rated_entries', False):
  102. request.session['rated_entries'] = []
  103. if not entry.pk in request.session['rated_entries']:
  104. request.session['rated_entries'].append(entry.pk)
  105. request.session.modified = True
  106. self.feedback.entry = entry
  107. if key.startswith('up'):
  108. entry.upvotes += 1
  109. self.feedback.validation = "P"
  110. if key.startswith('down'):
  111. entry.downvotes += 1
  112. self.feedback.validation = "N"
  113. entry.save()
  114. self.feedback.save()
  115. if request.is_ajax():
  116. return TemplateResponse(
  117. request,
  118. 'frequently/partials/feedback_form.html',
  119. {
  120. 'feedback_entry': entry.pk,
  121. 'feedback': self.feedback,
  122. },
  123. )
  124. elif key.startswith('feedback'):
  125. try:
  126. feedback_id = int(key.replace('feedback', ''))
  127. try:
  128. self.feedback = Feedback.objects.get(pk=feedback_id)
  129. except Feedback.DoesNotExist:
  130. return HttpResponseNotFound()
  131. except ValueError:
  132. return HttpResponseNotFound()
  133. self.feedback.remark = request.POST.get("remark")
  134. self.feedback.save()
  135. if request.is_ajax():
  136. return TemplateResponse(
  137. request,
  138. 'frequently/partials/feedback_form.html',
  139. {'feedback_send': True},
  140. )
  141. return self.get(self, request, *args, **kwargs)
  142. class EntryCategoryListView(EntryMixin, ListView):
  143. """
  144. Main view to display all categories and their entries.
  145. """
  146. model = EntryCategory
  147. template_name = "frequently/entry_list.html"
  148. def get_queryset(self):
  149. """
  150. Customized to get the ordered categories and entries from the Mixin.
  151. """
  152. self.queryset = super(EntryCategoryListView, self).get_queryset()
  153. return self.get_ordered_entries(self.queryset)
  154. class EntryDetailView(EntryMixin, DetailView):
  155. """
  156. Main view to display one entry.
  157. """
  158. model = Entry
  159. template_name = "frequently/entry_list.html"
  160. def get_object(self, **kwargs):
  161. obj = super(EntryDetailView, self).get_object(**kwargs)
  162. obj.last_view_date = timezone.now()
  163. obj.amount_of_views += 1
  164. obj.save()
  165. return obj
  166. def get_context_data(self, **kwargs):
  167. context = super(EntryDetailView, self).get_context_data(**kwargs)
  168. context.update({
  169. 'rated_entries': self.request.session.get('rated_entries', False),
  170. 'object_list': self.get_ordered_entries(),
  171. })
  172. for key in self.request.POST.keys():
  173. if key.startswith('down') or key.startswith('up'):
  174. context.update({
  175. 'feedback_entry': int(
  176. key.replace('up', '').replace('down', '')),
  177. 'feedback': self.feedback,
  178. })
  179. return context
  180. return context
  181. class EntryCreateView(CreateView):
  182. """
  183. Feedback submission form view.
  184. """
  185. model = Entry
  186. form_class = EntryForm
  187. def get_form_kwargs(self):
  188. kwargs = super(EntryCreateView, self).get_form_kwargs()
  189. if self.request.user.is_authenticated():
  190. kwargs.update({
  191. 'owner': self.request.user,
  192. })
  193. return kwargs
  194. def get_success_url(self):
  195. return reverse('frequently_list')