-
-
Notifications
You must be signed in to change notification settings - Fork 165
/
server.py
97 lines (87 loc) · 4 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
import logging
import sys
import json
import numpy as np
import random
import pandas as pd
from .params import ObservationParam
def convert(o):
if isinstance(o, np.generic): return o.item()
raise TypeError
def start_server(arena, host, port, disable_logs):
from flask import Flask, request, abort, Response
from flask_cors import CORS
import requests
cli = sys.modules['flask.cli']
cli.show_server_banner = lambda *x: None
app = Flask(__name__)
CORS(app)
shutdown_token = str(random.randrange(2<<63))
log = logging.getLogger('werkzeug')
log.disabled = disable_logs
app.logger.disabled = disable_logs
arena._stop_server = lambda: requests.get('http://' + host + ':' + str(port) + '/shutdown?token=' + shutdown_token)
@app.route("/", methods=['GET'])
def main():
result = {
'version': '1.2.0',
'api': 'arenar_api',
'timestamp': arena.timestamp*1000,
'availableParams': arena.list_available_params(),
'availablePlots': [plot.info for plot in arena.get_supported_plots()],
'options': { 'attributes': arena.enable_attributes, 'customParams': arena.enable_custom_params }
}
return Response(json.dumps(result, default=convert), content_type='application/json')
def get_params(request):
result = {}
custom_params = False
for param_type in ['model', 'observation', 'variable', 'dataset']:
param_label = request.args.get(param_type)
if (not param_label is None) and param_label.startswith('{') and param_label.endswith('}') and param_type == 'observation':
custom_params = True
try:
obj = json.loads(param_label)
data = result['model'].explainer.data if not result['model'] is None else None
if data is None:
raise Exception('Custom observation requires model param')
df = data.head(n=0).append({ k: v for k, v in obj.items() if k in data.columns }, ignore_index=True,).tail(n=1).reset_index(drop=True)
param = ObservationParam(df, 0)
result[param_type] = param
except:
raise Exception('Invalid custom observation')
else:
param_value = arena.find_param_value(param_type, request.args.get(param_type))
if not param_value is None:
result[param_type] = param_value
return (result, custom_params)
@app.route("/<string:plot_type>", methods=['GET'])
def get_plot(plot_type):
if plot_type == 'timestamp':
return Response(json.dumps({'timestamp': arena.timestamp * 1000}, default=convert), content_type='application/json')
elif plot_type == 'shutdown':
if request.args.get('token') != shutdown_token:
abort(403)
return
shutdown = request.environ.get('werkzeug.server.shutdown')
if shutdown is None:
raise Exception('Failed to stop the server.')
shutdown()
return ''
params, custom_params = get_params(request)
if not arena.enable_custom_params and custom_params:
abort(403)
return
try:
result = arena.plots_manager.get_plot(plot_type, params, cache=not custom_params)
return Response(json.dumps(result.serialize(), default=convert), content_type='application/json')
except Exception as e:
abort(404)
return
@app.route("/attribute/<string:param_type>/<string:param_label>", methods=['GET'])
def get_attribute(param_type, param_label):
if not param_type in ['model', 'variable', 'observation', 'dataset']:
abort(404)
return
result = arena.get_param_attributes(param_type, param_label)
return Response(json.dumps(result, default=convert), content_type='application/json')
app.run(host=host, port=port)