added title & image address for list serializers in private & project chats

Author: VeryBigSad

chats/models.py

@@ -160,7 +160,7 @@ def get_chat(cls, user1, user2) -> "DirectChat":
     def get_last_messages(self, message_count):
         return self.messages.order_by("-created_at")[:message_count]
 
-    def get_other_user(self, user):
+    def get_other_user(self, user) -> User:
         return self.users.exclude(pk=user.pk).first()
 
     @classmethod

chats/serializers.py

@@ -22,13 +22,19 @@ def get_users(cls, chat: ProjectChat):
     def get_last_message(cls, chat: DirectChat):
         return DirectChatMessageListSerializer(chat.get_last_message()).data
 
+    def get_title(self, chat: DirectChat):
+        request = self.context.get("request")
+        user = request.user
+        return chat.get_other_user(user).get_full_name()
+
+    def get_image_address(self, chat: DirectChat):
+        request = self.context.get("request")
+        user = request.user
+        return chat.get_other_user(user).avatar
+
     class Meta:
         model = DirectChat
-        fields = [
-            "id",
-            "users",
-            "last_message",
-        ]
+        fields = ["id", "users", "last_message", "title", "image_address"]
 
 
 class DirectChatDetailSerializer(serializers.ModelSerializer):
@@ -53,9 +59,17 @@ class ProjectChatListSerializer(serializers.ModelSerializer):
     def get_last_message(cls, chat: ProjectChat):
         return ProjectChatMessageListSerializer(chat.get_last_message()).data
 
+    @classmethod
+    def get_image_address(cls, chat: ProjectChat):
+        return chat.project.image_address
+
+    @classmethod
+    def get_name(cls, chat: ProjectChat):
+        return chat.project.name
+
     class Meta:
         model = ProjectChat
-        fields = ["id", "project", "last_message"]
+        fields = ["id", "project", "last_message", "name", "image_address"]
 
 
 class ProjectChatDetailSerializer(serializers.ModelSerializer):