Content disposition header

2 files changed, 4 insertions, 2 deletions

diff --git a/main.py b/main.py
index a4766b1..2309bf8 100644
--- a/main.py
+++ b/main.py
@@ -50,7 +50,7 @@ def create_app(test_config=None):
         def remove_file(response):
             os.remove(complete_path)
             return response
-        return send_from_directory(app.config['UPLOAD_FOLDER'], filepath)
+        return send_from_directory(app.config['UPLOAD_FOLDER'], filepath, as_attachment=True)
 
     @app.route('/', methods=['GET', 'POST'])
     def upload_file():
@@ -180,7 +180,7 @@ def create_app(test_config=None):
                     os.remove(complete_path)
                     return response
 
-            return send_from_directory(app.config['UPLOAD_FOLDER'], filepath)
+            return send_from_directory(app.config['UPLOAD_FOLDER'], filepath, as_attachment=True)
 
     class APIBulkDownloadCreator(Resource):
         schema = {
diff --git a/test/test_api.py b/test/test_api.py
index 2029820..de297c4 100644
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -157,6 +157,7 @@ class Mat2APITestCase(unittest.TestCase):
 
         request = self.app.get(data['download_link'])
         self.assertEqual(request.status_code, 200)
+        self.assertIn('attachment; filename=test_name.cleaned.jpg', request.headers['Content-Disposition'])
 
         request = self.app.get(data['download_link'])
         self.assertEqual(request.status_code, 404)
@@ -219,6 +220,7 @@ class Mat2APITestCase(unittest.TestCase):
         self.assertEqual(request.headers['Content-Length'], '1596')
 
         request = self.app.get(response['download_link'])
+        self.assertIn('attachment; filename=files.', request.headers['Content-Disposition'])
         zip_response = zipfile.ZipFile(BytesIO(request.data))
         self.assertEquals(2, len(zip_response.namelist()))
         for name in zip_response.namelist():