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

fix transfer partners

Browse Source
This commit is contained in:
Kuroshini 2020-01-28 21:18:25 +03:00
parent d2e28f519a
commit 24f4abc76d
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/partner/migrations/0005_auto_20200128_1754.py
View File

@ -10,7 +10,7 @@ def delete_unused_partners(apps, schema_editor):
ids_to_preserve = [] ids_to_preserve = []
for p_t_e in PartnerToEstablishment.objects.all(): for p_t_e in PartnerToEstablishment.objects.all():
ids_to_preserve.append(p_t_e.partner.pk) ids_to_preserve.append(p_t_e.partner.pk)
PartnerToEstablishment.objects.exclude(id__in=ids_to_preserve).delete() Partner.objects.exclude(id__in=ids_to_preserve).delete()
class Migration(migrations.Migration): class Migration(migrations.Migration):

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

@ -43,10 +43,13 @@ class PartnerSerializer(serializers.Serializer):
return establishment return establishment
def create(self, validated_data): def create(self, validated_data):
establishment = validated_data.pop('establishment')
obj, _ = Partner.objects.update_or_create( obj, _ = 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.save()
return obj return obj