diff --git a/backend/secfit/workouts/serializers.py b/backend/secfit/workouts/serializers.py index 5d5a3ce5ad8867dbc1a80a7d7403f309cbc43a34..9456c0f6ceab146f3921bfcb7b2c94d4656c6a99 100644 --- a/backend/secfit/workouts/serializers.py +++ b/backend/secfit/workouts/serializers.py @@ -172,21 +172,21 @@ class WorkoutSerializer(serializers.HyperlinkedModelSerializer): return instance def handle_workout_files(self, files, files_data): - for file, file_data in zip(files.all(), files_data): - file.file = file_data.get("file", file.file) - - # If new files have been added, creating new WorkoutFiles - if len(files_data) > len(files.all()): - for i in range(len(files.all()), len(files_data)): - WorkoutFile.objects.create( - workout=instance, - owner=instance.owner, - file=files_data[i].get("file"), - ) - # Else if files have been removed, delete WorkoutFiles - elif len(files_data) < len(files.all()): - for i in range(len(files_data), len(files.all())): - files.all()[i].delete() + for file, file_data in zip(files.all(), files_data): + file.file = file_data.get("file", file.file) + + # If new files have been added, creating new WorkoutFiles + if len(files_data) > len(files.all()): + for i in range(len(files.all()), len(files_data)): + WorkoutFile.objects.create( + workout=instance, + owner=instance.owner, + file=files_data[i].get("file"), + ) + # Else if files have been removed, delete WorkoutFiles + elif len(files_data) < len(files.all()): + for i in range(len(files_data), len(files.all())): + files.all()[i].delete() def get_owner_username(self, obj): """Returns the owning user's username