1 files changed, 18 insertions, 3 deletions

diff --git a/server.py b/server.py
index 8937d4b..85ffad0 100644
--- a/server.py
+++ b/server.py
@@ -1,10 +1,20 @@
+"""
+server.py - sets up and runs the flask server
+"""
 import os
 from flask import Flask
 from siteconfig import siteconfig
 from flask_compress import Compress
+from flask_caching import Cache
 
 app = Flask(__name__)
 compress = Compress()
+cache = Cache(
+    config={
+        'CACHE_TYPE': siteconfig.CACHE_TYPE,
+        'CACHE_DEFAULT_TIMEOUT': siteconfig.CACHE_DEFAULT_TIMEOUT,
+    }
+)
 
 from views import *
 
@@ -25,17 +35,22 @@ def setup():
     else:
         s = "./templates/site/"
         top_dirs = [
-            x for x in os.listdir(s) \
-                if os.path.isdir(s + x) and not x.startswith(".")
+            x
+            for x in os.listdir(s)
+            if os.path.isdir(s + x) and not x.startswith(".")
         ]
         app.config.update({'MAIN_SITE_DIRS': sorted(top_dirs)})
 
     app.config.update({'DOMAIN': siteconfig.DOMAIN})
     app.config.update({'HOME_TITLE': siteconfig.HOME_TITLE})
-    app.config.update({'COMPRESS_MIMETYPES': siteconfig.COMPRESS_MIMETYPES})
+    app.config.update(
+        {'COMPRESS_MIMETYPES': siteconfig.COMPRESS_MIMETYPES}
+    )
+
 
 # Setup needs to come first to be compatible with wsgi
 setup()
 if __name__ == "__main__":
     compress.init_app(app)
+    cache.init_app(app)
     app.run()