Merge pull request #188 from PROCOLLAB-github/dev

Пофиксил 500-ки в чатах драфтовых проеектов

Author: yakser

chats/views.py

@@ -22,6 +22,7 @@
     DirectChatDetailSerializer,
 )
 from chats.utils import get_all_files
+from files.models import UserFile
 from files.serializers import UserFileSerializer
 
 User = get_user_model()
@@ -124,12 +125,15 @@ class ProjectChatMessageList(ListCreateAPIView):
     pagination_class = MessageListPagination
 
     def get_queryset(self):
-        return (
-            ProjectChat.objects.get(id=self.kwargs["pk"])
-            .messages.filter(is_deleted=False)
-            .order_by("-created_at")
-            .all()
-        )
+        try:
+            return (
+                ProjectChat.objects.get(id=self.kwargs["pk"])
+                .messages.filter(is_deleted=False)
+                .order_by("-created_at")
+                .all()
+            )
+        except ProjectChat.DoesNotExist:
+            return ProjectChat.objects.none()
 
     def post(self, request, *args, **kwargs):
         # TODO: try to create a message in a chat. If chat doesn't exist, create it and then create a message.
@@ -155,6 +159,8 @@ class ProjectChatFileList(ListCreateAPIView):
     permission_classes = [IsProjectChatMember]
 
     def get_queryset(self):
-        messages = ProjectChat.objects.get(id=self.kwargs["pk"]).messages.all()
-
-        return get_all_files(messages)
+        try:
+            messages = ProjectChat.objects.get(id=self.kwargs["pk"]).messages.all()
+            return get_all_files(messages)
+        except ProjectChat.DoesNotExist:
+            return UserFile.objects.none()