Implement freeze.py

1 files changed, 15 insertions, 1 deletions

diff --git a/freeze.py b/freeze.py
index d0f2bf2..78ed52c 100644
--- a/freeze.py
+++ b/freeze.py
@@ -1,7 +1,21 @@
 from flask_frozen import Freezer
-from app import app
+from app import app, BLOG_CONTENT_DIR, WIKI_CONTENT_DIR
+from os import listdir
 
 freezer = Freezer(app)
 
+@freezer.register_generator
+def blog_post():
+    for f in listdir(BLOG_CONTENT_DIR):
+        y, m, d, *title = f[:-3].split('-')
+        slug = '-'.join(title)
+        yield {'y': y, 'm': m, 'd': d, 'slug': slug}
+
+@freezer.register_generator
+def wiki_page():
+    for f in listdir(WIKI_CONTENT_DIR):
+        slug = f[:-3]
+        yield {'slug': slug}
+
 if __name__ == '__main__':
     freezer.freeze()