piecrust2: piecrust/dataproviders/pageiterator.py comparison

bake: Support the `draft` setting.

author	Ludovic Chabant <ludovic@chabant.com>
date	Wed, 04 Oct 2017 20:40:43 -0700
parents	1d0364614665
children	45ad976712ec

comparison

equal deleted inserted replaced

-:7ecb946bfafd
+:abc52a6262a1
 def _ensureSorter(self):
 if self._has_sorter:
 return
 if self._is_content_source:
+# For content sources, the default sorting is reverse
+# date/time sorting.
 self._it = DateSortIterator(self._it, reverse=True)
 self._has_sorter = True
 def _initIterator(self):
 if self._is_content_source:
 self._it = PageContentSourceIterator(self._source)
+app = self._source.app
+if app.config.get('baker/is_baking'):
+# While baking, automatically exclude any page with
+# the `draft` setting.
+draft_setting = app.config['baker/no_bake_setting']
+self._it = NoDraftsIterator(self._it, draft_setting)
 else:
 self._it = GenericSourceIterator(self._source)
 def _unload(self):
 self._initIterator()
 def __iter__(self):
 source = self.source
 yield from source.getAllPages()
+class NoDraftsIterator:
+def __init__(self, source, no_draft_setting):
+self.it = source
+self.no_draft_setting = no_draft_setting
+def __iter__(self):
+nds = self.no_draft_setting
+yield from filter(lambda i: not i.config.get(nds), self.it)
 class PaginationDataBuilderIterator:
 def __init__(self, it):
 self.it = it
 def __iter__(self):

Mercurial > piecrust2