From 0843bf2af4dc02e6041e56d5b33bf38b3aedfbd3 Mon Sep 17 00:00:00 2001 From: Trey t Date: Tue, 4 Jul 2023 12:16:19 -0500 Subject: [PATCH] WIP --- workout/serializers.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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()