WIP

workout/serializers.py

@@ -4,6 +4,8 @@ from exercise.serializers import ExerciseSerializer
 from registered_user.serializers import GetRegisteredUserSerializer
 from muscle.models import ExerciseMuscle
 from equipment.models import WorkoutEquipment
+from video.models import Video
+from video.serializers import VideoSerializer
 
 class WorkoutExerciseSerializer(serializers.ModelSerializer):
     exercise = ExerciseSerializer(read_only=True)
@@ -62,10 +64,13 @@ class WorkoutSerializer(serializers.ModelSerializer):
 class WorkoutDetailSerializer(serializers.ModelSerializer):
     exercises = serializers.SerializerMethodField()
     registered_user = serializers.SerializerMethodField()
+    male_videos = serializers.SerializerMethodField()
+    female_videos = serializers.SerializerMethodField()
+    both_videos = serializers.SerializerMethodField()
 
     class Meta:
         model = Workout
-        fields = ('id', 'name', 'description', 'exercises', 'registered_user')
+        fields = ('id', 'name', 'description', 'exercises', 'registered_user', 'male_videos', 'female_videos', 'both_videos')
         depth = 1
    
     def get_exercises(self, obj): 
@@ -77,6 +82,18 @@ class WorkoutDetailSerializer(serializers.ModelSerializer):
         objs = RegisteredUser.objects.get(pk=obj.registered_user.pk)
         data = GetRegisteredUserSerializer(objs, many=False).data
         return data
+    
+    def get_male_videos(self, obj):
+        data = Video.objects.filter(gender=1).values_list('video_file', flat=True)
+        return data
+    
+    def get_female_videos(self, obj):
+        data = Video.objects.filter(gender=2).values_list('video_file', flat=True)
+        return data
+    
+    def get_both_videos(self, obj):
+        data = Video.objects.filter(gender=3).values_list('video_file', flat=True)
+        return data
 
 class GetCompleteWorkoutSerializer(serializers.ModelSerializer):
     workout = WorkoutSerializer(many=False, read_only=True)