WIP

workout/admin.py

@@ -17,7 +17,7 @@ class SupersetInline(admin.StackedInline):
 
 @admin.register(Workout)
 class WorkoutAdmin(ImportExportModelAdmin):
-    list_display = ("name", "description", "registered_user", "created_at", "updated_at")
+    list_display = ("name", "description", "estimated_time", "registered_user", "created_at", "updated_at")
     inlines = [
         SupersetInline,
     ]

workout/migrations/0012_workout_estimated_time.py

@@ -0,0 +1,18 @@
+# Generated by Django 4.2.2 on 2023-07-25 16:43
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('workout', '0011_alter_completedworkout_notes'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='workout',
+            name='estimated_time',
+            field=models.FloatField(blank=True, max_length=255, null=True),
+        ),
+    ]

workout/models.py

@@ -20,6 +20,7 @@ class Workout(models.Model):
       RegisteredUser,
       on_delete=models.CASCADE
     )
+    estimated_time = models.FloatField(max_length=255, blank=True, null=True)
 
     def __str__(self):
         return self.name #+ " : " + self.description + " | by: " + self.registered_user.nick_name

workout/serializers.py

@@ -82,7 +82,7 @@ class WorkoutDetailSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = Workout
-        fields = ('id', 'name', 'description', 'supersets', 'registered_user', 'male_videos', 'female_videos', 'both_videos')
+        fields = ('id', 'name', 'description', 'supersets', 'registered_user', 'male_videos', 'female_videos', 'both_videos', 'estimated_time', )
         depth = 1
    
     def get_supersets(self, obj): 

workout/urls.py

@@ -10,4 +10,6 @@ urlpatterns = [
    path('create/', views.add_workout, name='create new workout'),
    path('planned_workouts/', views.workouts_planned_by_logged_in_user, name='planned workout for user'),
    path('plan_workout/', views.plan_workout, name='plan workout'),
+
+   path('add_from_files/', views.add_from_files, name='plan workout'),
 ]

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)