init commit

workout/tasks.py

@@ -0,0 +1,101 @@
+from celery import shared_task
+import json
+import os
+from .models import *
+from .serializers import *
+from django.core.cache import cache
+from superset.models import Superset, SupersetExercise
+from exercise.models import Exercise
+
+@shared_task()
+def add_from_files_tasks():
+    sample_urls = [{
+            "file": os.getcwd() + "/workout/trey_all_workouts.json",
+            "user_id": 2
+        },{
+            "file": os.getcwd() + "/workout/cho_all_workouts.json",
+            "user_id": 6
+        }]
+    
+    for sample_url in sample_urls:
+        with open(sample_url["file"]) 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=sample_url["user_id"]),
+                    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()
+            
+    cache.delete('all_workouts')