save rendered images to disk so page loads faster

tdhopper · tdhopper · commit b6454563b5b3 · 2017-06-23T13:12:50.000-04:00
diff --git a/render.py b/render.py
@@ -9,6 +9,8 @@
 import markdown
 import logging
 import subprocess
+import base64
+import hashlib
 
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -46,7 +48,14 @@ def image_from_cell(cell):
     try:
         for c in cell['outputs']:
             if 'data' in c and 'image/png' in c['data']:
-                return c['data']['image/png'].replace("\n", "").strip()
+                base64_img = c['data']['image/png'].replace("\n", "").strip()
+                filename = hashlib.md5()
+                filename.update(base64_img.encode('ascii'))
+                web_path = "/img/plots/{}.png".format(filename.hexdigest())
+                full_path = "web" + web_path
+                with open(full_path, "wb") as fh:
+                    fh.write(base64.b64decode(base64_img))
+                return web_path
     except KeyError as e:
         logging.error("Can't find image in cell: %s", cell['source'])
         raise e
diff --git a/web/t_index.html b/web/t_index.html
@@ -131,8 +131,8 @@ <h3 class="card-title mb-1">{{ name }}</h3>
                     <div class="container-fluid">
                       <div class="row">
                         <div class="col-lg-6 col-xs-12 p-lg-0">
-                          <a href="data:image/png;base64,{{ plot['image'] }}" data-toggle="lightbox" data-title="{{ name }} - {{ plot['package'] }}">
-                            <img src="data:image/png;base64,{{ plot['image'] }}" class="img-fluid">
+                          <a href="{{ plot['image'] }}" data-toggle="lightbox" data-title="{{ name }} - {{ plot['package'] }}">
+                            <img src="{{ plot['image'] }}" class="img-fluid">
                           </a>
                         </div>
                         <div class="col-lg-6 col-xs-12 p-0 right-col d-inline-flex flex-column">
