X-Git-Url: https://git.mdrn.pl/django-pagination.git/blobdiff_plain/d4c2881356622fbe0653c1e5e183f2c0dba8e79b..951cf70f69e9ec7f60db366f90477fe0eefb0be5:/pagination/middleware.py diff --git a/pagination/middleware.py b/pagination/middleware.py index b99a7e3..f8a2a6f 100644 --- a/pagination/middleware.py +++ b/pagination/middleware.py @@ -1,10 +1,17 @@ +def get_page(self): + """ + A function which will be monkeypatched onto the request to get the current + integer representing the current page. + """ + try: + return int(self.REQUEST['page']) + 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, ValueError): - request.page = 1 \ No newline at end of file + request.__class__.page = property(get_page) \ No newline at end of file