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()