From 22ae890dfe29c7afeb9300307239edbd11a418cd Mon Sep 17 00:00:00 2001 From: Fredrik Unger Date: Sun, 10 Apr 2011 16:30:42 +0200 Subject: [PATCH] Adding method to write a sitemap.txt and writing it when something changed. missing & removed sets controll if we need a new sitemap.txt --- src/tree-cutter.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) 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() -- 2.30.2