-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathserver.py
124 lines (90 loc) · 2.53 KB
/
server.py
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
123
124
from flask import Flask
from flask import render_template
from flask import request, session
from celery import Celery
from celery.task.control import revoke
import json
"""
from flaskcelery.tasks import make_celery
app = Flask('server')
app.config.update(
CELERY_BROKER_URL='amqp://guest@localhost//',
)
celery = make_celery(app)
# celery task
@celery.task()
def add_together(a, b):
import time
time.sleep(5)
return a + b
"""
app = Flask(__name__)
app.secret_key = 'some random secret key'
from flaskcelery.tasks import rates
def get_plot_names(s):
begin = 0
plotname = []
while True:
index = s.find("plot(", begin) # plot(hello, 'hi')
if -1 == index:
break
next = s.find(")", index)
left = s.find("\'", index)
right = s.find("\'", left + 1)
plotname.append(s[left + 1 : right])
begin = next + 1
return plotname
# webapp
@app.route("/")
def index():
return render_template('index.html')
@app.route("/start/", methods=['POST'])
def start():
print request.path
print request.method
# temporarily store session (very hacky, should store in DB)
print session
data = json.loads(request.data)
#exec data['code']
#print data['code']
task_id = rates.delay(data['code'], data['currency_pair']).id
session['TASK_ID'] = task_id
print task_id
response = {}
response['plotnames'] = get_plot_names(data['code'])
print response
return json.dumps(response)
@app.route("/stop/", methods=['POST'])
def stop():
print session
response = {}
response['status'] = 'fail'
# a task is already running
if "TASK_ID" not in session:
response['status'] = 'no_task_running'
else:
task_id = session['TASK_ID']
print ("A task is running with id: %s" % task_id)
revoke(task_id, terminate=True)
print ("TASK REVOKED")
del(session['TASK_ID'])
response['status'] = 'task_stopped'
print response
return json.dumps(response)
@app.route("/get-plot-points/")
def get_plot_points():
response = {}
if "TASK_ID" not in session:
response['status'] = 'no_task_running'
else:
response['status'] = 'task_running'
# running a task
task_id = session['TASK_ID']
task = rates.AsyncResult(task_id)
if task.info is not None:
response['data'] = task.info['plot_data']
print response
return json.dumps(response)
if __name__ == "__main__":
app.debug = True
app.run(host='0.0.0.0')