X-Git-Url: https://git.mdrn.pl/django-pagination.git/blobdiff_plain/8df2d3c3985857213effe95e92f86afb2f804290..786f413a03df1e4d0485213d8c1d796d15ac114d:/pagination/middleware.py?ds=sidebyside

diff --git a/pagination/middleware.py b/pagination/middleware.py
index 5e917c5..676f909 100644
--- a/pagination/middleware.py
+++ b/pagination/middleware.py
@@ -1,10 +1,17 @@
+def get_page(self, suffix):
+    """
+    A function which will be monkeypatched onto the request to get the current
+    integer representing the current page.
+    """
+    try:
+        return int(self.REQUEST['page%s' % suffix])
+    except (KeyError, ValueError, TypeError):
+        return 1
+
 class PaginationMiddleware(object):
     """
     Inserts a variable representing the current page onto the request object if
     it exists in either **GET** or **POST** portions of the request.
     """
     def process_request(self, request):
-        try:
-            request.page = int(request.REQUEST['page'])
-        except (KeyError, ValueError, TypeError):
-            request.page = 1
\ No newline at end of file
+        request.__class__.page = get_page
\ No newline at end of file