You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

paginator.py 6.1KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194
  1. import collections.abc
  2. import inspect
  3. import warnings
  4. from math import ceil
  5. from django.utils.deprecation import RemovedInDjango31Warning
  6. from django.utils.functional import cached_property
  7. from django.utils.inspect import method_has_no_args
  8. from django.utils.translation import gettext_lazy as _
  9. class UnorderedObjectListWarning(RuntimeWarning):
  10. pass
  11. class InvalidPage(Exception):
  12. pass
  13. class PageNotAnInteger(InvalidPage):
  14. pass
  15. class EmptyPage(InvalidPage):
  16. pass
  17. class Paginator:
  18. def __init__(self, object_list, per_page, orphans=0,
  19. allow_empty_first_page=True):
  20. self.object_list = object_list
  21. self._check_object_list_is_ordered()
  22. self.per_page = int(per_page)
  23. self.orphans = int(orphans)
  24. self.allow_empty_first_page = allow_empty_first_page
  25. def validate_number(self, number):
  26. """Validate the given 1-based page number."""
  27. try:
  28. if isinstance(number, float) and not number.is_integer():
  29. raise ValueError
  30. number = int(number)
  31. except (TypeError, ValueError):
  32. raise PageNotAnInteger(_('That page number is not an integer'))
  33. if number < 1:
  34. raise EmptyPage(_('That page number is less than 1'))
  35. if number > self.num_pages:
  36. if number == 1 and self.allow_empty_first_page:
  37. pass
  38. else:
  39. raise EmptyPage(_('That page contains no results'))
  40. return number
  41. def get_page(self, number):
  42. """
  43. Return a valid page, even if the page argument isn't a number or isn't
  44. in range.
  45. """
  46. try:
  47. number = self.validate_number(number)
  48. except PageNotAnInteger:
  49. number = 1
  50. except EmptyPage:
  51. number = self.num_pages
  52. return self.page(number)
  53. def page(self, number):
  54. """Return a Page object for the given 1-based page number."""
  55. number = self.validate_number(number)
  56. bottom = (number - 1) * self.per_page
  57. top = bottom + self.per_page
  58. if top + self.orphans >= self.count:
  59. top = self.count
  60. return self._get_page(self.object_list[bottom:top], number, self)
  61. def _get_page(self, *args, **kwargs):
  62. """
  63. Return an instance of a single page.
  64. This hook can be used by subclasses to use an alternative to the
  65. standard :cls:`Page` object.
  66. """
  67. return Page(*args, **kwargs)
  68. @cached_property
  69. def count(self):
  70. """Return the total number of objects, across all pages."""
  71. c = getattr(self.object_list, 'count', None)
  72. if callable(c) and not inspect.isbuiltin(c) and method_has_no_args(c):
  73. return c()
  74. return len(self.object_list)
  75. @cached_property
  76. def num_pages(self):
  77. """Return the total number of pages."""
  78. if self.count == 0 and not self.allow_empty_first_page:
  79. return 0
  80. hits = max(1, self.count - self.orphans)
  81. return ceil(hits / self.per_page)
  82. @property
  83. def page_range(self):
  84. """
  85. Return a 1-based range of pages for iterating through within
  86. a template for loop.
  87. """
  88. return range(1, self.num_pages + 1)
  89. def _check_object_list_is_ordered(self):
  90. """
  91. Warn if self.object_list is unordered (typically a QuerySet).
  92. """
  93. ordered = getattr(self.object_list, 'ordered', None)
  94. if ordered is not None and not ordered:
  95. obj_list_repr = (
  96. '{} {}'.format(self.object_list.model, self.object_list.__class__.__name__)
  97. if hasattr(self.object_list, 'model')
  98. else '{!r}'.format(self.object_list)
  99. )
  100. warnings.warn(
  101. 'Pagination may yield inconsistent results with an unordered '
  102. 'object_list: {}.'.format(obj_list_repr),
  103. UnorderedObjectListWarning,
  104. stacklevel=3
  105. )
  106. class QuerySetPaginator(Paginator):
  107. def __init__(self, *args, **kwargs):
  108. warnings.warn(
  109. 'The QuerySetPaginator alias of Paginator is deprecated.',
  110. RemovedInDjango31Warning, stacklevel=2,
  111. )
  112. super().__init__(*args, **kwargs)
  113. class Page(collections.abc.Sequence):
  114. def __init__(self, object_list, number, paginator):
  115. self.object_list = object_list
  116. self.number = number
  117. self.paginator = paginator
  118. def __repr__(self):
  119. return '<Page %s of %s>' % (self.number, self.paginator.num_pages)
  120. def __len__(self):
  121. return len(self.object_list)
  122. def __getitem__(self, index):
  123. if not isinstance(index, (int, slice)):
  124. raise TypeError
  125. # The object_list is converted to a list so that if it was a QuerySet
  126. # it won't be a database hit per __getitem__.
  127. if not isinstance(self.object_list, list):
  128. self.object_list = list(self.object_list)
  129. return self.object_list[index]
  130. def has_next(self):
  131. return self.number < self.paginator.num_pages
  132. def has_previous(self):
  133. return self.number > 1
  134. def has_other_pages(self):
  135. return self.has_previous() or self.has_next()
  136. def next_page_number(self):
  137. return self.paginator.validate_number(self.number + 1)
  138. def previous_page_number(self):
  139. return self.paginator.validate_number(self.number - 1)
  140. def start_index(self):
  141. """
  142. Return the 1-based index of the first object on this page,
  143. relative to total objects in the paginator.
  144. """
  145. # Special case, return zero if no items.
  146. if self.paginator.count == 0:
  147. return 0
  148. return (self.paginator.per_page * (self.number - 1)) + 1
  149. def end_index(self):
  150. """
  151. Return the 1-based index of the last object on this page,
  152. relative to total objects found (hits).
  153. """
  154. # Special case for the last page because there can be orphans.
  155. if self.number == self.paginator.num_pages:
  156. return self.paginator.count
  157. return self.number * self.paginator.per_page