From 91d00222d692b80a8eb22f0f193479545846ac89 Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Mon, 7 Aug 2017 22:01:15 +0300
Subject: Implement freeze.py

---
 freeze.py | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

(limited to 'freeze.py')

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()
-- 
cgit v1.3.1