This commit is contained in:
Trey t
2023-07-09 09:45:15 -05:00
parent 887c83e421
commit dda7b15e23

View File

@@ -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
@@ -55,6 +56,9 @@ class WorkoutSerializer(serializers.ModelSerializer):
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()
registered_user = serializers.SerializerMethodField()