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')