gen-sitemap: build tree

diff --git a/gen-sitemap.py b/gen-sitemap.py
index 1afe424131947fac5bd56bb175d6ab734661fb71..1a78a7fb84b8fda32dbfdd1a3712ad8aa922668c 100755 (executable)
--- a/gen-sitemap.py
+++ b/gen-sitemap.py
@@ -20,6 +20,10 @@ os.chdir(args.root_dir)
 fullpath = os.getcwd()
 fp_len = len(fullpath)
 
+tree = ('', [])
+current_subtree = tree[1]
+save_level = 0
+
 for dirpath, dirs, files in sorted(os.walk(fullpath)):
     try:
         dirpath = dirpath[fp_len:]
@@ -30,6 +34,17 @@ for dirpath, dirs, files in sorted(os.walk(fullpath)):
                 raise StopIteration
         parts = dirpath.split('/')
         level = len(parts) - 2
-        print '    ' * level + parts[-1]
+        if level < save_level:
+            current_subtree = tree[1]
+            for i in range(level):
+                current_subtree = current_subtree[-1][1]
+            current_subtree.append((parts[-1], []))
+        elif level == save_level:
+            current_subtree.append((parts[-1], []))
+        elif level > save_level:
+            assert level - save_level == 1
+            current_subtree = current_subtree[-1][1]
+            current_subtree.append((parts[-1], []))
+        save_level = level
     except StopIteration:
         pass