WIP

workout/views.py

@@ -26,8 +26,8 @@ def all_workouts(request):
 
 @api_view(['GET'])
 def workout_details(request, workout_id):
-    users = Workout.objects.get(pk=workout_id)
-    serializer = WorkoutDetailSerializer(users, many=False)
+    workout = Workout.objects.get(pk=workout_id)
+    serializer = WorkoutDetailSerializer(workout, many=False)
     return Response(data=serializer.data, status=status.HTTP_200_OK)
 
 @api_view(['POST'])
@@ -73,6 +73,7 @@ def add_workout(request):
         workout = serializer.save(registered_user=registered_user)
         workout.save()
 
+        workout_total_time = 0
         for superset in exercise_data:
             name = superset["name"]
             rounds = superset["rounds"]
@@ -87,6 +88,7 @@ def add_workout(request):
             )
             superset.save()
       
+            superset_total_time = 0
             for exercise in exercises:
                 exercise_id = exercise["id"]
                 exercise_obj = Exercise.objects.get(pk=exercise_id)
@@ -102,12 +104,22 @@ def add_workout(request):
                     supersetExercise.weight = exercise["weight"]
                 if "reps" in exercise:
                     supersetExercise.reps = exercise["reps"]
+                    superset_total_time += exercise["reps"] * exercise_obj.estimated_rep_duration
                 if "duration" in exercise:
                     supersetExercise.duration = exercise["duration"]
+                    superset_total_time += exercise["duration"]
+
                 supersetExercise.save()
 
-            superset_order += 1
+            superset.estimated_time = superset_total_time
+            superset.save()
 
+            workout_total_time += (superset_total_time * rounds)
+
+            superset_order += 1
+        workout.estimated_time = workout_total_time
+        workout.save()
+        
         return Response(serializer.data, status=status.HTTP_201_CREATED)
     return Response(serializer.errors, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
@@ -132,3 +144,87 @@ def plan_workout(request):
         serializer.save()
         return Response(serializer.data, status=status.HTTP_201_CREATED)
     return Response(serializer.errors, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+@api_view(['GET'])
+def add_from_files(request):
+    with open("/Users/treyt/Desktop/code/werkout/werkout_api/workout/sample_workout.json") as user_file:
+        file_contents = user_file.read()
+        parsed_json = json.loads(file_contents)
+        
+        for item in parsed_json:
+            workout_name = item["name"]
+            workout_description = item["description"]
+            workout_created = item["created"]
+            
+            workout_obj = Workout.objects.create(
+                registered_user = RegisteredUser.objects.get(pk=2),
+                description = workout_description,
+                name = workout_name,
+                created_at = workout_created
+            )
+
+            workout_obj.save()
+            workout_obj.created_at = workout_created
+            workout_obj.save(update_fields=['created_at'])
+            workout_total_time = 0
+
+            supersets = item["supersets"]
+            superset_order = 1
+            for superset in supersets:
+                superset_name = superset["name"]
+                superset_rounds = superset["rounds"]
+
+                superset_obj = Superset.objects.create(
+                    workout=workout_obj, 
+                    name=superset_name,
+                    rounds=superset_rounds,
+                    order=superset_order
+                )
+                
+                superset_obj.save()
+            
+                superset_order += 1
+
+                exercises = superset["exercises"]
+                exercise_order = 1
+
+                superset_total_time = 0
+                for exercise in exercises:
+                    side = exercise["side"]
+                    name = exercise["name"]
+                    
+                    duration = exercise["duration"]
+                    reps = exercise["reps"]
+                    side = exercise["side"]
+
+                    exercise_obj = None
+                    if len(side) > 0:
+                        exercise_obj = Exercise.objects.get(name=name, side=side)
+                    else:
+                        exercise_obj = Exercise.objects.get(name=name, side="")
+                    
+                    supersetExercise = SupersetExercise.objects.create(
+                        superset=superset_obj, 
+                        exercise=exercise_obj,
+                        order=exercise_order
+                    )
+                    
+                    if reps != 0:
+                        supersetExercise.reps = reps
+                        superset_total_time += reps * exercise_obj.estimated_rep_duration
+                    if reps == 0 and duration != 0:
+                        supersetExercise.duration = duration
+                        superset_total_time += exercise["duration"]
+                    supersetExercise.save()
+
+                    exercise_order += 1
+
+                superset_obj.estimated_time = superset_total_time
+                superset_obj.save()
+
+                workout_total_time += (superset_total_time * superset_rounds)
+                    
+            workout_obj.estimated_time = workout_total_time
+            workout_obj.save()
+
+        return Response(status=status.HTTP_200_OK)