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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
|
#!/usr/bin/env python3
import asyncio
import json
import os
from datetime import datetime, timedelta
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from aiohttp import web
from home import http
from home.util import parse_addr
from home.config import config, is_development_mode
from home.database import BotsDatabase, SensorsDatabase, InverterDatabase
from home.api.types import BotType, TemperatureSensorLocation, SoundSensorLocation
from home.media import SoundRecordStorage
class AuthError(Exception):
def __init__(self, message: str):
super().__init__()
self.message = message
class WebAPIServer(http.HTTPServer):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.app.middlewares.append(self.validate_auth)
self.get('/', self.get_index)
self.get('/sensors/data/', self.GET_sensors_data)
self.get('/sound-sensors/hits/', self.GET_sound_sensors_hits)
self.post('/sound-sensors/hits/', self.POST_sound_sensors_hits)
self.post('/log/bot-request/', self.POST_bot_request_log)
self.post('/log/openwrt/', self.POST_openwrt_log)
self.get('/recordings/list/', self.GET_recordings_list)
@staticmethod
@web.middleware
async def validate_auth(req: http.Request, handler):
def get_token() -> str:
name = 'X-Token'
if name in req.headers:
return req.headers[name]
return req.query['token']
try:
token = get_token()
except KeyError:
raise AuthError('no token')
if token != config['api']['token']:
raise AuthError('invalid token')
return await handler(req)
@staticmethod
async def get_index(req: http.Request):
message = "nothing here, keep lurking"
if is_development_mode():
message += ' (dev mode)'
return http.Response(text=message, content_type='text/plain')
async def GET_sensors_data(self, req: http.Request):
try:
hours = int(req.query['hours'])
if hours < 1 or hours > 24:
raise ValueError('invalid hours value')
except KeyError:
hours = 1
sensor = TemperatureSensorLocation(int(req.query['sensor']))
dt_to = datetime.now()
dt_from = dt_to - timedelta(hours=hours)
db = SensorsDatabase()
data = db.get_temperature_recordings(sensor, (dt_from, dt_to))
return self.ok(data)
async def GET_sound_sensors_hits(self, req: http.Request):
location = SoundSensorLocation(int(req.query['location']))
after = int(req.query['after'])
kwargs = {}
if after is None:
last = int(req.query['last'])
if last is None:
raise ValueError('you must pass `after` or `last` params')
else:
if not 0 < last < 100:
raise ValueError('invalid last value: must be between 0 and 100')
kwargs['last'] = last
else:
kwargs['after'] = datetime.fromtimestamp(after)
data = BotsDatabase().get_sound_hits(location, **kwargs)
return self.ok(data)
async def POST_sound_sensors_hits(self, req: http.Request):
hits = []
data = await req.post()
for hit, count in json.loads(data['hits']):
if not hasattr(SoundSensorLocation, hit.upper()):
raise ValueError('invalid sensor location')
if count < 1:
raise ValueError(f'invalid count: {count}')
hits.append((SoundSensorLocation[hit.upper()], count))
BotsDatabase().add_sound_hits(hits, datetime.now())
return self.ok()
async def POST_bot_request_log(self, req: http.Request):
data = await req.post()
try:
user_id = int(data['user_id'])
except KeyError:
user_id = 0
try:
message = data['message']
except KeyError:
message = ''
bot = BotType(int(data['bot']))
# validate message
if message.strip() == '':
raise ValueError('message can\'t be empty')
# add record to the database
BotsDatabase().add_request(bot, user_id, message)
return self.ok()
async def POST_openwrt_log(self, req: http.Request):
data = await req.post()
try:
logs = data['logs']
except KeyError:
logs = ''
# validate it
logs = json.loads(logs)
assert type(logs) is list, "invalid json data (list expected)"
lines = []
for line in logs:
assert type(line) is list, "invalid line type (list expected)"
assert len(line) == 2, f"expected 2 items in line, got {len(line)}"
assert type(line[0]) is int, "invalid line[0] type (int expected)"
assert type(line[1]) is str, "invalid line[1] type (str expected)"
lines.append((
datetime.fromtimestamp(line[0]),
line[1]
))
BotsDatabase().add_openwrt_logs(lines)
return self.ok()
async def GET_recordings_list(self, req: http.Request):
data = await req.post()
try:
extended = bool(int(data['extended']))
except KeyError:
extended = False
node = data['node']
root = os.path.join(config['recordings']['directory'], node)
if not os.path.isdir(root):
raise ValueError(f'invalid node {node}: no such directory')
storage = SoundRecordStorage(root)
files = storage.getfiles(as_objects=extended)
if extended:
files = list(map(lambda file: file.__dict__(), files))
return self.ok(files)
# start of the program
# --------------------
if __name__ == '__main__':
config.load('web_api')
loop = asyncio.get_event_loop()
server = WebAPIServer(parse_addr(config['server']['listen']))
server.run()
|
