From 6f70383649f4f46ab9c3a0ef24ffc9def1bbe5dd Mon Sep 17 00:00:00 2001
From: Ivlev Denis <di-erz@yandex.ru>
Date: Fri, 9 Feb 2018 18:46:52 +0300
Subject: [PATCH] Fix ImageObject structure

---
 api/v1/serializers/course.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/api/v1/serializers/course.py b/api/v1/serializers/course.py
index 7d7c8053..ba6d73c7 100644
--- a/api/v1/serializers/course.py
+++ b/api/v1/serializers/course.py
@@ -138,14 +138,14 @@ class CourseCreateSerializer(serializers.ModelSerializer):
                     i.position = c['data']['position']
                     i.title = c['data']['title']
                     i.course = course
-                    i.img = ImageObject.objects.get(id=c['data']['img']['id'])
+                    i.img = ImageObject.objects.get(id=c['data']['img'])
                     i.save()
                 else:
                     i = Image.objects.create(
                         position=c['data']['position'],
                         title=c['data']['title'],
                         course=course,
-                        img=Image.objects.get(id=c['data']['img']['id']),
+                        img=ImageObject.objects.get(id=c['data']['img']),
                     )
             elif c['type'] == 'image-text':
                 if 'id' in c['data'] and c['data']['id']:
@@ -153,7 +153,7 @@ class CourseCreateSerializer(serializers.ModelSerializer):
                     it.position = c['data']['position']
                     it.title = c['data']['title']
                     it.course = course
-                    it.img = ImageObject.objects.get(id=c['data']['img']['id'])
+                    it.img = ImageObject.objects.get(id=c['data']['img'])
                     it.txt = c['data']['txt']
                     it.save()
                 else:
@@ -161,7 +161,7 @@ class CourseCreateSerializer(serializers.ModelSerializer):
                         position=c['data']['position'],
                         title=c['data']['title'],
                         course=course,
-                        img=Image.objects.get(id=c['data']['img']['id']),
+                        img=ImageObject.objects.get(id=c['data']['img']),
                         txt=c['data']['txt'],
                     )
             elif c['type'] == 'video':
@@ -222,22 +222,22 @@ class CourseCreateSerializer(serializers.ModelSerializer):
                     i.position = c['data']['position']
                     i.title = c['data']['title']
                     i.course = course
-                    i.img = Image.objects.get(id=c['data']['img']['id'])
+                    i.img = ImageObject.objects.get(id=c['data']['img'])
                     i.save()
                 else:
                     i = Image.objects.create(
                         position=c['data']['position'],
                         title=c['data']['title'],
                         course=course,
-                        img=Image.objects.get(id=c['data']['img']['id']),
+                        img=ImageObject.objects.get(id=c['data']['img']),
                     )
-            elif c['type'] == 'imagetext':
+            elif c['type'] == 'image-text':
                 if 'id' in c['data'] and c['data']['id']:
                     it = ImageText.objects.get(id=c['data']['id'])
                     it.position = c['data']['position']
                     it.title = c['data']['title']
                     it.course = course
-                    it.img = Image.objects.get(id=c['data']['img']['id'])
+                    it.img = ImageObject.objects.get(id=c['data']['img'])
                     it.txt = c['data']['txt']
                     it.save()
                 else:
@@ -245,7 +245,7 @@ class CourseCreateSerializer(serializers.ModelSerializer):
                         position=c['data']['position'],
                         title=c['data']['title'],
                         course=course,
-                        img=Image.objects.get(id=c['data']['img']['id']),
+                        img=ImageObject.objects.get(id=c['data']['img']),
                         txt=c['data']['txt'],
                     )
             elif c['type'] == 'video':