X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=gen-sitemap.py;h=333069f0122df57fdad96776356d9ed175467509;hb=3b607b20bbfbc5b8a2282d57b4c1f99f07f9d272;hp=088f83819e4c8d491fa0ac69e4eb8ab632c3cb06;hpb=faa336e3b58a3a51edcfd5da191dc0dcdac3b2be;p=phdru.name%2Fphdru.name.git

diff --git a/gen-sitemap.py b/gen-sitemap.py
index 088f838..333069f 100755
--- 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()
@@ -49,23 +50,24 @@ for dirpath, dirs, files in sorted(os.walk(fullpath)):
         pass
 
 def _tree2html(tree, level=0):
-    indent = "  " * level
-    strings = []
-    strings.append(indent + "<ul>")
-
-    level += 1
-    indent2 = "  " * level
-
+    subparts = []
+    indent = "  " * (level + 1)
     for title, subtree in tree:
         if subtree:
-            strings.append(indent2 + "<li>%s" % title)
-            strings.append(_tree2html(subtree, level+1))
-            strings.append(indent2 + "</li>")
+            subparts.append(indent + "<li>%s" % title)
+            subparts.append(_tree2html(subtree, level+2))
+            subparts.append(indent + "</li>")
         else:
-            strings.append(indent2 + "<li>%s</li>" % title)
+            subparts.append(indent + "<li>%s</li>" % title)
+    s = "\n".join(subparts)
+
+    parts = []
+    indent = "  " * level
+    parts.append(indent + "<ul>")
+    parts.append(s)
+    parts.append(indent + "</ul>")
 
-    strings.append(indent + "</ul>")
-    return '\n'.join(strings)
+    return "\n".join(parts)
 
 sitemap_tmpl = ["""\
 #extends phd_site