paginator.py 974 B

1234567891011121314151617181920212223242526272829
  1. from django.conf import settings
  2. from django.core.paginator import Paginator, Page
  3. class EnhancedPaginator(Paginator):
  4. def __init__(self, object_list, per_page, **kwargs):
  5. per_page = getattr(settings, 'PAGINATE_COUNT', 50)
  6. super(EnhancedPaginator, self).__init__(object_list, per_page, **kwargs)
  7. def _get_page(self, *args, **kwargs):
  8. return EnhancedPage(*args, **kwargs)
  9. class EnhancedPage(Page):
  10. def smart_pages(self):
  11. # Show first page, last page, next/previous two pages, and current page
  12. n = self.number
  13. pages_wanted = [1, n - 2, n - 1, n, n + 1, n + 2, self.paginator.num_pages]
  14. page_list = sorted(set(self.paginator.page_range).intersection(pages_wanted))
  15. # Insert skip markers
  16. skip_pages = [x[1] for x in zip(page_list[:-1], page_list[1:]) if (x[1] - x[0] != 1)]
  17. for i in skip_pages:
  18. page_list.insert(page_list.index(i), False)
  19. return page_list