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

diff --git a/pagination/middleware.py b/pagination/middleware.py
index 8a541fb..676f909 100644
--- a/pagination/middleware.py
+++ b/pagination/middleware.py
@@ -1,6 +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['page'])
-        except KeyError:
-            request.page = 1
\ No newline at end of file
+        request.__class__.page = get_page
\ No newline at end of file