From 2df24632e046e39225f4c5f728fc64ab524e8274 Mon Sep 17 00:00:00 2001
From: Dharanidhar Reddy <dharanidhar.reddy@hashedin.com>
Date: Sat, 15 Jun 2019 18:15:09 +0530
Subject: [PATCH] Implemented Pagination decorator for View

---
 rest_framework/decorators.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/rest_framework/decorators.py b/rest_framework/decorators.py
index 671754fa1..579a811fa 100644
--- a/rest_framework/decorators.py
+++ b/rest_framework/decorators.py
@@ -7,10 +7,13 @@ based views, as well as the `@detail_route` and `@list_route` decorators, which
 used to annotate methods on viewsets that should be included by routers.
 """
 import types
+from functools import wraps
+from django.db.models import QuerySet
 
 from django.forms.utils import pretty_name
 
 from rest_framework.views import APIView
+from rest_framework.response import Response
 
 
 def api_view(http_method_names=None):
@@ -120,6 +123,32 @@ def schema(view_inspector):
     return decorator
 
 
+def paginate(view_method):
+    """
+    Adds Pagination to the method
+    """
+    @wraps(view_method)
+    def inner(view, *args, **kwargs):
+        assert isinstance(view, APIView), (
+                "paginate must be applied on an APIView, but applied on %s." % APIView
+        )
+
+        queryset = view_method(view, *args, **kwargs)
+
+        assert isinstance(queryset, (list, QuerySet)), (
+            "paginate expects a list or a QuerySet but got %s." % queryset
+        )
+
+        page = view.paginate_queryset(queryset)
+        if page is not None:
+            serializer = view.get_serializer(page, many=True)
+            return view.get_paginated_response(serializer.data)
+
+        serializer = view.get_serializer(queryset, many=True)
+        return Response(serializer.data)
+    return inner
+
+
 def action(methods=None, detail=None, url_path=None, url_name=None, **kwargs):
     """
     Mark a ViewSet method as a routable action.