fix transfer partners

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

View File

@ -10,7 +10,7 @@ def delete_unused_partners(apps, schema_editor):
ids_to_preserve = []
for p_t_e in PartnerToEstablishment.objects.all():
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):

View File

@ -43,10 +43,13 @@ class PartnerSerializer(serializers.Serializer):
return establishment
def create(self, validated_data):
establishment = validated_data.pop('establishment')
obj, _ = Partner.objects.update_or_create(
old_id=validated_data['old_id'],
defaults=validated_data,
)
obj.establishment.set([establishment])
obj.save()
return obj