piecrust2: piecrust/data/iterators.py annotate

annotate piecrust/data/iterators.py @ 550:6f216c1ab6b1

bake: Add a flag to know which record entries got collapsed from last run. This makes it possible to find entries for things that were actually baked during the current run, as opposed to skipped because they were "clean".

author	Ludovic Chabant <ludovic@chabant.com>
date	Tue, 04 Aug 2015 21:22:30 -0700
parents	cb3446be44b7
children	911a054f4fbd

rev	line source
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	1 import logging
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	2 from piecrust.data.filters import PaginationFilter
455 cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	3 from piecrust.environment import AbortedSourceUseError
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 from piecrust.events import Event
455 cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	5 from piecrust.sources.base import PageSource
242 f130365568ff internal: Code reorganization to put less stuff in `sources.base`. Ludovic Chabant <ludovic@chabant.com> parents: 233 diff changeset	6 from piecrust.sources.interfaces import IPaginationSource
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9 logger = logging.getLogger(__name__)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	10
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	11
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	12 class SliceIterator(object):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	13 def __init__(self, it, offset=0, limit=-1):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	14 self.it = it
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	15 self.offset = offset
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	16 self.limit = limit
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17 self.current_page = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	18 self.has_more = False
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19 self.inner_count = -1
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20 self.next_page = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 self.prev_page = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22 self._cache = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24 def __iter__(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	25 if self._cache is None:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	26 inner_list = list(self.it)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	27 self.inner_count = len(inner_list)
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	28
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	29 if self.limit > 0:
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	30 self.has_more = self.inner_count > (self.offset + self.limit)
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	31 self._cache = inner_list[self.offset:self.offset + self.limit]
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	32 else:
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	33 self.has_more = False
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	34 self._cache = inner_list[self.offset:]
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	35
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 if self.current_page:
21 53b9df2263dc Fix for pages listing pages from other sources. Ludovic Chabant <ludovic@chabant.com> parents: 16 diff changeset	37 try:
53b9df2263dc Fix for pages listing pages from other sources. Ludovic Chabant <ludovic@chabant.com> parents: 16 diff changeset	38 idx = inner_list.index(self.current_page)
53b9df2263dc Fix for pages listing pages from other sources. Ludovic Chabant <ludovic@chabant.com> parents: 16 diff changeset	39 except ValueError:
53b9df2263dc Fix for pages listing pages from other sources. Ludovic Chabant <ludovic@chabant.com> parents: 16 diff changeset	40 idx = -1
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	41 if idx >= 0:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	42 if idx < self.inner_count - 1:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	43 self.next_page = inner_list[idx + 1]
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	44 if idx > 0:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	45 self.prev_page = inner_list[idx - 1]
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	46
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	47 return iter(self._cache)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	48
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	49
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50 class SettingFilterIterator(object):
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	51 def __init__(self, it, fil_conf, setting_accessor=None):
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52 self.it = it
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	53 self.fil_conf = fil_conf
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54 self._fil = None
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	55 self.setting_accessor = setting_accessor
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	56
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	57 def __iter__(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	58 if self._fil is None:
280 8c0c53a315ae data: Correctly build pagination filters when we know items are pages. Ludovic Chabant <ludovic@chabant.com> parents: 242 diff changeset	59 self._fil = PaginationFilter(value_accessor=self.setting_accessor)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	60 self._fil.addClausesFromConfig(self.fil_conf)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	61
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	62 for i in self.it:
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	63 if self._fil.pageMatches(i):
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	64 yield i
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	65
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	66
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	67 class NaturalSortIterator(object):
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	68 def __init__(self, it, reverse=False):
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	69 self.it = it
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	70 self.reverse = reverse
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	71
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	72 def __iter__(self):
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	73 return iter(sorted(self.it, reverse=self.reverse))
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	74
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	75
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	76 class SettingSortIterator(object):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77 def __init__(self, it, name, reverse=False, value_accessor=None):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	78 self.it = it
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	79 self.name = name
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	80 self.reverse = reverse
226 e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	81 self.value_accessor = value_accessor or self._default_value_accessor
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	82
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	83 def __iter__(self):
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	84 return iter(sorted(self.it, key=self._key_getter,
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	85 reverse=self.reverse))
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	86
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	87 def _key_getter(self, item):
226 e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	88 key = self.value_accessor(item, self.name)
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	89 if key is None:
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	90 return 0
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	91 return key
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	92
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	93 @staticmethod
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	94 def _default_value_accessor(item, name):
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	95 try:
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	96 return getattr(item, name)
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	97 except AttributeError:
e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	98 return None
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	99
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	100
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	101 class PaginationFilterIterator(object):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	102 def __init__(self, it, fil):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 self.it = it
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	104 self._fil = fil
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	105
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	106 def __iter__(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	107 for page in self.it:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108 if self._fil.pageMatches(page):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	109 yield page
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	110
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	111
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	112 class PageIterator(object):
408 fd8e39254da0 debug: Better debug info output for iterators, providers, and linkers. Ludovic Chabant <ludovic@chabant.com> parents: 280 diff changeset	113 debug_render = []
fd8e39254da0 debug: Better debug info output for iterators, providers, and linkers. Ludovic Chabant <ludovic@chabant.com> parents: 280 diff changeset	114 debug_render_doc_dynamic = ['_debugRenderDoc']
fd8e39254da0 debug: Better debug info output for iterators, providers, and linkers. Ludovic Chabant <ludovic@chabant.com> parents: 280 diff changeset	115 debug_render_not_empty = True
fd8e39254da0 debug: Better debug info output for iterators, providers, and linkers. Ludovic Chabant <ludovic@chabant.com> parents: 280 diff changeset	116
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	117 def __init__(self, source, current_page=None, pagination_filter=None,
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	118 offset=0, limit=-1, locked=False):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	119 self._source = source
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	120 self._current_page = current_page
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	121 self._locked = False
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	122 self._pages = source
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	123 self._pagesData = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	124 self._pagination_slicer = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	125 self._has_sorter = False
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	126 self._next_page = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	127 self._prev_page = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	128 self._iter_event = Event()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	129
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	130 if isinstance(source, IPaginationSource):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	131 src_it = source.getSourceIterator()
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	132 if src_it is not None:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	133 self._pages = src_it
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	134
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	135 # Apply any filter first, before we start sorting or slicing.
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	136 if pagination_filter is not None:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	137 self._simpleNonSortedWrap(PaginationFilterIterator,
226 e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	138 pagination_filter)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	140 if offset > 0 or limit > 0:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	141 self.slice(offset, limit)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	142
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	143 self._locked = locked
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	144
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	145 @property
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146 def total_count(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	147 self._load()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	148 if self._pagination_slicer is not None:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	149 return self._pagination_slicer.inner_count
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	150 return len(self._pagesData)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	151
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	152 @property
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	153 def next_page(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154 self._load()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	155 return self._next_page
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	156
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	157 @property
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158 def prev_page(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	159 self._load()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	160 return self._prev_page
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	161
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	162 def __len__(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	163 self._load()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	164 return len(self._pagesData)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	165
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	166 def __getitem__(self, key):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	167 self._load()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	168 return self._pagesData[key]
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	169
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	170 def __iter__(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	171 self._load()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	172 self._iter_event.fire()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	173 return iter(self._pagesData)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	174
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	175 def __getattr__(self, name):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	176 if name[:3] == 'is_' or name[:3] == 'in_':
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177 def is_filter(value):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	178 conf = {'is_%s' % name[3:]: value}
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	179 accessor = self._getSettingAccessor()
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	180 return self._simpleNonSortedWrap(SettingFilterIterator, conf,
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	181 accessor)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	182 return is_filter
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	184 if name[:4] == 'has_':
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	185 def has_filter(value):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	186 conf = {name: value}
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	187 accessor = self._getSettingAccessor()
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	188 return self._simpleNonSortedWrap(SettingFilterIterator, conf,
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	189 accessor)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	190 return has_filter
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	191
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	192 if name[:5] == 'with_':
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	193 def has_filter(value):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	194 conf = {'has_%s' % name[5:]: value}
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	195 accessor = self._getSettingAccessor()
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	196 return self._simpleNonSortedWrap(SettingFilterIterator, conf,
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	197 accessor)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	198 return has_filter
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	199
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	200 return self.__getattribute__(name)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	201
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	202 def skip(self, count):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	203 return self._simpleWrap(SliceIterator, count)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	204
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	205 def limit(self, count):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	206 return self._simpleWrap(SliceIterator, 0, count)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	207
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	208 def slice(self, skip, limit):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	209 return self._simpleWrap(SliceIterator, skip, limit)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	210
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	211 def filter(self, filter_name):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	212 if self._current_page is None:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	213 raise Exception("Can't use `filter()` because no parent page was "
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	214 "set for this page iterator.")
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	215 filter_conf = self._current_page.config.get(filter_name)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	216 if filter_conf is None:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	217 raise Exception("Couldn't find filter '%s' in the configuration "
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	218 "header for page: %s" %
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	219 (filter_name, self._current_page.path))
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	220 accessor = self._getSettingAccessor()
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	221 return self._simpleNonSortedWrap(SettingFilterIterator, filter_conf,
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	222 accessor)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	223
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	224 def sort(self, setting_name=None, reverse=False):
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	225 self._ensureUnlocked()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	226 self._unload()
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	227 if setting_name is not None:
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	228 accessor = self._getSettingAccessor()
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	229 self._pages = SettingSortIterator(self._pages, setting_name,
226 e9dc18a275ff data: Add ability for `IPaginationSource`s to specify how to get settings. Ludovic Chabant <ludovic@chabant.com> parents: 21 diff changeset	230 reverse, accessor)
10 cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	231 else:
cd35d356ccce Fix some bugs with iterators, add some unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 6 diff changeset	232 self._pages = NaturalSortIterator(self._pages, reverse)
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	233 self._has_sorter = True
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	234 return self
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	235
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	236 def reset(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	237 self._ensureUnlocked()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	238 self._unload
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	239 return self
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	240
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	241 @property
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	242 def _has_more(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	243 self._load()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	244 if self._pagination_slicer:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	245 return self._pagination_slicer.has_more
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	246 return False
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	247
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	248 def _simpleWrap(self, it_class, args, *kwargs):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	249 self._ensureUnlocked()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	250 self._unload()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	251 self._ensureSorter()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	252 self._pages = it_class(self._pages, args, *kwargs)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	253 if self._pagination_slicer is None and it_class is SliceIterator:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	254 self._pagination_slicer = self._pages
16 3a13b43e77c1 Correctly set the current page on a pagination slicer. Ludovic Chabant <ludovic@chabant.com> parents: 10 diff changeset	255 self._pagination_slicer.current_page = self._current_page
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	256 return self
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	257
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	258 def _simpleNonSortedWrap(self, it_class, args, *kwargs):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	259 self._ensureUnlocked()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	260 self._unload()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	261 self._pages = it_class(self._pages, args, *kwargs)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	262 return self
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	263
233 4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	264 def _getSettingAccessor(self):
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	265 accessor = None
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	266 if isinstance(self._source, IPaginationSource):
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	267 accessor = self._source.getSettingAccessor()
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	268 return accessor
4379d8f8f831 internal: Removing some dependency of filters and iterators on pages. Ludovic Chabant <ludovic@chabant.com> parents: 228 diff changeset	269
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	270 def _ensureUnlocked(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	271 if self._locked:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	272 raise Exception(
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	273 "This page iterator has been locked, probably because "
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	274 "you're trying to tamper with pagination data.")
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	275
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	276 def _ensureSorter(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	277 if self._has_sorter:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	278 return
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	279 if isinstance(self._source, IPaginationSource):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	280 sort_it = self._source.getSorterIterator(self._pages)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	281 if sort_it is not None:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	282 self._pages = sort_it
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	283 self._has_sorter = True
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	284
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	285 def _unload(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	286 self._pagesData = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	287 self._next_page = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	288 self._prev_page = None
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	289
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	290 def _load(self):
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	291 if self._pagesData is not None:
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	292 return
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	293
455 cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	294 if (self._current_page is not None and
cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	295 self._current_page.app.env.abort_source_use and
cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	296 isinstance(self._source, PageSource)):
cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	297 logger.debug("Aborting iteration from %s." %
cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	298 self._current_page.ref_spec)
cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	299 raise AbortedSourceUseError()
cb3446be44b7 bake: Abort "render first" jobs if we start using other pages. Ludovic Chabant <ludovic@chabant.com> parents: 408 diff changeset	300
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	301 self._ensureSorter()
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	302
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	303 it_chain = self._pages
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	304 is_pgn_source = False
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	305 if isinstance(self._source, IPaginationSource):
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	306 is_pgn_source = True
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	307 tail_it = self._source.getTailIterator(self._pages)
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	308 if tail_it is not None:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	309 it_chain = tail_it
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	310
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	311 self._pagesData = list(it_chain)
f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	312
6 f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	313 if is_pgn_source and self._current_page and self._pagination_slicer:
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	314 pn = [self._pagination_slicer.prev_page,
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	315 self._pagination_slicer.next_page]
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	316 pn_it = self._source.getTailIterator(iter(pn))
f5ca5c5bed85 More Python 3 fixes, modularization, and new unit tests. Ludovic Chabant <ludovic@chabant.com> parents: 3 diff changeset	317 self._prev_page, self._next_page = (list(pn_it))
3 f485ba500df3 Gigantic change to basically make PieCrust 2 vaguely functional. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	318
408 fd8e39254da0 debug: Better debug info output for iterators, providers, and linkers. Ludovic Chabant <ludovic@chabant.com> parents: 280 diff changeset	319 def _debugRenderDoc(self):
fd8e39254da0 debug: Better debug info output for iterators, providers, and linkers. Ludovic Chabant <ludovic@chabant.com> parents: 280 diff changeset	320 return "Contains %d items" % len(self)
fd8e39254da0 debug: Better debug info output for iterators, providers, and linkers. Ludovic Chabant <ludovic@chabant.com> parents: 280 diff changeset	321

Mercurial > piecrust2

annotate piecrust/data/iterators.py @ 550:6f216c1ab6b1