add session clear endpoint

Author: rhsimplex

picasso/picasso.py

@@ -29,6 +29,7 @@
 import io
 import time
 import inspect
+import shutil
 from operator import itemgetter
 from tempfile import mkdtemp
 from importlib import import_module
@@ -102,7 +103,6 @@ def initialize_new_session():
         session['img_input_dir'] = mkdtemp()
         session['img_output_dir'] = mkdtemp()
 
-
 def get_visualizations():
     """Get visualization classes in context
 
@@ -221,6 +221,14 @@ def api_visualize():
     return jsonify(output=output)
 
 
+@app.route('/api/reset', methods=['GET'])
+def end_session():
+    shutil.rmtree(session['img_input_dir'])
+    shutil.rmtree(session['img_output_dir'])
+    session.clear()
+    return jsonify(ok='true')
+
+
 @app.route('/', methods=['GET', 'POST'])
 def landing():
     """Landing page for the application

tests/test_picasso.py

@@ -103,6 +103,10 @@ def test_listing_images(self, client):
         response = client.get(url_for('api_images'))
         assert response.status_code == 200
 
+    def test_end_session(self, client):
+        response = client.get(url_for('end_session'))
+        assert response.status_code == 200
+
 
 class TestBaseModel: