From 10620ac8813412c833cb74b4c7895ea67fe5aee4 Mon Sep 17 00:00:00 2001 From: Kuroshini Date: Fri, 31 Jan 2020 00:16:08 +0300 Subject: [PATCH] fix set uploaded files #2 --- apps/establishment/serializers/back.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/apps/establishment/serializers/back.py b/apps/establishment/serializers/back.py index 64d2c655..851bf5d4 100644 --- a/apps/establishment/serializers/back.py +++ b/apps/establishment/serializers/back.py @@ -748,21 +748,21 @@ class MenuBackOfficeSerializer(serializers.ModelSerializer): def update(self, instance, validated_data): uploaded_files_ids = validated_data.pop('uploads') if len(uploaded_files_ids): - uploaded_files = list(models.MenuFiles.objects.filter(id__in=uploaded_files_ids)) + uploaded_files_ids = list(models.MenuFiles.objects.filter(id__in=uploaded_files_ids)) establishment_kwargs = validated_data.pop('establishment') establishment = get_object_or_404(models.Establishment, **establishment_kwargs) validated_data['establishment'] = establishment instance = super().update(instance, validated_data) - instance.uploads.set(uploaded_files) + instance.uploads.set(uploaded_files_ids) return instance def create(self, validated_data): uploaded_files_ids = validated_data.pop('uploads') if len(uploaded_files_ids): - uploaded_files = list(models.MenuFiles.objects.filter(id__in=uploaded_files_ids)) + uploaded_files_ids = list(models.MenuFiles.objects.filter(id__in=uploaded_files_ids)) establishment_kwargs = validated_data.pop('establishment') establishment = get_object_or_404(models.Establishment, **establishment_kwargs) validated_data['establishment'] = establishment instance = models.Menu.objects.create(**validated_data) - instance.uploads.set(uploaded_files) + instance.uploads.set(uploaded_files_ids) return instance