Project

General

Profile

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)