From b36c675084a6a8c8668dc5789dfaeade1bb169fa Mon Sep 17 00:00:00 2001
From: sosokker <sirin_29@hotmail.co.th>
Date: Sun, 26 Nov 2023 11:47:15 +0700
Subject: [PATCH] Task serializer must test userid availability

---
 backend/tasks/tasks/serializers.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/backend/tasks/tasks/serializers.py b/backend/tasks/tasks/serializers.py
index d2863a0..12bea72 100644
--- a/backend/tasks/tasks/serializers.py
+++ b/backend/tasks/tasks/serializers.py
@@ -1,4 +1,5 @@
 from rest_framework import serializers
+from users.models import CustomUser
 from boards.models import ListBoard
 from tasks.models import Todo, RecurrenceTask, Habit
 
@@ -8,7 +9,14 @@ class TaskSerializer(serializers.ModelSerializer):
         fields = '__all__'
 
     def create(self, validated_data):
-        # Create a new task with validated data
+        user_id = validated_data.get('user')
+
+        try:
+            user = CustomUser.objects.get(id=user_id)
+        except CustomUser.DoesNotExist:
+            raise serializers.ValidationError("User with the provided ID does not exist.")
+
+        validated_data['user'] = user
         return Todo.objects.create(**validated_data)
 
 class TaskCreateSerializer(serializers.ModelSerializer):