changes

junior/views.py

@@ -0,0 +1,40 @@
+"""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)