From dda7b15e237caa2c9956d503d674757bb969de2b Mon Sep 17 00:00:00 2001 From: Trey t Date: Sun, 9 Jul 2023 09:45:15 -0500 Subject: [PATCH] WIP --- workout/serializers.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/workout/serializers.py b/workout/serializers.py index 01bec54..0f4c5ce 100644 --- a/workout/serializers.py +++ b/workout/serializers.py @@ -39,6 +39,7 @@ class WorkoutSerializer(serializers.ModelSerializer): registered_user = GetRegisteredUserSerializer(many=False, read_only=True) muscles = serializers.SerializerMethodField() equipment = serializers.SerializerMethodField() + exercise_count = serializers.SerializerMethodField() class Meta: model = Workout @@ -54,6 +55,9 @@ class WorkoutSerializer(serializers.ModelSerializer): 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)) + + def get_exercise_count(self, obj): + return WorkoutExercise.objects.filter(workout=obj).count() class WorkoutDetailSerializer(serializers.ModelSerializer): exercises = serializers.SerializerMethodField()