This commit is contained in:
Trey t
2023-07-04 12:16:19 -05:00
parent fc18231728
commit 0843bf2af4

View File

@@ -46,14 +46,14 @@ class WorkoutSerializer(serializers.ModelSerializer):
# depth = 1 # depth = 1
def get_muscles(self, obj): 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) muscles_names = ExerciseMuscle.objects.filter(exercise__id__in=exercises).values_list('muscle__name', flat=True)
return list(set(muscles_names)) return list(set(muscles_names))
def get_equipment(self, obj): def get_equipment(self, obj):
exercises = WorkoutExercise.objects.filter(workout=obj).values_list('id') exercises = WorkoutExercise.objects.filter(workout=obj).values_list('exercise__id')
muscles_names = WorkoutEquipment.objects.filter(exercise__id__in=exercises).values_list('equipment__name', flat=True) equipment_names = WorkoutEquipment.objects.filter(exercise__id__in=exercises).values_list('equipment__name', flat=True)
return list(set(muscles_names)) return list(set(equipment_names))
class WorkoutDetailSerializer(serializers.ModelSerializer): class WorkoutDetailSerializer(serializers.ModelSerializer):
exercises = serializers.SerializerMethodField() exercises = serializers.SerializerMethodField()