1
|
#!/usr/bin/python
|
2
|
|
3
|
import json
|
4
|
|
5
|
from flask import Flask, jsonify, request
|
6
|
from flask.json import JSONEncoder
|
7
|
|
8
|
from DLIQueryResolver import QueryResolver
|
9
|
from DLI_ObjectResolver import DLIObject, DLIRelation
|
10
|
from logger import dlilogger
|
11
|
|
12
|
|
13
|
class MyJSONEncoder(JSONEncoder):
|
14
|
def default(self, obj):
|
15
|
if isinstance(obj, DLIObject):
|
16
|
return obj.__dict__
|
17
|
if isinstance(obj, DLIRelation):
|
18
|
return obj.__dict__
|
19
|
return super(MyJSONEncoder, self).default(obj)
|
20
|
|
21
|
|
22
|
app = Flask(__name__)
|
23
|
app.debug = False
|
24
|
app.json_encoder = MyJSONEncoder
|
25
|
|
26
|
host = "http://localhost:8983/solr/DLIF-index-cleaned"
|
27
|
base_dnet_url = "http://localhost:8080/dli"
|
28
|
# host = "http://node0-p-dli.d4science.org:8983/solr/DLIF-index-cleaned"
|
29
|
# base_dnet_url = "http://node0-p-dli.d4science.org:8080/dli"
|
30
|
|
31
|
|
32
|
# q = QueryResolver("http://localhost:8983/solr/DLIF-index-cleaned_shard1_replica1")
|
33
|
|
34
|
class InvalidUsage(Exception):
|
35
|
status_code = 400
|
36
|
|
37
|
def __init__(self, message, status_code=None, payload=None):
|
38
|
Exception.__init__(self)
|
39
|
self.message = message
|
40
|
if status_code is not None:
|
41
|
self.status_code = status_code
|
42
|
self.payload = payload
|
43
|
|
44
|
def to_dict(self):
|
45
|
rv = dict(self.payload or ())
|
46
|
rv['message'] = self.message
|
47
|
return rv
|
48
|
|
49
|
|
50
|
# Routes
|
51
|
@app.route('/')
|
52
|
def root():
|
53
|
return app.send_static_file('index.html')
|
54
|
|
55
|
|
56
|
@app.route('/<path:path>')
|
57
|
def static_proxy(path):
|
58
|
# send_static_file will guess the correct MIME type
|
59
|
return app.send_static_file(path)
|
60
|
|
61
|
|
62
|
@app.route('/api/item/<path:identifier>', methods=['post', 'get'])
|
63
|
def getItem(identifier):
|
64
|
if identifier:
|
65
|
q = QueryResolver(host, base_dnet_url)
|
66
|
return jsonify(result=q.get_item(identifier))
|
67
|
else:
|
68
|
raise InvalidUsage('This view is gone', status_code=410)
|
69
|
|
70
|
|
71
|
@app.route('/api/stats/', methods=['post', 'get'])
|
72
|
def stats():
|
73
|
q = QueryResolver(host, base_dnet_url)
|
74
|
return jsonify(stats=q.get_stats())
|
75
|
|
76
|
|
77
|
@app.route('/api/ds_info/', methods=['post', 'get'])
|
78
|
def info():
|
79
|
q = QueryResolver(host, base_dnet_url)
|
80
|
if 'id' in request.form:
|
81
|
id = request.form['id']
|
82
|
name = request.form['name']
|
83
|
result = q.get_ds_info(id, name)
|
84
|
return jsonify(info=result)
|
85
|
result = q.get_all_ds()
|
86
|
return jsonify(info=result)
|
87
|
|
88
|
|
89
|
@app.route('/api/stats_detail/', methods=['post', 'get'])
|
90
|
def stats_detail():
|
91
|
q = QueryResolver(host, base_dnet_url)
|
92
|
return jsonify(stats=q.get_stats_detail())
|
93
|
|
94
|
@app.route('/api/makeQuery/', methods=[ 'get'])
|
95
|
def makeQuery():
|
96
|
if 'q' in request.args:
|
97
|
query = request.args['q']
|
98
|
|
99
|
q = QueryResolver(host, base_dnet_url)
|
100
|
try:
|
101
|
result = q.query(q=query, browse_fields=["entitytype", "provenance"], start=0, only_result=True,
|
102
|
filter_query=query)
|
103
|
except Exception as e:
|
104
|
print e
|
105
|
return jsonify(result=result)
|
106
|
|
107
|
|
108
|
return jsonify(error={'error_code': 'an error occur during query on server'})
|
109
|
|
110
|
|
111
|
|
112
|
|
113
|
|
114
|
@app.route('/api/post/', methods=['post', 'get'])
|
115
|
def query_post():
|
116
|
action = None
|
117
|
query = None
|
118
|
only_result = False
|
119
|
filter_query = None
|
120
|
start = 0
|
121
|
if 'action' in request.form:
|
122
|
action = request.form['action']
|
123
|
if 'query' in request.form:
|
124
|
query = request.form['query']
|
125
|
if 'only_result' in request.form:
|
126
|
only_result = True
|
127
|
|
128
|
if 'browse_field' in request.form:
|
129
|
try:
|
130
|
filter_query = json.loads(request.form['browse_field'])
|
131
|
except:
|
132
|
app.logger.error("Unable to parse browse_field, current value %s" % request.form['browse_field'])
|
133
|
|
134
|
if 'start' in request.form:
|
135
|
start = int(request.form['start'])
|
136
|
|
137
|
if action == 'query' and query is not None:
|
138
|
q = QueryResolver(host, base_dnet_url)
|
139
|
try:
|
140
|
result = q.query(q=query, browse_fields=["entitytype", "provenance"], start=start, only_result=only_result,
|
141
|
filter_query=filter_query)
|
142
|
except Exception as e:
|
143
|
print e
|
144
|
return jsonify(result=result)
|
145
|
else:
|
146
|
return jsonify(error={'error_code': 'an error occur during query on server'})
|
147
|
|
148
|
|
149
|
if __name__ == '__main__':
|
150
|
app.logger.addHandler(dlilogger)
|
151
|
app.run(debug=False, host='0.0.0.0', processes=5)
|