jira-16 filter, invite and remove API

junior/views.py

@@ -1,9 +1,12 @@
 """Junior view file"""
-from rest_framework import viewsets, status
+from rest_framework import viewsets, status, generics,views
 from rest_framework.permissions import IsAuthenticated
+from rest_framework.pagination import PageNumberPagination
+from rest_framework.response import Response
 """Django app import"""
 from junior.models import Junior
-from .serializers import CreateJuniorSerializer, JuniorDetailListSerializer, AddJuniorSerializer
+from .serializers import (CreateJuniorSerializer, JuniorDetailListSerializer, AddJuniorSerializer,\
+                          RemoveJuniorSerializer)
 from guardian.models import Guardian
 from base.messages import ERROR_CODE, SUCCESS_CODE
 from account.utils import custom_response, custom_error_response
@@ -76,3 +79,70 @@ class AddJuniorAPIView(viewsets.ModelViewSet):
             serializer.save()
             return custom_response(SUCCESS_CODE['3021'], serializer.data, response_status=status.HTTP_200_OK)
         return custom_error_response(serializer.error, response_status=status.HTTP_400_BAD_REQUEST)
+
+class InvitedJuniorAPIView(viewsets.ModelViewSet):
+    """Junior list of assosicated guardian"""
+
+    serializer_class = JuniorDetailListSerializer
+    queryset = Junior.objects.all()
+    permission_classes = [IsAuthenticated]
+    pagination_class = PageNumberPagination
+
+    def get_queryset(self):
+        """Get the queryset for the view"""
+        guardian = Guardian.objects.filter(user__email=self.request.user).last()
+        junior_queryset = Junior.objects.filter(guardian_code__icontains=str(guardian.guardian_code),
+                                         is_invited=True)
+        return junior_queryset
+    def list(self, request, *args, **kwargs):
+        """ junior list"""
+        queryset = self.get_queryset()
+        paginator = self.pagination_class()
+        paginated_queryset = paginator.paginate_queryset(queryset, request)
+        serializer = JuniorDetailListSerializer(paginated_queryset, many=True)
+        return custom_response(None, serializer.data, response_status=status.HTTP_200_OK)
+
+
+class FilterJuniorAPIView(viewsets.ModelViewSet):
+    """Update guardian profile"""
+    serializer_class = JuniorDetailListSerializer
+    permission_classes = [IsAuthenticated]
+    pagination_class = PageNumberPagination
+    queryset = Junior.objects.all()
+
+    def get_queryset(self):
+        """Get the queryset for the view"""
+        title = self.request.GET.get('title')
+        guardian_data = Guardian.objects.filter(user__email=self.request.user).last()
+        queryset = Junior.objects.filter(guardian_code__icontains=str(guardian_data.guardian_code),
+                                         is_invited=True, auth__first_name=title)
+        return queryset
+
+    def list(self, request, *args, **kwargs):
+        """Create guardian profile"""
+        queryset = self.get_queryset()
+        paginator = self.pagination_class()
+        paginated_queryset = paginator.paginate_queryset(queryset, request)
+        serializer = JuniorDetailListSerializer(paginated_queryset, many=True)
+        return custom_response(None, serializer.data, response_status=status.HTTP_200_OK)
+
+
+class RemoveJuniorAPIView(views.APIView):
+    """Eater Update API"""
+    serializer_class = RemoveJuniorSerializer
+    model = Junior
+    permission_classes = [IsAuthenticated]
+
+    def put(self, request, format=None):
+        junior_id = self.request.GET.get('id')
+        guardian = Guardian.objects.filter(user__email=self.request.user).last()
+        junior_queryset = Junior.objects.filter(id=junior_id, guardian_code__icontains=str(guardian.guardian_code),
+                                                is_invited=True).last()
+        if junior_queryset:
+            serializer = RemoveJuniorSerializer(junior_queryset, data=request.data, partial=True)
+            if serializer.is_valid():
+                serializer.save()
+                return custom_response(SUCCESS_CODE['3022'], serializer.data, 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['2034'], response_status=status.HTTP_400_BAD_REQUEST)