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