Merge pull request #117 from PROCOLLAB-github/feature/chats-files

Added route for all files in a chat

Author: yakser

chats/models.py

@@ -297,13 +297,13 @@ class FileToMessage(models.Model):
     direct_message = models.ForeignKey(
         DirectChatMessage,
         on_delete=models.CASCADE,
-        related_name="file_to_direct_message",
+        related_name="file_to_message",
         null=True,
     )
     project_message = models.ForeignKey(
         ProjectChatMessage,
         on_delete=models.CASCADE,
-        related_name="file_to_direct_message",
+        related_name="file_to_message",
         null=True,
     )
 

chats/urls.py

@@ -7,23 +7,35 @@
     ProjectChatMessageList,
     ProjectChatDetail,
     DirectChatDetail,
+    ProjectChatFileList,
+    DirectChatFileList,
 )
 
 app_name = "chats"
 
 urlpatterns = [
     path("directs/", DirectChatList.as_view(), name="direct-chat-list"),
     path("directs/<slug:pk>/", DirectChatDetail.as_view(), name="direct-chat-detail"),
-    path("projects/", ProjectChatList.as_view(), name="project-chat-list"),
-    path("projects/<int:pk>/", ProjectChatDetail.as_view(), name="project-chat-detail"),
     path(
         "directs/<slug:pk>/messages/",
         DirectChatMessageList.as_view(),
         name="direct-chat-messages",
     ),
+    path(
+        "directs/<slug:pk>/files/",
+        DirectChatFileList.as_view(),
+        name="direct-chat-files",
+    ),
+    path("projects/", ProjectChatList.as_view(), name="project-chat-list"),
+    path("projects/<int:pk>/", ProjectChatDetail.as_view(), name="project-chat-detail"),
     path(
         "projects/<int:pk>/messages/",
         ProjectChatMessageList.as_view(),
         name="project-chat-messages",
     ),
+    path(
+        "projects/<int:pk>/files/",
+        ProjectChatFileList.as_view(),
+        name="project-chat-files",
+    ),
 ]

chats/utils.py

@@ -115,3 +115,15 @@ async def match_files_and_messages(file_urls, messages):
             project_message=messages["project_message"],
             file=file,
         )
+
+
+def get_all_files(messages):
+    # looks like something bad
+    files = []
+    for message in messages:
+        files_in_message = [
+            file_to_message.file for file_to_message in message.file_to_message.all()
+        ]
+        files.extend(files_in_message)
+
+    return files

chats/views.py

@@ -1,13 +1,10 @@
 from django.contrib.auth import get_user_model
 
-# from django.db import transaction
 from rest_framework import status
 from rest_framework.generics import (
     ListAPIView,
     ListCreateAPIView,
     RetrieveAPIView,
-    # GenericAPIView,
-    # get_object_or_404,
 )
 
 from rest_framework.permissions import IsAuthenticated
@@ -23,8 +20,8 @@
     ProjectChatDetailSerializer,
     DirectChatDetailSerializer,
 )
-
-# from files.helpers import FileAPI, get_file_data
+from chats.utils import get_all_files
+from files.serializers import UserFileSerializer
 
 User = get_user_model()
 
@@ -136,3 +133,23 @@ def post(self, request, *args, **kwargs):
         self.perform_create(serializer)
         headers = self.get_success_headers(serializer.data)
         return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
+
+
+class DirectChatFileList(ListCreateAPIView):
+    serializer_class = UserFileSerializer
+    permission_classes = [IsAuthenticated]
+
+    def get_queryset(self):
+        messages = self.request.user.direct_chats.get(id=self.kwargs["pk"]).messages.all()
+
+        return get_all_files(messages)
+
+
+class ProjectChatFileList(ListCreateAPIView):
+    serializer_class = UserFileSerializer
+    permission_classes = [IsAuthenticated]
+
+    def get_queryset(self):
+        messages = ProjectChat.objects.get(id=self.kwargs["pk"]).messages.all()
+
+        return get_all_files(messages)