From f9d7137d5e3f280171f21af399d02b9782baa3ca Mon Sep 17 00:00:00 2001 From: Ivlev Denis Date: Sun, 22 Apr 2018 20:53:47 +0300 Subject: [PATCH] Fix 500 error when save gallery --- api/v1/serializers/mixins.py | 38 +++++++++++++++++++----------------- 1 file changed, 20 insertions(+), 18 deletions(-) diff --git a/api/v1/serializers/mixins.py b/api/v1/serializers/mixins.py index 52f1aedc..e70d69e4 100644 --- a/api/v1/serializers/mixins.py +++ b/api/v1/serializers/mixins.py @@ -170,15 +170,16 @@ class DispatchContentMixin(object): g.save() if 'images' in cdata: for image in cdata['images']: - if 'id' in image and image['id']: - gi = GalleryImage.objects.get(id=image['id']) - gi.img = ImageObject.objects.get(id=image['img']) - gi.save() - else: - gi = GalleryImage.objects.create( - gallery=g, - img=ImageObject.objects.get(id=image['img']) - ) + if 'img' in image and image['img']: + if 'id' in image and image['id']: + gi = GalleryImage.objects.get(id=image['id']) + gi.img = ImageObject.objects.get(id=image['img']) + gi.save() + else: + gi = GalleryImage.objects.create( + gallery=g, + img=ImageObject.objects.get(id=image['img']) + ) else: g = Gallery( position=cdata['position'], @@ -190,15 +191,16 @@ class DispatchContentMixin(object): g.save() if 'images' in cdata: for image in cdata['images']: - if 'id' in image and image['id']: - gi = GalleryImage.objects.get(id=image['id']) - gi.img = ImageObject.objects.get(id=image['img']) - gi.save() - else: - gi = GalleryImage.objects.create( - gallery=g, - img=ImageObject.objects.get(id=image['img']) - ) + if 'img' in image and image['img']: + if 'id' in image and image['id']: + gi = GalleryImage.objects.get(id=image['id']) + gi.img = ImageObject.objects.get(id=image['img']) + gi.save() + else: + gi = GalleryImage.objects.create( + gallery=g, + img=ImageObject.objects.get(id=image['img']) + ) class DispatchMaterialMixin(object):