from django.shortcuts import render
from exercise.models import Exercise
from muscle.models import Muscle, ExerciseMuscle
from equipment.models import Equipment, WorkoutEquipment
from rest_framework.decorators import api_view, authentication_classes, permission_classes
from rest_framework.authentication import TokenAuthentication
from rest_framework.permissions import IsAuthenticated, IsAdminUser
from rest_framework.response import Response
from rest_framework import status
from django.core.cache import cache

# Create your views here.
@api_view(['GET'])
@authentication_classes([TokenAuthentication])
@permission_classes([IsAuthenticated])
def sync_equipment(request):
    all_exercise = Exercise.objects.all()
    for exercise in all_exercise:
        all_equipment = (exercise.equipment_required or '').split(',')
        for equipment in all_equipment:
            if len(equipment) > 0:
                try:
                    equipment_obj = Equipment.objects.get(name=equipment.lower())
                    WorkoutEquipment.objects.create(exercise=exercise, equipment=equipment_obj)
                except Equipment.DoesNotExist:
                    pass


    return Response(status=status.HTTP_200_OK)

@api_view(['GET'])
@authentication_classes([TokenAuthentication])
@permission_classes([IsAuthenticated])
def sync_muscle_groups(request):
    all_exercise = Exercise.objects.all()
    for exercise in all_exercise:
        all_muscle_groups = (exercise.muscle_groups or '').split(',')
        for muscle_group in all_muscle_groups:
            if len(muscle_group) > 0:
                try:
                    muscle_obj = Muscle.objects.get(name=muscle_group.lower())
                    ExerciseMuscle.objects.create(exercise=exercise, muscle=muscle_obj)
                except Muscle.DoesNotExist:
                    pass


    return Response(status=status.HTTP_200_OK)

@api_view(['POST'])
@authentication_classes([TokenAuthentication])
@permission_classes([IsAuthenticated, IsAdminUser])
def clear_redis(request):
    cache.clear()
    return Response(status=status.HTTP_200_OK)