-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathserve.py
89 lines (75 loc) · 2.6 KB
/
serve.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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import sqlite3
from flask import g
from flask import Flask
from flask import request
from flask_cors import CORS
import json
app = Flask(__name__)
CORS(app)
DATABASE = 'packets.db'
DB_QUERY_START = """SELECT packet_from, packet_to, decoded, id, rx_time, rx_snr,
hop_limit, rx_rssi, channel, want_ack, priority, delayed, port, rowid FROM
packets """
def get_db():
db = getattr(g, '_database', None)
if db is None:
db = g._database = sqlite3.connect(DATABASE)
db.row_factory = sqlite3.Row
return db
def query_db(query, args=(), one=False):
cur = get_db().execute(query, args)
rv = cur.fetchall()
cur.close()
return (rv[0] if rv else None) if one else rv
def row_to_dict(row):
return {
"from": row['packet_from'],
"to": row['packet_to'],
"decoded": row['decoded'],
"id": row['id'],
"rxTime": row['rx_time'],
"rxSnr": row['rx_snr'],
"hopLimit": row['hop_limit'],
"rxRssi": row['rx_rssi'],
"channel": row['channel'],
"wantAck": row['want_ack'],
"priority": row['priority'],
"delayed": row['delayed'],
"port": row['port'],
"rowId": row['rowid'],
}
@app.teardown_appcontext
def close_connection(exception):
db = getattr(g, '_database', None)
if db is not None:
db.close()
@app.route("/packets/after/<int:rowid>")
def get_packets_after(rowid):
"""Returns a JSON reprsentation of ALL the packets AFTER a certain rowid"""
response = []
for row in query_db(DB_QUERY_START + 'WHERE rowid > ?;', (rowid,)):
response.append(row_to_dict(row))
return response
@app.route("/packets/one-after/<int:rowid>")
def get_one_packet_after(rowid):
"""Returns a JSON reprsentation of THE NEXT packet AFTER a certain rowid"""
response = []
for row in query_db(DB_QUERY_START + 'WHERE rowid > ? LIMIT 1;', (rowid,)):
response.append(row_to_dict(row))
return response
@app.route("/packets/range/<int:start>-<int:end>")
def get_range(start, end):
"""Returns a JSON reprsentation of the packets AFTER one rxTime but BEFORE
another"""
response = []
for row in query_db(DB_QUERY_START + 'WHERE rx_time >= ? AND rx_time <= ?;',
(start, end)):
response.append(row_to_dict(row))
return response
@app.route("/packets")
def get_packets():
"""Returns a JSON representation of ALL of the packets in the DB"""
response = []
for row in query_db(DB_QUERY_START + ';'):
response.append(row_to_dict(row))
return response