1 |
55154
|
sandro.lab
|
from flask import Flask
|
2 |
|
|
from flask import jsonify, request
|
3 |
|
|
from eu.dnetlib.exporter.ScrollResult import ScrollResult
|
4 |
|
|
|
5 |
|
|
app = Flask(__name__)
|
6 |
|
|
|
7 |
|
|
scroll_map = {}
|
8 |
|
|
|
9 |
|
|
|
10 |
|
|
@app.route("/new_scan", methods=['get'])
|
11 |
|
|
def new_scan():
|
12 |
|
|
timestamp = request.args.get('timestamp')
|
13 |
|
|
print(timestamp)
|
14 |
|
|
active = request.args.get('active')
|
15 |
|
|
if active is not None and 'false' == active.lower():
|
16 |
|
|
active = False
|
17 |
|
|
else:
|
18 |
|
|
active = True
|
19 |
|
|
s = ScrollResult()
|
20 |
|
|
s.generate_scroll(timestamp=timestamp, only_active=active)
|
21 |
|
|
scroll_map[str(s.id)] = s
|
22 |
|
|
return jsonify(s.next_page())
|
23 |
|
|
|
24 |
|
|
|
25 |
|
|
@app.route("/scan/<s_id>")
|
26 |
|
|
def continue_scan(s_id):
|
27 |
|
|
if s_id in scroll_map:
|
28 |
|
|
try:
|
29 |
|
|
return jsonify(scroll_map[s_id].next_page())
|
30 |
|
|
except Exception as e:
|
31 |
|
|
print(e)
|
32 |
|
|
del(scroll_map[s_id])
|
33 |
|
|
return jsonify([])
|
34 |
|
|
return jsonify([])
|
35 |
|
|
|
36 |
|
|
|
37 |
|
|
if __name__ == '__main__':
|
38 |
|
|
print("Starting flask")
|
39 |
55215
|
sandro.lab
|
app.run(host='0.0.0.0', debug=False)
|