diff --git a/boards/models.py b/boards/models.py --- a/boards/models.py +++ b/boards/models.py @@ -240,11 +240,7 @@ class Post(models.Model): def get_images_count(self): images_count = 1 if self.image else 0 - - # TODO Use query here - for reply in self.replies: - if reply.image: - images_count += 1 + images_count += self.replies.filter(image_width__gt=0).count() return images_count