zod-backend/junior/views.py

"""Junior view file"""
from rest_framework import viewsets, status
from rest_framework.permissions import IsAuthenticated
"""Django app import"""
from junior.models import Junior
from .serializers import CreateJuniorSerializer
from guardian.models import Guardian
from base.messages import ERROR_CODE, SUCCESS_CODE
from account.utils import custom_response, custom_error_response
# Create your views here.
class UpdateJuniorProfile(viewsets.ViewSet):
    """Update junior profile"""
    serializer_class = CreateJuniorSerializer
    permission_classes = [IsAuthenticated]

    def create(self, request, *args, **kwargs):
        """Use CreateJuniorSerializer"""
        serializer = CreateJuniorSerializer(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 ValidateGuardianCode(viewsets.ViewSet):
    """Check guardian code exist or not"""
    permission_classes = [IsAuthenticated]

    def list(self, request, *args, **kwargs):
        """check guardian code"""
        guardian_code = self.request.GET.get('guardian_code').split(',')
        for code in guardian_code:
            guardian_data = Guardian.objects.filter(guardian_code=code).exists()
            if guardian_data:
                return custom_response(SUCCESS_CODE['3013'], response_status=status.HTTP_200_OK)
            else:
                return custom_error_response(ERROR_CODE["2022"], response_status=status.HTTP_400_BAD_REQUEST)