Add gen-sitemap.py

[phdru.name/phdru.name.git] / gen-sitemap.py

diff --git a/gen-sitemap.py b/gen-sitemap.py
new file mode 100755 (executable)
index 0000000..d57d422
--- /dev/null
+++ b/gen-sitemap.py
@@ -0,0 +1,21 @@
+#! /usr/bin/env python
+
+import argparse
+from fnmatch import fnmatch
+import os
+
+parser = argparse.ArgumentParser(description='Generate sitemap')
+parser.add_argument('root_dir', help='Root dicrectory')
+args = parser.parse_args()
+
+os.chdir(args.root_dir)
+fullpath = os.getcwd()
+fp_len = len(fullpath)
+
+for dirpath, dirs, files in sorted(os.walk(fullpath)):
+    dirpath = dirpath[fp_len:]
+    if not dirpath:
+        continue
+    parts = dirpath.split('/')
+    level = len(parts) - 2
+    print '    ' * level + parts[-1]