fix HasChatUnreadsView

Author: VeryBigSad

chats/views.py

@@ -219,14 +219,14 @@ def get(self, request, *args, **kwargs):
         user = request.user
         # get all user chats
         direct_messages = user.direct_chats.all().prefetch_related("messages")
-        project_messages = user.get_project_chats().prefetch_related("messages")
-
-        has_direct_messages_unread = direct_messages.filter(
-            messages__is_read=False
-        ).exists()
-        has_project_messages_unread = project_messages.filter(
-            messages__is_read=False
-        ).exists()
+        project_chats = user.get_project_chats().prefetch_related("messages")
+
+        has_direct_messages_unread = (
+            direct_messages.filter(messages__is_read=False).distinct().exists()
+        )
+        has_project_messages_unread = (
+            project_chats.filter(messages__is_read=False).distinct().exists()
+        )
         return Response(
             {"has_unreads": has_direct_messages_unread or has_project_messages_unread}
         )

users/models.py

@@ -1,5 +1,6 @@
 from django.contrib.auth.models import AbstractUser
 from django.db import models
+from chats.models import ProjectChat
 
 from users.constants import (
     ADMIN,
@@ -134,13 +135,8 @@ def calculate_ordering_score(self) -> int:
         return score
 
     def get_project_chats(self) -> list:
-        collaborations = self.collaborations.prefetch_related(
-            "project__project_chats"
-        ).all()
-        projects = []
-        for collaboration in collaborations:
-            projects.extend(list(collaboration.project.project_chats.all()))
-        return projects
+        user_project_ids = self.collaborations.all().values_list("project_id", flat=True)
+        return ProjectChat.objects.filter(project__in=user_project_ids)
 
     def get_key_skills(self) -> list[str]:
         return [skill.strip() for skill in self.key_skills.split(",") if skill.strip()]