From 00380ac6020179de15929afc1b29b7dc5ce094e4 Mon Sep 17 00:00:00 2001
From: jaseg <git@jaseg.net>
Date: Sun, 29 Sep 2019 13:30:38 +0200
Subject: Add STL renderer

---
 webapp/pogojig.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'webapp/pogojig.py')

diff --git a/webapp/pogojig.py b/webapp/pogojig.py
index baaca5a..9312cf5 100644
--- a/webapp/pogojig.py
+++ b/webapp/pogojig.py
@@ -93,9 +93,12 @@ def upload_svg():
         flash(f'Error uploading SVG file: {"; ".join(msg for elem in upload_form.errors.values() for msg in elem)}', 'error')
     return redirect(url_for('jigerator'))
 
-@app.route('/render/download')
-def render_download():
-    return send_file(tempfile_path(f'renders.zip'),
+@app.route('/render/download/<file>')
+def render_download(file):
+    if file not in ['jig.stl', 'pcb_shape.dxf', 'kicad.zip', 'sources.zip']:
+        abort(404)
+
+    return send_file(tempfile_path(file),
             mimetype='application/zip',
             as_attachment=True,
             attachment_filename=f'{path.splitext(session["filename"])[0]}_pogojig.zip')
-- 
cgit