From 3e21e6e13b1b5c3d87643f9246c44273f49c41d6 Mon Sep 17 00:00:00 2001
From: burnettk <burnettk@users.noreply.github.com>
Date: Tue, 31 May 2022 13:26:31 -0400
Subject: [PATCH] beginning of upload stuff

---
 .../routes/admin_blueprint/admin_blueprint.py | 25 ++++++++++++++++++-
 .../templates/process_model_show.html         |  5 ++++
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/src/spiff_workflow_webapp/routes/admin_blueprint/admin_blueprint.py b/src/spiff_workflow_webapp/routes/admin_blueprint/admin_blueprint.py
index e9536898..3684f3ad 100644
--- a/src/spiff_workflow_webapp/routes/admin_blueprint/admin_blueprint.py
+++ b/src/spiff_workflow_webapp/routes/admin_blueprint/admin_blueprint.py
@@ -1,7 +1,7 @@
 """APIs for dealing with process groups, process models, and process instances."""
 from typing import Any
 
-from flask import Blueprint, render_template
+from flask import Blueprint, render_template, redirect, url_for
 from flask_bpmn.models.db import db
 from flask import request
 
@@ -47,6 +47,29 @@ def process_model_show_file(process_model_id, file_name):
     return render_template('process_model_show.html', process_model=process_model, bpmn_xml=bpmn_xml, files=files, current_file_name=file_name)
 
 
+@admin_blueprint.route("/process-models/<process_model_id>/upload-file", methods=["POST"])
+def process_model_upload_file(process_model_id):
+    """Process_model_upload_file."""
+    process_model = ProcessModelService().get_spec(process_model_id)
+    bpmn_xml = SpecFileService.get_data(process_model, file_name)
+    files = SpecFileService.get_files(process_model, extension_filter="bpmn")
+
+    if 'file' not in request.files:
+            flash('No file part')
+            return redirect(url_for('admin.process_model_show', process_model_id=process_model.id))
+    file = request.files['file']
+    # If the user does not select a file, the browser submits an
+    # empty file without a filename.
+    if file.filename == '':
+        flash('No selected file')
+        return redirect(request.url)
+    if file and allowed_file(file.filename):
+        filename = secure_filename(file.filename)
+        file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+        return redirect(url_for('download_file', name=filename))
+
+    return render_template('process_model_show.html', process_model=process_model, bpmn_xml=bpmn_xml, files=files, current_file_name=file_name)
+
 @admin_blueprint.route("/process_models/<process_model_id>/edit", methods=["GET"])
 def process_model_edit(process_model_id):
     """Edit_bpmn."""
diff --git a/src/spiff_workflow_webapp/routes/admin_blueprint/templates/process_model_show.html b/src/spiff_workflow_webapp/routes/admin_blueprint/templates/process_model_show.html
index e7b13740..316cf50e 100644
--- a/src/spiff_workflow_webapp/routes/admin_blueprint/templates/process_model_show.html
+++ b/src/spiff_workflow_webapp/routes/admin_blueprint/templates/process_model_show.html
@@ -65,6 +65,11 @@ html, body, #canvas {
   </ul>
   {% endif %}
 
+  <form method=post action="/admin/process-models/{{process_model.id}}/upload-file" enctype=multipart/form-data>
+    <input type=file name=file>
+    <input type=submit value=Upload>
+  </form>
+
   <div id="canvas"></div>
 
   <meta id="bpmn_xml" data-name="{{bpmn_xml}}">