Add type to serializers

api/v1/serializers/content.py

@@ -55,6 +55,7 @@ class ImageObjectSerializer(serializers.ModelSerializer):
 
 
 class ImageCreateSerializer(serializers.ModelSerializer):
+    type = serializers.SerializerMethodField()
 
     class Meta:
         model = Image
@@ -65,22 +66,28 @@ class ImageCreateSerializer(serializers.ModelSerializer):
             'title',
             'position',
             'img',
+            'type',
             'created_at',
             'update_at',
         )
 
         read_only_fields = (
             'id',
+            'type',
             'created_at',
             'update_at',
         )
 
+    def get_type(self, object):
+        return 'image'
+
 
 class ImageSerializer(ImageCreateSerializer):
     img = ImageObjectSerializer()
 
 
 class TextCreateSerializer(serializers.ModelSerializer):
+    type = serializers.SerializerMethodField()
 
     class Meta:
         model = Text
@@ -100,12 +107,16 @@ class TextCreateSerializer(serializers.ModelSerializer):
             'update_at',
         )
 
+    def get_type(self, object):
+        return 'text'
+
 
 class TextSerializer(TextCreateSerializer):
     pass
 
 
 class ImageTextCreateSerializer(serializers.ModelSerializer):
+    type = serializers.SerializerMethodField()
 
     class Meta:
         model = ImageText
@@ -127,12 +138,16 @@ class ImageTextCreateSerializer(serializers.ModelSerializer):
             'update_at',
         )
 
+    def get_type(self, object):
+        return 'image-text'
+
 
 class ImageTextSerializer(ImageTextCreateSerializer):
     img = ImageObjectSerializer()
 
 
 class VideoCreateSerializer(serializers.ModelSerializer):
+    type = serializers.SerializerMethodField()
 
     class Meta:
         model = Video
@@ -152,6 +167,9 @@ class VideoCreateSerializer(serializers.ModelSerializer):
             'update_at',
         )
 
+    def get_type(self, object):
+        return 'video'
+
 
 class VideoSerializer(VideoCreateSerializer):
     pass
@@ -181,6 +199,7 @@ class GalleryImageSerializer(GalleryImageCreateSerializer):
 
 
 class GallerySerializer(serializers.ModelSerializer):
+    type = serializers.SerializerMethodField()
     gallery_images = GalleryImageSerializer(many=True)
 
     class Meta:
@@ -202,6 +221,9 @@ class GallerySerializer(serializers.ModelSerializer):
             'update_at',
         )
 
+    def get_type(self, object):
+        return 'images'
+
 
 class ContentSerializer(serializers.ModelSerializer):