Dynamically generate sample images

Author: tylermenezes

api/.gitignore

@@ -3,3 +3,4 @@ zip/
 __pycache__/
 build/
 remote/
+preview/

api/example.json

@@ -0,0 +1,48 @@
+{
+  "id": "region",
+  "name": "Hometown",
+  "webname": "region",
+  "abbr": "NIL",
+  "location": {
+    "lat": 0,
+    "lng": 0
+  },
+  "timezone": "America/Los_Angeles",
+  "timezone_group": "Pacific",
+  "current_event": {
+    "id": "example",
+    "name": "CodeDay Hometown Winter 1970",
+    "region_name": "Hometown",
+    "region_id": "region",
+    "webname": "region",
+    "timezone": "America/Los_Angeles",
+    "hashtag": "region",
+    "urls": {
+      "home": "https://codeday.org/region",
+      "register": "https://codeday.org/region/register"
+    },
+    "starts_at": 1581796800,
+    "ends_at": 1581883200,
+    "venue": {
+      "name": "VenueCorp",
+      "address": {
+        "line_1": "555 Post St",
+        "line_2": null,
+        "city": "SF",
+        "state": "CA",
+        "postal": "94102",
+        "country": "US"
+      },
+      "full_address": "555 Post St, SF CA 94102, US"
+    },
+    "loaners": {
+      "total": 0,
+      "available": 0
+    },
+    "venue_agreement": null,
+    "stripe_public_key": "pk_v0AIOIy377403GN0FKltGR9gOAAUe",
+    "notice": null,
+    "overflow_event": null,
+    "has_related_events": false
+  }
+}

api/generator.py

@@ -1,5 +1,6 @@
 import datetime, pytz, os, subprocess
 from jinja2 import Environment, FileSystemLoader, select_autoescape
+import hashlib
 
 from starlette.responses import RedirectResponse, FileResponse
 
@@ -13,6 +14,7 @@ def __init__(self, data, promo=None, promoFor=None):
     self.supported_formats = ['svg', 'pdf', 'png']
     self.promo = promo
     self.promoFor = promoFor
+    self.hash = hashlib.md5('{}_{}'.format(promo,promoFor).encode('utf8')).hexdigest()
 
     self.data = data
     for key in data:
@@ -29,7 +31,9 @@ def __init__(self, data, promo=None, promoFor=None):
 
   def get_cache(self, file_format, template_name='', full=True):
     basedir = os.path.dirname(os.path.realpath(__file__)) if full else ''
-    return '{}/generated/{}/{}/{}_{}_{}'.format(basedir, self.current_event['id'], file_format, self.promo, self.promoFor, template_name.replace('svg', file_format))
+    fname = '{}_{}'.format(self.hash, template_name.replace('svg', file_format)) if template_name else ''
+    return '{}/generated/{}/{}/{}'.format(basedir, self.current_event['id'], file_format, fname)
+
 
   def make_poster(self, template_name, file_format):
     file_format = file_format.lower()

api/main.py

@@ -1,17 +1,16 @@
 import requests, json, shutil
 
-from fastapi import FastAPI
+from fastapi import FastAPI, BackgroundTasks
 from jinja2 import Environment, FileSystemLoader, select_autoescape
 
 from starlette.middleware.cors import CORSMiddleware
 from starlette.responses import HTMLResponse, FileResponse, JSONResponse
 from starlette.staticfiles import StaticFiles
 
 from generator import PosterGenerator
-from tasks import sync_templates, cleanup
+from tasks import run_tasks
 
-sync_templates()
-cleanup()
+run_tasks()
 
 env = Environment(
   loader= FileSystemLoader('./remote/templates/template'),
@@ -20,7 +19,7 @@
 
 app = FastAPI()
 
-app.mount("/preview", StaticFiles(directory="./remote/templates/preview"), name="preview")
+app.mount("/preview", StaticFiles(directory="./preview"), name="preview")
 app.mount("/static", StaticFiles(directory="./build/static"), name="static")
 
 app.add_middleware(
@@ -40,9 +39,8 @@ def root():
   return HTMLResponse(file_get_contents("./build/index.html"))
 
 @app.get('/sync')
-def sync():
-  sync_templates()
-  cleanup()
+def sync(background_tasks: BackgroundTasks):
+  background_tasks.add_task(run_tasks)
   return HTMLResponse('ok')
 
 @app.get("/render/{id}/{template}/{file_format}")
@@ -62,7 +60,7 @@ def generate_all(id, promo=None):
     eventJson = json.loads(eventRequest.text)
   except:
     return "No event found with id {}".format(id),404
-  PosterGenerator(eventJson, promo).make_posters(env.list_templates())
+  PosterGenerator(eventJson, promo, promoFor).make_posters(env.list_templates())
 
   return FileResponse(shutil.make_archive('zip/{}'.format(id), 'zip', 'generated/{}'.format(eventJson.current_event['id'])), filename='{}.zip'.format(id))
 

api/tasks.py

@@ -1,6 +1,9 @@
 import time, threading
 import os, tempfile, zipfile, shutil
 from urllib import request
+from generator import PosterGenerator
+from jinja2 import Environment, FileSystemLoader, select_autoescape
+import json
 
 def sync_templates():
   zip_path = os.path.join(tempfile.mkdtemp(), 'templates.zip')
@@ -20,6 +23,25 @@ def sync_templates():
   shutil.rmtree('remote/templates_old', ignore_errors=True)
   os.remove(zip_path)
 
+
+  # Generate samples
+  env = Environment(
+    loader= FileSystemLoader('./remote/templates/template'),
+    autoescape=select_autoescape(['svg'])
+  )
+
+  with open('example.json', 'r') as exFile:
+    data=json.loads(exFile.read())
+
+  shutil.rmtree('generated/example', ignore_errors=True)
+  PosterGenerator(data, 'PROMO', '20% off').make_posters(env.list_templates())
+  shutil.rmtree('preview', ignore_errors=True)
+  os.rename('generated/example/png', 'preview')
+  shutil.rmtree('generated/example', ignore_errors=True)
+  for f in os.listdir('preview'):
+    os.rename('preview/{}'.format(f), 'preview/{}'.format(f[str.index(f, '_')+1:]));
+
+
 def cleanup():
   for dir in ('generated', 'zip'):
     if (os.path.exists(dir)):
@@ -31,7 +53,11 @@ def cleanup():
           else:
             os.remove(path)
 
-if __name__ == "__main__":
-  print("Running tasks...")
+def run_tasks():
   sync_templates()
   cleanup()
+
+
+if __name__ == "__main__":
+  print("Running tasks...")
+  run_tasks()