Speed up getting new post count for favorites. Get favorites at page start, not only by JS

boards/context_processors.py

@@ -5,7 +5,7 b' from boards.models.user import Notificat'
 __author__ = 'neko259'
 
 from boards import settings
-from boards.models import Post, Tag
+from boards.models import Post, Tag, Thread
 
 CONTEXT_SITE_NAME = 'site_name'
 CONTEXT_VERSION = 'version'
@@ -20,6 +20,7 b" CONTEXT_TAGS_STR = 'tags_str'"
 CONTEXT_IMAGE_VIEWER = 'image_viewer'
 CONTEXT_HAS_FAV_THREADS = 'has_fav_threads'
 CONTEXT_POW_DIFFICULTY = 'pow_difficulty'
+CONTEXT_NEW_POST_COUNT = 'new_post_count'
 
 
 def get_notifications(context, request):
@@ -31,11 +32,23 b' def get_notifications(context, request):'
             SETTING_LAST_NOTIFICATION_ID)
 
         new_notifications_count = Notification.objects.get_notification_posts(
-                usernames=usernames, last=last_notification_id).count()
+                usernames=usernames, last=last_notification_id).only('id').count()
     context[CONTEXT_NEW_NOTIFICATIONS_COUNT] = new_notifications_count
     context[CONTEXT_USERNAMES] = usernames
 
 
+def get_new_post_count(context, request):
+    settings_manager = get_settings_manager(request)
+    fav_threads = settings_manager.get_fav_threads()
+    fav_thread_ops = Post.objects.filter(id__in=fav_threads.keys()) \
+        .order_by('-pub_time').only('thread_id', 'pub_time')
+
+    ops = [{'op': op, 'last_id': fav_threads[str(op.id)]} for op in fav_thread_ops]
+    count = Thread.objects.get_new_post_count(ops)
+    if count > 0:
+        context[CONTEXT_NEW_POST_COUNT] = '(+{})'.format(count)
+
+
 def user_and_ui_processor(request):
     context = dict()
 
@@ -64,5 +77,6 b' def user_and_ui_processor(request):'
         len(settings_manager.get_fav_threads()) > 0
 
     get_notifications(context, request)
+    get_new_post_count(context, request)
 
     return context

boards/models/post/__init__.py

@@ -163,6 +163,9 b' class Post(models.Model, Viewable):'
     def get_thread(self):
         return self.thread
 
+    def get_thread_id(self):
+        return self.thread_id
+
     def get_threads(self) -> QuerySet:
         """
         Gets post's thread.

boards/models/thread.py

@@ -68,7 +68,7 b' class ThreadManager(models.Manager):'
         # TODO Use classes instead of dicts
         for data in datas:
             if data['last_id'] != FAV_THREAD_NO_UPDATES:
-                q = (Q(id=data['op'].get_thread().id)
+                q = (Q(id=data['op'].get_thread_id())
                      & Q(multi_replies__id__gt=data['last_id']))
                 if query is None:
                     query = q

boards/static/css/base.css

@@ -141,10 +141,6 b' textarea, input {'
     margin: 1ex;
 }
 
-#new-fav-post-count {
-    display: none;
-}
-
 .hidden_post {
     opacity: 0.2;
 }

boards/static/js/main.js

@@ -90,8 +90,6 b' function updateFavPosts() {'
 }
 
 function initFavPanel() {
-    updateFavPosts();
-
     if ($('#fav-panel-btn').length > 0) {
         setInterval(updateFavPosts, FAV_POST_UPDATE_PERIOD);
         $('#fav-panel-btn').click(function() {

boards/templates/boards/base.html

@@ -40,7 +40,7 b''
         <a href="{% url 'feed' %}" title="{% trans 'Feed' %}">{% trans 'feed' %}</a>,
         <a href="{% url 'random' %}" title="{% trans 'Random images' %}">{% trans 'random' %}</a>{% if has_fav_threads %},
 
-            <a href="#" id="fav-panel-btn">{% trans 'favorites' %} <span id="new-fav-post-count"></span></a>
+            <a href="#" id="fav-panel-btn">{% trans 'favorites' %} <span id="new-fav-post-count" {% if not new_post_count %}style="display: none" {% endif %}>{{ new_post_count }}</span></a>
         {% endif %}
 
         {% if usernames %}