added auth key for upload

1 files changed, 24 insertions, 13 deletions

diff --git a/r_api.py b/r_api.py
index f8d614c..9dbd83e 100644
--- a/r_api.py
+++ b/r_api.py
@@ -57,22 +57,23 @@ def get_data():
                                               stderr=subprocess.PIPE)
             ret_data, err = processed_data.communicate()
             is_plot = str(os.path.isfile(plot_path))
-            plot_path_req = (API_URL_PLOT,
-                             '?user_id=' + user_id,
+            plot_path_req = (API_URL_PLOT +
+                             '?user_id=' + user_id +
                              '&R_file_id=' + R_file_id)
             response_body = {
                 "data": ret_data.decode("utf-8"),
                 "error": err.decode("utf-8"),
                 "is_plot": is_plot,
-                "plot_path_req": plot_path_req
+                "plot_path_req": plot_path_req,
+                "status": "200"
             }
         else:
             response_body = {
-                "auth_error": "Invalid authentication request",
+                "status": "Invalid authentication request",
             }
     else:
         response_body = {
-            "auth_error": "400",
+            "status": "400",
         }
     result = jsonify(response_body)
     return result
@@ -92,14 +93,24 @@ def get_plot():
 
 @app.route('/upload-temp-file', methods=['GET', 'POST'])
 def upload_file():
-    if request.method == 'POST':
-        f = request.files['file']
-        user_id = request.form.get('user_id')
-        user_dir = TEMP_DIR + user_id
-        uploaded_file = secure_filename(f.filename)
-        f.save(os.path.join(user_dir, uploaded_file))
-        print("done")
-        return 'file uploaded successfully'
+    headers = request.headers
+    auth = headers.get("X-Api-Key")
+    if auth == AUTH_KEY:
+        if request.method == 'POST':
+            f = request.files['file']
+            user_id = request.form.get('user_id')
+            user_dir = TEMP_DIR + user_id
+            if not os.path.exists(user_dir):
+                os.makedirs(user_dir)
+            uploaded_file = secure_filename(f.filename)
+            f.save(os.path.join(user_dir, uploaded_file))
+            print("done")
+            return 'file uploaded successfully'
+        else:
+            print("Post request fail")
+    else:
+        print("Wrong authentication key")
+        return jsonify({"message": "ERROR: Unauthorized"}), 401
 
 
 if __name__ == '__main__':