mirror of
https://github.com/HamzaSha1/zod-backend.git
synced 2025-07-16 02:16:16 +00:00
79 lines
3.7 KiB
Python
79 lines
3.7 KiB
Python
"""Views of Guardian"""
|
|
"""Third party Django app"""
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework import viewsets, status
|
|
"""Import Django app"""
|
|
from .serializers import UserSerializer, CreateGuardianSerializer, TaskSerializer, TaskDetailsSerializer
|
|
from .models import Guardian, JuniorTask
|
|
from junior.models import Junior
|
|
from account.models import UserEmailOtp
|
|
from .tasks import generate_otp
|
|
from account.utils import send_otp_email
|
|
from account.utils import custom_response, custom_error_response
|
|
from base.messages import ERROR_CODE, SUCCESS_CODE
|
|
# Create your views here.
|
|
class SignupViewset(viewsets.ModelViewSet):
|
|
"""Signup view set"""
|
|
serializer_class = UserSerializer
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
"""Create user profile"""
|
|
if request.data['user_type'] in ['1', '2']:
|
|
serializer = UserSerializer(context=request.data['user_type'], data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
"""Generate otp"""
|
|
otp = generate_otp()
|
|
UserEmailOtp.objects.create(email=request.data['email'], otp=otp, user_type=str(request.data['user_type']))
|
|
"""Send email to the register user"""
|
|
send_otp_email(request.data['email'], otp)
|
|
return custom_response(SUCCESS_CODE['3001'], {"email_otp": otp},
|
|
response_status=status.HTTP_200_OK)
|
|
return custom_error_response(serializer.errors, response_status=status.HTTP_400_BAD_REQUEST)
|
|
else:
|
|
return custom_error_response(ERROR_CODE['2028'], response_status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
class UpdateGuardianProfile(viewsets.ViewSet):
|
|
"""Update guardian profile"""
|
|
serializer_class = CreateGuardianSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
"""Create guardian profile"""
|
|
serializer = CreateGuardianSerializer(context={"user":request.user,
|
|
"first_name":request.data.get('first_name', ''),
|
|
"last_name": request.data.get('last_name',' ')},
|
|
data=request.data)
|
|
if serializer.is_valid():
|
|
"""save serializer"""
|
|
serializer.save()
|
|
return custom_response(None, serializer.data,response_status=status.HTTP_200_OK)
|
|
return custom_error_response(serializer.errors, response_status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
|
|
class TaskListAPIView(viewsets.ModelViewSet):
|
|
"""Update guardian profile"""
|
|
serializer_class = TaskDetailsSerializer
|
|
permission_classes = [IsAuthenticated]
|
|
|
|
def list(self, request, *args, **kwargs):
|
|
"""Create guardian profile"""
|
|
queryset = JuniorTask.objects.filter(guardian__user=request.user)
|
|
print("queryset==>",queryset)
|
|
serializer = TaskDetailsSerializer(queryset, many=True)
|
|
print("serializer.data===>",serializer.data)
|
|
return custom_response(None, serializer.data, response_status=status.HTTP_200_OK)
|
|
|
|
class CreateTaskAPIView(viewsets.ModelViewSet):
|
|
"""create task for junior"""
|
|
serializer_class = TaskSerializer
|
|
|
|
def create(self, request, *args, **kwargs):
|
|
print("request.data===>",request.data)
|
|
image = request.data['image']
|
|
serializer = TaskSerializer(context={"user":request.user, "image":image}, data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return custom_response(SUCCESS_CODE['3018'], serializer.data, response_status=status.HTTP_200_OK)
|
|
return custom_error_response(serializer.errors, response_status=status.HTTP_400_BAD_REQUEST)
|