From 6e1073dd2c760a8ef2dcae4fbf19a8b301587b6b Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Mon, 12 Feb 2018 09:46:07 +0300
Subject: [PATCH] Add type to serializers

---
 api/v1/serializers/content.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/api/v1/serializers/content.py b/api/v1/serializers/content.py
index e1159b2e..3743371f 100644
--- a/api/v1/serializers/content.py
+++ b/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):