forked from ito-org/backend-sqlite
/
app.py
56 lines (44 loc) · 1.43 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#!/usr/bin/env python3
from flask import Flask, Response, request, g
import binascii
import db
app = Flask(__name__)
DATABASE = 'temp.db'
def get_db_connection():
con = getattr(g, '_con', None)
if con is None:
con = g._database = db.prepare_db(DATABASE)
return con
@app.teardown_appcontext
def close_connection(exception):
con = getattr(g, '_con', None)
if con is not None:
con.close()
@app.route('/post_uuids', methods=["POST"])
def post_uuids():
UUID_LENGTH = 16
print(request.content_type)
assert request.content_type == 'application/octet-stream'
data = request.get_data()
assert len(data) % UUID_LENGTH == 0
uuids = (data[i:i + UUID_LENGTH] for i in range(0, len(data), UUID_LENGTH))
db.insert_into_db(get_db_connection(), uuids)
return ""
@app.route('/get_uuids', methods=["GET"])
def get_uuids():
# TODO handle invalid uuid length
request_uuid = request.args.get('uuid', '')
request_uuid = binascii.unhexlify(request_uuid)
data = db.query_subsequent(get_db_connection(), request_uuid)
if request.accept_mimetypes['application/json']:
def generator():
yield "["
first_uuid = next(data)
if first_uuid:
yield '"{}"'.format(first_uuid.hex())
for uuid in data:
yield ', "{}"'.format(uuid.hex())
yield "]"
return Response(generator(), mimetype="application/json")
if request.accept_mimetypes['application/octet-stream']:
return Response((uuid for uuid in data), mimetype="application/octet-stream")