WIP

workout/views.py

@@ -12,6 +12,7 @@ from rest_framework.authentication import TokenAuthentication
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.decorators import authentication_classes
 from rest_framework.decorators import permission_classes
+from django.shortcuts import get_object_or_404
 import json
 
 
@@ -59,7 +60,6 @@ def add_workout(request):
 
     # exercise_data = dict(request.POST)["exercise_data"]
     exercise_data = request.data["exercise_data"]
-    print(exercise_data)
 
     if exercise_data is None:
         return Response({"exercise_data": [ "missing" ] }, status=status.HTTP_400_BAD_REQUEST)
@@ -99,4 +99,17 @@ def workouts_planned_by_logged_in_user(request):
     registered_user = RegisteredUser.objects.get(user=request.user)    
     workouts = PlannedWorkout.objects.filter(registered_user=registered_user)
     serializer = PlannedWorkoutSerializer(workouts, many=True)
-    return Response(serializer.data, status=status.HTTP_201_CREATED)
+    return Response(serializer.data, status=status.HTTP_200_OK)
+
+@api_view(['POST'])
+@authentication_classes([TokenAuthentication])
+@permission_classes([IsAuthenticated])
+def plan_workout(request):
+    registered_user = RegisteredUser.objects.get(user=request.user)    
+    serializer = POSTPlannedWorkoutSerializer(data=request.data, 
+                                              context = {"registered_user":registered_user.pk})
+    
+    if serializer.is_valid():
+        serializer.save()
+        return Response(serializer.data, status=status.HTTP_201_CREATED)
+    return Response(serializer.errors, status=status.HTTP_500_INTERNAL_SERVER_ERROR)