1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
from config import *
import subprocess
import flask
from flask import send_file
from flask import Flask, url_for, jsonify, request
from werkzeug import secure_filename
import json
import os
import os.path
import base64
import re
from os.path import abspath, dirname, exists, isfile, join, splitext
import config
if not os.path.exists(TEMP_DIR):
os.makedirs(TEMP_DIR)
app = Flask(__name__)
@app.route('/')
def api_root():
headers = request.headers
auth = headers.get("X-Api-Key")
if auth == AUTH_KEY:
return jsonify({"message": "OK: Authorized"}), 200
else:
return jsonify({"message": "ERROR: Unauthorized"}), 401
@app.route('/rscript', methods=['GET', 'POST'])
def get_data():
# Validate the request body contains JSON
headers = request.headers
auth = headers.get("X-Api-Key")
if auth == AUTH_KEY:
if request.is_json:
req_data = json.loads(request.get_json())
user_id = req_data["user_id"]
user_dir = TEMP_DIR + user_id
R_file_id = req_data["R_file_id"]
code = req_data["code"]
if not os.path.exists(user_dir):
os.makedirs(user_dir)
file_path = user_dir + '/' + R_file_id + '.R'
plot_path = user_dir + '/' + R_file_id + '.png'
f = open(file_path, "w")
f.write('png("{0}");\n'.format(plot_path))
f.write('\n')
f.write(code)
f.write('\n')
f.write("while (!is.null(dev.list())) dev.off()")
f.close()
processed_data = subprocess.Popen(['Rscript', file_path],
stdout=subprocess.PIPE,
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 +
'&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,
"status": "200"
}
else:
response_body = {
"status": "Invalid authentication request",
}
else:
response_body = {
"status": "400",
}
result = jsonify(response_body)
return result
@app.route('/plot', methods=['GET', 'POST'])
def get_plot():
user_id = request.args.get('user_id')
R_file_id = request.args.get('R_file_id')
user_dir = TEMP_DIR + user_id
plot_path = user_dir + '/' + R_file_id + '.png'
try:
return send_file(plot_path, mimetype='image/png', as_attachment=True)
except Exception as e:
print("Error generated in")
@app.route('/upload-temp-file', methods=['GET', 'POST'])
def upload_file():
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__':
if (PRODUCTION == True):
app.run(debug=False)
else:
app.run(debug=True)
app.run(port=HTTP_PORT, host=HTTP_HOST)
|
