From 4fde19d23e4fe143b34954f78ecb1df2a591bf7d Mon Sep 17 00:00:00 2001
From: Kuroshini <o.hayatov@id-east.ru>
Date: Wed, 29 Jan 2020 16:53:04 +0300
Subject: [PATCH] fix partner transfer

---
 apps/transfer/serializers/partner.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/apps/transfer/serializers/partner.py b/apps/transfer/serializers/partner.py
index 1a41a5cc..9d79e00c 100644
--- a/apps/transfer/serializers/partner.py
+++ b/apps/transfer/serializers/partner.py
@@ -1,7 +1,7 @@
 from rest_framework import serializers
 
 from establishment.models import Establishment
-from partner.models import Partner
+from partner.models import Partner, PartnerToEstablishment
 
 
 class PartnerSerializer(serializers.Serializer):
@@ -44,12 +44,23 @@ class PartnerSerializer(serializers.Serializer):
 
     def create(self, validated_data):
         establishment = validated_data.pop('establishment')
-        obj, _ = Partner.objects.update_or_create(
+        url = validated_data.pop('url')
+        image = validated_data.pop('image')
+        obj, is_created = Partner.objects.update_or_create(
             old_id=validated_data['old_id'],
             defaults=validated_data,
         )
         obj.establishment.set([establishment])
+        if obj.created:
+            obj.images = [image]
+        else:
+            obj.images.append(image)
         obj.save()
+
+        p_t_e = PartnerToEstablishment.objects.filter(establishment=establishment, partner=obj).first()
+        p_t_e.url = url
+        p_t_e.image = image
+        p_t_e.save()
         return obj