Merge pull request #8 from JuanBenitez97/movie_posters

added custom pagination to get all movies endpoint

Author: Juan Benitez

README.md

@@ -94,5 +94,14 @@ http POST http://127.0.0.1:8000/api/v1/movies/ "Authorization: Token <YOUR_TOKEN
 http PUT http://127.0.0.1:8000/api/v1/movies/3 "Authorization: Token <YOUR_TOKEN>" title="AntMan and The Wasp" genre="Action" year=2018
 http DELETE http://127.0.0.1:8000/api/v1/movies/3 "Authorization: Token <YOUR_TOKEN>"
 ```
+
+### Pagination
+The API supports pagination, by default responses have a page_size=10 but if you want change that you can pass through params page=size=X
+```
+http http://127.0.0.1:8000/api/v1/movies/?page=1 "Authorization: Token <YOUR_TOKEN>"
+http http://127.0.0.1:8000/api/v1/movies/?page=3 "Authorization: Token <YOUR_TOKEN>"
+http http://127.0.0.1:8000/api/v1/movies/?page=3&page_size=15 "Authorization: Token <YOUR_TOKEN>"
+```
+
 Finally, I provide a DB to make these tests.
 

movies/pagination.py

@@ -0,0 +1,5 @@
+from rest_framework.pagination import PageNumberPagination
+
+class CustomPagination(PageNumberPagination):
+    page_size = 10
+    page_size_query_param = 'page_size'

movies/views.py

@@ -4,6 +4,7 @@
 from .models import Movie
 from .permissions import IsOwnerOrReadOnly, IsAuthenticated
 from .serializers import MovieSerializer
+from .pagination import CustomPagination
 
 class get_delete_update_movie(RetrieveUpdateDestroyAPIView):
     serializer_class = MovieSerializer
@@ -64,21 +65,21 @@ def delete(self, request, pk):
 class get_post_movies(ListCreateAPIView):
     serializer_class = MovieSerializer
     permission_classes = (IsAuthenticated,)
-    # Get all movies
+    pagination_class = CustomPagination
 
     def get_queryset(self):
        movies = Movie.objects.all()
        return movies
-    
-    def get(self, request):
 
+    # Get all movies
+    def get(self, request):
         movies = self.get_queryset()
-        serializer = MovieSerializer(movies, many=True)
-        return Response(serializer.data, status=status.HTTP_200_OK)
+        paginate_queryset = self.paginate_queryset(movies)
+        serializer = self.serializer_class(paginate_queryset, many=True)
+        return self.get_paginated_response(serializer.data)
 
     # Create a new movie
     def post(self, request):
-
         serializer = MovieSerializer(data=request.data)
         if serializer.is_valid():
             serializer.save(creator=request.user)