WIP

workout/serializers.py

@@ -6,7 +6,7 @@ from muscle.models import ExerciseMuscle
 from equipment.models import WorkoutEquipment
 from video.models import Video
 from video.serializers import VideoSerializer
-from superset.serializers import SupersetSerializer
+from superset.serializers import SupersetSerializer, SupersetExerciseSerializer
 from superset.models import Superset, SupersetExercise
 
 class WorkoutExerciseSerializer(serializers.ModelSerializer):
@@ -79,12 +79,23 @@ class WorkoutDetailSerializer(serializers.ModelSerializer):
     male_videos = serializers.SerializerMethodField()
     female_videos = serializers.SerializerMethodField()
     both_videos = serializers.SerializerMethodField()
+    all_superset_exercise = serializers.SerializerMethodField()
 
     class Meta:
         model = Workout
-        fields = ('id', 'name', 'description', 'supersets', 'registered_user', 'male_videos', 'female_videos', 'both_videos', 'estimated_time', )
+        fields = ('id', 'name', 'description', 'supersets', 'registered_user', 'male_videos', 'female_videos', 'both_videos', 'estimated_time', 'all_superset_exercise', )
         depth = 1
    
+    def get_all_superset_exercise(self, obj): 
+        all_superset_exercise = []
+        supersets = Superset.objects.filter(workout=obj).order_by('order')
+        for superset in supersets:
+            supersetExercises = SupersetExercise.objects.filter(superset=superset).order_by('order')
+            for x in range(0, superset.rounds):
+                all_superset_exercise += supersetExercises
+        data = SupersetExerciseSerializer(all_superset_exercise, many=True).data
+        return data
+    
     def get_supersets(self, obj): 
         objs = Superset.objects.filter(workout=obj).order_by('order')
         data = SupersetSerializer(objs, many=True).data