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