diff piecrust/data/provider.py @ 23:923699e816d0

Don't try to get the name of a source that doesn't have one. TODO: clean up code duplication.
author Ludovic Chabant <ludovic@chabant.com>
date Tue, 19 Aug 2014 08:34:16 -0700
parents 343d08ef5668
children 371a6c879ab9
line wrap: on
line diff
--- a/piecrust/data/provider.py	Mon Aug 18 23:20:43 2014 -0700
+++ b/piecrust/data/provider.py	Tue Aug 19 08:34:16 2014 -0700
@@ -70,8 +70,7 @@
     def _onIteration(self):
         if not self._ctx_set:
             eis = self._page.app.env.exec_info_stack
-            eis.current_page_info.render_ctx.used_source_names.add(
-                    self._source.name)
+            eis.current_page_info.render_ctx.addUsedSource(self._source.name)
             self._ctx_set = True
 
 
@@ -192,8 +191,7 @@
     def _onIteration(self):
         if not self._ctx_set:
             eis = self._page.app.env.exec_info_stack
-            eis.current_page_info.render_ctx.used_source_names.add(
-                    self._source.name)
+            eis.current_page_info.render_ctx.addUsedSource(self._source)
             self._ctx_set = True