From 47f890583218f541cc837a765d9e6140907be4c5 Mon Sep 17 00:00:00 2001 From: Kuroshini Date: Fri, 31 Jan 2020 00:14:49 +0300 Subject: [PATCH] fix set uploaded files --- apps/establishment/serializers/back.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/apps/establishment/serializers/back.py b/apps/establishment/serializers/back.py index 0b82308f..64d2c655 100644 --- a/apps/establishment/serializers/back.py +++ b/apps/establishment/serializers/back.py @@ -747,18 +747,22 @@ 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)) 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_ids) + instance.uploads.set(uploaded_files) 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)) 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_ids) + instance.uploads.set(uploaded_files) return instance