Mercurial > piecrust2

diff piecrust/processing/sitemap.py @ 853:f070a4fc033c

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
core: Continue PieCrust3 refactor, simplify pages. The asset pipeline is still the only function pipeline at this point. * No more `QualifiedPage`, and several other pieces of code deleted. * Data providers are simpler and more focused. For instance, the page iterator doesn't try to support other types of items. * Route parameters are proper known source metadata to remove the confusion between the two. * Make the baker and pipeline more correctly manage records and record histories. * Add support for record collapsing and deleting stale outputs in the asset pipeline.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 21 May 2017 00:06:59 -0700
parents 4850f8c21b6e
children 08e02c2a2a1a
line wrap: on
line diff
--- a/piecrust/processing/sitemap.py	Wed May 17 00:11:48 2017 -0700
+++ b/piecrust/processing/sitemap.py	Sun May 21 00:06:59 2017 -0700
@@ -1,7 +1,9 @@
+import os
+import os.path
 import time
 import logging
 import yaml
-from piecrust.data.iterators import PageIterator
+from piecrust.dataproviders.page_iterator import PageIterator
 from piecrust.processing.base import SimpleFileProcessor
 
 
@@ -36,11 +38,19 @@
         with open(in_path, 'r') as fp:
             sitemap = yaml.load(fp)
 
-        with open(out_path, 'w') as fp:
-            fp.write(SITEMAP_HEADER)
-            self._writeManualLocs(sitemap, fp)
-            self._writeAutoLocs(sitemap, fp)
-            fp.write(SITEMAP_FOOTER)
+        try:
+            with open(out_path, 'w') as fp:
+                fp.write(SITEMAP_HEADER)
+                self._writeManualLocs(sitemap, fp)
+                self._writeAutoLocs(sitemap, fp)
+                fp.write(SITEMAP_FOOTER)
+        except:
+            # If an exception occurs, delete the output file otherwise
+            # the pipeline will think the output was correctly produced.
+            if os.path.isfile(out_path):
+                logger.debug("Error occured, removing output sitemap.")
+                os.unlink(out_path)
+            raise
 
         return True