add sitemap generation templates

billryan · billryan · commit d6e2b4c316df · 2015-10-01T13:48:38.000+08:00
diff --git a/scripts/sitemap/__init__.py b/scripts/sitemap/__init__.py
diff --git a/scripts/sitemap/templates/sitemap.xml b/scripts/sitemap/templates/sitemap.xml
@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">
+    {% for page in pages %}
+    <url>
+        <loc>{{root_url}}/{{page['url']}}</loc>
+        <lastmod>{{page['lastmod']}}</lastmod>
+        <changefreq>{{freq}}</changefreq>
+    </url>
+    {% endfor %}
+</urlset>
diff --git a/scripts/sitemap_gen.py b/scripts/sitemap_gen.py
@@ -3,7 +3,7 @@
 
 import os
 from pathlib import Path
-from jinja2 import Template
+from jinja2 import Environment, FileSystemLoader
 from subprocess import check_output
 
 BASE_DIR = os.path.abspath(os.path.dirname(__file__))
@@ -31,13 +31,17 @@ def sitemap(suffix='.md'):
                 p = p.with_suffix('.html')
                 fn = p.as_posix().lower()
                 page = {}
-                page['date'] = date
+                page['lastmod'] = date
                 page['url'] = fn
                 pages.append(page)
     root_url = 'http://algorithm.yuanbin.me'
-    template = Template(os.path.join(BASE_DIR, 'sitemap.xml'))
-    template.render(root_url=root_url, pages=pages)
-    # print(sitemap_xml)
+    templates = os.path.join(BASE_DIR, 'sitemap' + os.sep + 'templates')
+    env = Environment(loader=FileSystemLoader(templates))
+    template = env.get_template('sitemap.xml')
+    sitemap_xml = template.render(root_url=root_url, pages=pages, freq='daily')
+    sitemap_fn = os.path.join(ROOT_DIR, 'sitemap.xml')
+    with open(sitemap_fn, 'w') as sf:
+        sf.write(sitemap_xml)
 
 
 if __name__ == "__main__":
