From: Fredrik Unger <fred@tree.se>
Date: Sun, 10 Apr 2011 14:30:42 +0000 (+0200)
Subject: Adding method to write a sitemap.txt and writing it when something changed.
X-Git-Tag: v1.0~10
X-Git-Url: https://source.tree.se/git?p=treecutter.git;a=commitdiff_plain;h=22ae890dfe29c7afeb9300307239edbd11a418cd

Adding method to write a sitemap.txt and writing it when something changed.

missing & removed sets controll if we need a new sitemap.txt
---

diff --git a/src/tree-cutter.py b/src/tree-cutter.py
index cf1ecb0..b21c091 100755
--- a/src/tree-cutter.py
+++ b/src/tree-cutter.py
@@ -295,6 +295,11 @@ class Sitemap():
         tokens = filter(None,re.split(r'(^/\w*/|\w*/)',link))
         self._tree.add(tokens,Link(link))
 
+    def write_map(self):
+        f = open(self._file,'w')
+        f.write('\n'.join(link.link() for link in self._tree))
+        f.close()
+
     def read_map(self):
         try:
             f = open(self._file)
@@ -506,11 +511,12 @@ missing = dir_.set() - sitemap.set()
 removed = sitemap.set() - dir_.set()
 for page in removed:
     print removed+' pages missing!!'
-
 for page in missing:
     print 'adding missing page '+page
-    sitemap.add_page(page)
-
+    sitemap.add_link(page)
+if len(missing & removed) != 0:
+    print 'writing new sitemap - please adjust if needed'
+    sitemap.write_map()
 sitemap.graph()