minor changes and caching home chart - closes #84

Luan
1 parent 6c7f0db8
Showing 1 changed file with 18 additions and 5 deletions Show diff stats
src/home/views.py
 from collections import OrderedDict
+from django.core.cache import cache
 from django.shortcuts import render
 from search.utils import trans
 from haystack.query import SearchQuerySet
+from proxy.models import WikiCollabCount, TicketCollabCount
 from super_archives.models import Thread
@@ -16,11 +18,22 @@ def index(request):
     latest_threads = Thread.objects.all()[:6]
     hottest_threads = Thread.highest_score.from_haystack()[:6]
-    count_types = OrderedDict()
-    for type in ['thread', 'ticket', 'wiki', 'changeset', 'attachment']:
-        count_types[trans(type)] = SearchQuerySet().filter(
-            type=type,
-        ).count()
+    count_types = cache.get('home_chart')
+    if count_types is None:
+        count_types = OrderedDict()
+        for type in ['thread', 'changeset', 'attachment']:
+            count_types[trans(type)] = SearchQuerySet().filter(
+                type=type,
+            ).count()
+
+        count_types[trans('ticket')] = sum([
+            ticket.count for ticket in TicketCollabCount.objects.all()
+        ])
+
+        count_types[trans('wiki')] = sum([
+            wiki.count for wiki in WikiCollabCount.objects.all()
+        ])
+        cache.set('home_chart', count_types, 60)
     context = {
         'hottest_threads': hottest_threads[:6],