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