28 lines
850 B
Python
28 lines
850 B
Python
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
|