Count attachments, not only images

boards/models/thread.py

@@ -140,6 +140,11 b' class Thread(models.Model):'
             .annotate(images_count=Count(
             'attachments')).aggregate(Sum('images_count'))['images_count__sum'] or 0
 
+    @cached_result(key_method=_get_cache_key)
+    def get_attachment_count(self) -> int:
+        return self.get_replies().annotate(attachment_count=Count('attachments'))\
+                .aggregate(Sum('attachment_count'))['attachment_count__sum'] or 0
+
     def can_bump(self) -> bool:
         """
         Checks if the thread can be bumped by replying to it.

boards/static/js/thread_update.js

@@ -182,7 +182,7 b' function getReplyCount() {'
 }
 
 function getImageCount() {
-    return $('.thread').find('img').length
+    return $('.thread').find('.image').length
 }
 
 /**

boards/templates/boards/post.html

@@ -153,7 +153,7 b''
         <div class="metadata">
             {% if need_open_link %}
                 ♥ {{ thread.get_reply_count }}
-                ❄ {{ thread.get_images_count }}
+                ❄ {{ thread.get_attachment_count }}
                 <a href="{% url 'thread_gallery' post.id %}">G</a>
                 <a href="{% url 'thread_tree' post.id %}">T</a>
             {% endif %}
@@ -163,4 +163,4 b''
         </div>
     {% endif %}
 </div>
-{% endautoescape %}
  No newline at end of file
+{% endautoescape %}

boards/templates/boards/thread.html

@@ -29,8 +29,8 b''
         {% with replies_count=thread.get_reply_count%}
             ♥ <span id="reply-count">{{ thread.get_reply_count }}</span>{% if thread.has_post_limit %}/{{ thread.max_posts }}{% endif %}
         {% endwith %}
-        {% with images_count=thread.get_images_count%}
-            ❄ <span id="image-count">{{ images_count }}</span>
+        {% with attachment_count=thread.get_attachment_count%}
+            ❄ <span id="image-count">{{ attachment_count }}</span>
         {% endwith %}
         {% trans 'Last update: ' %}<span id="last-update"><time datetime="{{ thread.last_edit_time|date:'c' }}">{{ thread.last_edit_time }}</time></span>
     </span>