1 files changed, 38 insertions, 0 deletions

diff --git a/chat_api/serializers.py b/chat_api/serializers.py
index 585bb36..15828be 100644
--- a/chat_api/serializers.py
+++ b/chat_api/serializers.py
@@ -15,3 +15,41 @@ class GroupSerializer(serializers.ModelSerializer):
         model = Group
         fields = '__all__'
         extra_fields = ('is_mine', )
+
+class DmSerializer(serializers.ModelSerializer):
+    is_mine = serializers.SerializerMethodField('_is_mine')
+
+    def _is_mine(self, obj):
+        request = self.context.get('request')
+        if request:
+            return request.user == obj.sender
+        return False
+    class Meta:
+        model = Dm
+        fields = '__all__'
+
+class GmSerializer(serializers.ModelSerializer):
+    is_mine = serializers.SerializerMethodField('_is_mine')
+
+    def _is_mine(self, obj):
+        request = self.context.get('request')
+        if request:
+            return request.user == obj.sender
+        return False
+    class Meta:
+        model = Gm
+        fields = '__all__'
+
+class GroupMessagesSerializer(serializers.ModelSerializer):
+    messages = serializers.SerializerMethodField('_messages')
+
+    def _messages(self, obj):
+        request = self.context.get('request')
+        if request:
+            gms = GmSerializer(obj.receiver_gm.all(), many=True, read_only=True, context={"request": request})
+            return gms.data
+        return []
+
+    class Meta:
+        model=Group
+        fields=('messages', )