diff --git a/workout/serializers.py b/workout/serializers.py index 884b878..622235e 100644 --- a/workout/serializers.py +++ b/workout/serializers.py @@ -46,14 +46,14 @@ class WorkoutSerializer(serializers.ModelSerializer): # depth = 1 def get_muscles(self, obj): - exercises = WorkoutExercise.objects.filter(workout=obj).values_list('id') + exercises = WorkoutExercise.objects.filter(workout=obj).values_list('exercise__id') muscles_names = ExerciseMuscle.objects.filter(exercise__id__in=exercises).values_list('muscle__name', flat=True) return list(set(muscles_names)) def get_equipment(self, obj): - exercises = WorkoutExercise.objects.filter(workout=obj).values_list('id') - muscles_names = WorkoutEquipment.objects.filter(exercise__id__in=exercises).values_list('equipment__name', flat=True) - return list(set(muscles_names)) + exercises = WorkoutExercise.objects.filter(workout=obj).values_list('exercise__id') + equipment_names = WorkoutEquipment.objects.filter(exercise__id__in=exercises).values_list('equipment__name', flat=True) + return list(set(equipment_names)) class WorkoutDetailSerializer(serializers.ModelSerializer): exercises = serializers.SerializerMethodField()