X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=gen-sitemap.py;h=333069f0122df57fdad96776356d9ed175467509;hb=3b607b20bbfbc5b8a2282d57b4c1f99f07f9d272;hp=1a78a7fb84b8fda32dbfdd1a3712ad8aa922668c;hpb=8383bd5528ad8c6f3395ab2aaac5f068514965fe;p=phdru.name%2Fphdru.name.git diff --git a/gen-sitemap.py b/gen-sitemap.py index 1a78a7f..333069f 100755 --- a/gen-sitemap.py +++ b/gen-sitemap.py @@ -3,6 +3,7 @@ import argparse from fnmatch import fnmatch import os +from news import write_if_changed parser = argparse.ArgumentParser(description='Generate sitemap') parser.add_argument('-x', '--exclude', action='append', @@ -11,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() @@ -38,13 +40,49 @@ for dirpath, dirs, files in sorted(os.walk(fullpath)): 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], [])) + #else: # level == save_level: + current_subtree.append((parts[-1], [])) save_level = level except StopIteration: pass + +def _tree2html(tree, level=0): + subparts = [] + indent = " " * (level + 1) + for title, subtree in tree: + if subtree: + subparts.append(indent + "