task list

junior/views.py

@@ -18,7 +18,8 @@ from rest_framework.pagination import PageNumberPagination
 from junior.models import Junior
 from .serializers import (CreateJuniorSerializer, JuniorDetailListSerializer, AddJuniorSerializer,\
                           RemoveJuniorSerializer)
-from guardian.models import Guardian
+from guardian.models import Guardian, JuniorTask
+from guardian.serializers import TaskDetailsSerializer
 from base.messages import ERROR_CODE, SUCCESS_CODE
 from base.constants import NUMBER
 from account.utils import custom_response, custom_error_response
@@ -184,3 +185,26 @@ class RemoveJuniorAPIView(views.APIView):
             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)
+
+
+class JuniorTaskListAPIView(viewsets.ModelViewSet):
+    """Update guardian profile"""
+    serializer_class = TaskDetailsSerializer
+    permission_classes = [IsAuthenticated]
+    pagination_class = PageNumberPagination
+    queryset = JuniorTask.objects.all()
+
+    def list(self, request, *args, **kwargs):
+        """Create guardian profile"""
+        status_value = self.request.GET.get('status')
+        if str(status_value) == '0':
+            queryset = JuniorTask.objects.filter(junior__auth=request.user).order_by('due_date', 'created_at')
+        else:
+            queryset = JuniorTask.objects.filter(junior__auth=request.user,
+                                                 task_status=status_value).order_by('due_date','created_at')
+        paginator = self.pagination_class()
+        # use Pagination
+        paginated_queryset = paginator.paginate_queryset(queryset, request)
+        # use TaskDetailsSerializer serializer
+        serializer = TaskDetailsSerializer(paginated_queryset, many=True)
+        return custom_response(None, serializer.data, response_status=status.HTTP_200_OK)