can create a workout

workout/views.py

@@ -48,3 +48,45 @@ def workouts_completed_by_logged_in_user(request):
     workouts = CompletedWorkout.objects.filter(registered_user=registered_user)
     serializer = GetCompleteWorkoutSerializer(workouts, many=True)
     return Response(serializer.data, status=status.HTTP_201_CREATED)
+
+@api_view(['POST'])
+@authentication_classes([TokenAuthentication])
+@permission_classes([IsAuthenticated])
+def add_workout(request):
+    registered_user = RegisteredUser.objects.get(user=request.user)   
+    if registered_user is None:
+        return Response(status=status.HTTP_400_BAD_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)
+    if len(exercise_data) < 1:
+        return Response({"exercise_data": [ "empty" ] }, status=status.HTTP_400_BAD_REQUEST)
+
+    serializer = POSTCompleteWorkoutSerializer(data=request.data)
+
+    if serializer.is_valid():
+        new_obj = serializer.save(registered_user=registered_user)
+        new_obj.save()
+
+        for data in exercise_data:
+            print(data)
+            id = data["id"]
+            if id is None:
+                continue
+      
+            exercise = Exercise.objects.get(pk=id)
+            completed_workout = WorkoutExercise.objects.create(
+                workout=new_obj, 
+                exercise=exercise
+            )
+            if "weight" in data:
+                completed_workout.weight = data["weight"]
+            if "reps" in data:
+                completed_workout.reps = data["reps"]
+            completed_workout.save()
+
+        return Response(serializer.data, status=status.HTTP_201_CREATED)
+    return Response(serializer.errors, status=status.HTTP_500_INTERNAL_SERVER_ERROR)