23 lines
833 B
Python
23 lines
833 B
Python
from rest_framework import serializers
|
|
from .models import *
|
|
from muscle.models import ExerciseMuscle
|
|
from equipment.models import WorkoutEquipment
|
|
from muscle.serializers import ExerciseMuscleSerializer
|
|
|
|
class ExerciseSerializer(serializers.ModelSerializer):
|
|
muscles = serializers.SerializerMethodField()
|
|
equipment = serializers.SerializerMethodField()
|
|
audio_url = serializers.ReadOnlyField()
|
|
video_url = serializers.ReadOnlyField()
|
|
|
|
class Meta:
|
|
model = Exercise
|
|
fields = '__all__'
|
|
|
|
def get_muscles(self, obj):
|
|
objs = ExerciseMuscle.objects.filter(exercise=obj).values_list('muscle__name', flat=True)
|
|
return objs
|
|
|
|
def get_equipment(self, obj):
|
|
objs = WorkoutEquipment.objects.filter(exercise=obj).values_list('equipment__name', flat=True)
|
|
return objs |