From b147040c53513ef76ab140faa86592778e4341d9 Mon Sep 17 00:00:00 2001
From: Trey t <treytartt@fastmail.com>
Date: Tue, 4 Jul 2023 11:46:55 -0500
Subject: [PATCH] WIP

---
 db.sqlite3             | Bin 2052096 -> 2052096 bytes
 workout/serializers.py |  16 ++++++++++++++--
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/db.sqlite3 b/db.sqlite3
index 787a66e00c5bcb99fdf3199bbd81b6df6a9c59ad..6e1f5bfa73a2567a8f53bc1956b18cc3a56f852d 100644
GIT binary patch
delta 673
zcmZwDziSg=7zgls?@e<*Ozyc>?VY(cNhwtvnq1!ZdOsq9(LrnR$52Yfg2`}2YN&-y
zrI8MugcOupJGv-HhTB5_06_-_-J;--E-t!65T7H}A@+gK`wkD!m*;)kN4xE#-Ht_i
zhcmZG@9lcKM0od0Ng~ADACwOUWfBj{qSevU&==qroWM8e!%z6$ojiUAYKkhvOb#PD
zc`xQ@P98?m2eBe4*X5xc%F&#z-su+Cze-W~Nj%)EaYd-r2$sNX#usgG(RPY9uQFcc
zPRVyY#+<V|zFTo1wP4<4qX_Z82v)U&k{2vF%pHq3?fA0o1*$^(M*QrvxIe9@wY&fi
z;Wbb#);6>}K8k32@QPd$lDcb3xb3WBjm3?2U2=RsIJ4t&*R$_8cZ!YX(?`j@+1yz%
zlB{{f7@HWa<}>eK*8JZuIc>R|SCXKuu(cgN*<1~`A2;Lpw|I}pIasE!1<Nm0dR2V@
zJ$M1n;Wor`H<ZhyTRi@ljRK&0c7hVJz|&MDw9le7pl@)rA;1uRM0%fE(G{IeNgL3B
zI@IwKOo-|Qn#m+{CQnX-9kdD*rO*UQBMHe!K`J_jGAN5Q1V~4dXbPQ27tlpCjSOU>
SODKmdG=pYQK5i8X9s3XQQKoeO

delta 364
zcmXxcu}cDB7{~E@@4Y;=^PX2ryFBxBDOxngXbBpsL1+@kTiv8xP#l~B4~=&V$CiRP
z@ZG{cAgH0C2BF$?X{n_k`X%&y`D_or_Ndh!wfZg_2e|__zVX{7RtaBACS#fItlXQG
znVOY#w=WEuYxGTX`lNT7(lks??kJwu{!h;D^(=Q{9ce!36*JmcB#R<`6c+q9Q#PLT
zU{LUTs`IROGs3WyMkjPd+*WqlPGgVR23@kfLDIia>PN4#h(k3_r{@;mwr*)iXLPto
z758JCEG$eO6LOch2&xHLBYquqsHxExeMs@bUAZNA%B<5O%sV9O9=C#HtJm~67_bBp
qFd+&th{H1c{UyK#0s%=#!3wNG8Zxj34y;2KT*$!&<W<x2`a8cKt86v^

diff --git a/workout/serializers.py b/workout/serializers.py
index 5ac12e9..884b878 100644
--- a/workout/serializers.py
+++ b/workout/serializers.py
@@ -2,6 +2,8 @@ from rest_framework import serializers
 from .models import *
 from exercise.serializers import ExerciseSerializer
 from registered_user.serializers import GetRegisteredUserSerializer
+from muscle.models import ExerciseMuscle
+from equipment.models import WorkoutEquipment
 
 class WorkoutExerciseSerializer(serializers.ModelSerializer):
     exercise = ExerciseSerializer(read_only=True)
@@ -12,8 +14,6 @@ class WorkoutExerciseSerializer(serializers.ModelSerializer):
         model = WorkoutExercise
         fields = ('workout', 'exercise','weight','reps','duration','duration_audio','weight_audio', 'created_at',)
 
-
-
 class CompleteWorkoutSerializer(serializers.ModelSerializer):
     class Meta:
         model = CompletedWorkout
@@ -37,11 +37,23 @@ class CompleteWorkoutSerializer(serializers.ModelSerializer):
 class WorkoutSerializer(serializers.ModelSerializer):
     # exercise = WorkoutExerciseSerializer(source='workout_exercise_workout', many=True)
     registered_user = GetRegisteredUserSerializer(many=False, read_only=True)
+    muscles = serializers.SerializerMethodField()
+    equipment = serializers.SerializerMethodField()
 
     class Meta:
         model = Workout
         fields = '__all__'
         # depth = 1
+    
+    def get_muscles(self, obj): 
+        exercises = WorkoutExercise.objects.filter(workout=obj).values_list('id')
+        muscles_names = ExerciseMuscle.objects.filter(exercise__id__in=exercises).values_list('muscle__name', flat=True)
+        return list(set(muscles_names))
+    
+    def get_equipment(self, obj): 
+        exercises = WorkoutExercise.objects.filter(workout=obj).values_list('id')
+        muscles_names = WorkoutEquipment.objects.filter(exercise__id__in=exercises).values_list('equipment__name', flat=True)
+        return list(set(muscles_names))
 
 class WorkoutDetailSerializer(serializers.ModelSerializer):
     exercises = serializers.SerializerMethodField()