from rest_framework import serializers
from .models import *
from exercise.models import Exercise
from exercise.serializers import ExerciseSerializer

class SupersetExerciseSerializer(serializers.ModelSerializer):
    exercise = serializers.SerializerMethodField()

    class Meta:
        model = SupersetExercise
        fields = '__all__'
    
    def get_exercise(self, obj): 
        data = ExerciseSerializer(obj.exercise, many=False).data
        return data

class SupersetSerializer(serializers.ModelSerializer):
    exercises = serializers.SerializerMethodField()
    
    class Meta:
        model = Superset
        fields = '__all__'
    
    def get_exercises(self, obj): 
        objs = SupersetExercise.objects.filter(superset=obj).order_by('order')
        data = SupersetExerciseSerializer(objs, many=True).data
        return data