1
|
#!/usr/bin/python
|
2
|
|
3
|
from flask import Flask, jsonify, request
|
4
|
from flask.json import JSONEncoder
|
5
|
from eu.dnetlib.es_connector import DLIESConnector
|
6
|
from flask_prometheus import monitor
|
7
|
|
8
|
from logger import dlilogger
|
9
|
|
10
|
DLI_INDEX = 'dli'
|
11
|
|
12
|
ES_HOST = 'ip-90-147-167-27.ct1.garrservices.it'
|
13
|
|
14
|
|
15
|
class MyJSONEncoder(JSONEncoder):
|
16
|
def default(self, obj):
|
17
|
return obj.__dict__
|
18
|
|
19
|
|
20
|
app = Flask(__name__)
|
21
|
app.debug = False
|
22
|
app.json_encoder = MyJSONEncoder
|
23
|
|
24
|
base_dnet_url = "http://localhost:8280/app"
|
25
|
|
26
|
|
27
|
# q = QueryResolver("http://localhost:8983/solr/DLIF-index-cleaned_shard1_replica1")
|
28
|
|
29
|
class InvalidUsage(Exception):
|
30
|
status_code = 400
|
31
|
|
32
|
def __init__(self, message, status_code=None, payload=None):
|
33
|
Exception.__init__(self)
|
34
|
self.message = message
|
35
|
if status_code is not None:
|
36
|
self.status_code = status_code
|
37
|
self.payload = payload
|
38
|
|
39
|
def to_dict(self):
|
40
|
rv = dict(self.payload or ())
|
41
|
rv['message'] = self.message
|
42
|
return rv
|
43
|
|
44
|
|
45
|
# Routes
|
46
|
@app.route('/')
|
47
|
def root():
|
48
|
return app.send_static_file('index.html')
|
49
|
|
50
|
|
51
|
@app.route('/<path:path>')
|
52
|
def static_proxy(path):
|
53
|
# send_static_file will guess the correct MIME type
|
54
|
return app.send_static_file(path)
|
55
|
|
56
|
|
57
|
@app.route('/api/item/<path:identifier>', methods=['post', 'get'])
|
58
|
def get_item(identifier):
|
59
|
if identifier:
|
60
|
object_type = request.form.get('type')
|
61
|
start = request.form.get('from')
|
62
|
if len(object_type) == 0:
|
63
|
object_type = None
|
64
|
if len(start) == 0:
|
65
|
start = None
|
66
|
else:
|
67
|
start = int(start)
|
68
|
connector = DLIESConnector(ES_HOST, DLI_INDEX)
|
69
|
return jsonify(result=connector.item_by_id(identifier, object_type, start))
|
70
|
else:
|
71
|
raise InvalidUsage('This view is gone', status_code=410)
|
72
|
|
73
|
|
74
|
@app.route('/api/stats/', methods=['post', 'get'])
|
75
|
def stats():
|
76
|
raise Exception("Method to be implemented")
|
77
|
# q = QueryResolver(host, base_dnet_url)
|
78
|
# return jsonify(stats=q.get_stats())
|
79
|
|
80
|
|
81
|
@app.route('/api/ds_info/', methods=['post', 'get'])
|
82
|
def info():
|
83
|
raise Exception("Method to be implemented")
|
84
|
# q = QueryResolver(host, base_dnet_url)
|
85
|
# if 'id' in request.form:
|
86
|
# id = request.form['id']
|
87
|
# name = request.form['name']
|
88
|
# result = q.get_ds_info(id, name)
|
89
|
# return jsonify(info=result)
|
90
|
# result = q.get_all_ds()
|
91
|
# return jsonify(info=result)
|
92
|
|
93
|
|
94
|
@app.route('/api/stats_detail/', methods=['post', 'get'])
|
95
|
def stats_detail():
|
96
|
return ""
|
97
|
|
98
|
# q = QueryResolver(host, base_dnet_url)
|
99
|
# return jsonify(stats=q.get_stats_detail())
|
100
|
|
101
|
|
102
|
@app.route('/api/post/', methods=['post', 'get'])
|
103
|
def query_post():
|
104
|
action = None
|
105
|
query = None
|
106
|
filter_key = None
|
107
|
filter_value = None
|
108
|
result = {}
|
109
|
|
110
|
start = 0
|
111
|
if 'action' in request.form:
|
112
|
action = request.form['action']
|
113
|
if 'query' in request.form:
|
114
|
query = request.form['query']
|
115
|
if 'start' in request.form:
|
116
|
start = int(request.form['start'])
|
117
|
if 'filter' in request.form:
|
118
|
filter_key = request.form['filter']
|
119
|
|
120
|
if action == 'query' and query is not None:
|
121
|
connector = DLIESConnector(ES_HOST, DLI_INDEX)
|
122
|
try:
|
123
|
result = connector.simple_query(query, start=start, user_filter=filter_key)
|
124
|
except Exception as e:
|
125
|
print e
|
126
|
return jsonify(result=result)
|
127
|
else:
|
128
|
return jsonify(error={'error_code': 'an error occur during query on server'})
|
129
|
|
130
|
|
131
|
if __name__ == '__main__':
|
132
|
app.logger.addHandler(dlilogger)
|
133
|
monitor(app, port=8000)
|
134
|
app.run()
|