Feat(blog): Encode tags

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

diff --git a/gen-sitemap.py b/gen-sitemap.py
index aaff97a7c3001f77707b32876da324d3bc20b73a..32c729dfabefcc7d9047e9c2a442aaae46bb8ed6 100755 (executable)
--- a/gen-sitemap.py
+++ b/gen-sitemap.py
@@ -12,10 +12,11 @@ parser.add_argument('root_dir', help='Root dicrectory')
 args = parser.parse_args()
 
 exclude = []
-for pat in args.exclude:
-    exclude.append(pat)
-    if not pat.endswith('/*'):
-        exclude.append(pat + '/*')
+if args.exclude:
+    for pat in args.exclude:
+        exclude.append(pat)
+        if not pat.endswith('/*'):
+            exclude.append(pat + '/*')
 
 os.chdir(args.root_dir)
 fullpath = os.getcwd()
@@ -48,16 +49,20 @@ for dirpath, dirs, files in sorted(os.walk(fullpath)):
     except StopIteration:
         pass
 
-def _tree2html(tree, level=0):
+def _tree2html(tree, path='', level=0):
     subparts = []
     indent = "  " * (level + 1)
     for title, subtree in tree:
+        subpath = "%s/%s" % (path, title)
+        if subpath.startswith('/'):
+            subpath = subpath[1:]
+        href = '<a href="%s/">%s</a>' % (subpath, title)
         if subtree:
-            subparts.append(indent + "<li>%s" % title)
-            subparts.append(_tree2html(subtree, level+2))
+            subparts.append(indent + "<li>%s" % href)
+            subparts.append(_tree2html(subtree, subpath, level+2))
             subparts.append(indent + "</li>")
         else:
-            subparts.append(indent + "<li>%s</li>" % title)
+            subparts.append(indent + "<li>%s</li>" % href)
     s = "\n".join(subparts)
 
     parts = []