phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix partner transfer

Browse Source
This commit is contained in:
Kuroshini 2020-01-29 16:53:04 +03:00
parent 28d4a5ae42
commit 4fde19d23e
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/transfer/serializers/partner.py
View File

@ -1,7 +1,7 @@
from rest_framework import serializers from rest_framework import serializers
from establishment.models import Establishment from establishment.models import Establishment
from partner.models import Partner from partner.models import Partner, PartnerToEstablishment
class PartnerSerializer(serializers.Serializer): class PartnerSerializer(serializers.Serializer):
@ -44,12 +44,23 @@ class PartnerSerializer(serializers.Serializer):
def create(self, validated_data): def create(self, validated_data):
establishment = validated_data.pop('establishment') 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'], old_id=validated_data['old_id'],
defaults=validated_data, defaults=validated_data,
) )
obj.establishment.set([establishment]) obj.establishment.set([establishment])
if obj.created:
obj.images = [image]
else:
obj.images.append(image)
obj.save() 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 return obj