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
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
|
#!/usr/bin/env python3
import asyncio
import json
import os
import __py_include
from datetime import datetime, timedelta
from aiohttp import web
from homekit import http
from homekit.config import config, is_development_mode
from homekit.database import BotsDatabase, SensorsDatabase, InverterDatabase
from homekit.database.inverter_time_formats import *
from homekit.api.types import TemperatureSensorLocation, SoundSensorLocation
from homekit.media import SoundRecordStorage
def strptime_auto(s: str) -> datetime:
e = None
for fmt in (FormatTime, FormatDate):
try:
return datetime.strptime(s, fmt)
except ValueError as _e:
e = _e
raise e
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/openwrt/', self.POST_openwrt_log)
self.get('/inverter/consumed_energy/', self.GET_consumed_energy)
self.get('/inverter/grid_consumed_energy/', self.GET_grid_consumed_energy)
self.get('/recordings/list/', self.GET_recordings_list)
@staticmethod
@web.middleware
async def validate_auth(req: web.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: web.Request):
message = "nothing here, keep lurking"
if is_development_mode():
message += ' (dev mode)'
return web.Response(text=message, content_type='text/plain')
async def GET_sensors_data(self, req: web.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: web.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: web.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_openwrt_log(self, req: web.Request):
data = await req.post()
try:
logs = data['logs']
ap = int(data['ap'])
except KeyError:
logs = ''
ap = 0
# 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, ap)
return self.ok()
async def GET_recordings_list(self, req: web.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)
@staticmethod
def _get_inverter_from_to(req: web.Request):
s_from = req.query['from']
s_to = req.query['to']
dt_from = strptime_auto(s_from)
if s_to == 'now':
dt_to = datetime.now()
else:
dt_to = strptime_auto(s_to)
return dt_from, dt_to
async def GET_consumed_energy(self, req: web.Request):
dt_from, dt_to = self._get_inverter_from_to(req)
wh = InverterDatabase().get_consumed_energy(dt_from, dt_to)
return self.ok(wh)
async def GET_grid_consumed_energy(self, req: web.Request):
dt_from, dt_to = self._get_inverter_from_to(req)
wh = InverterDatabase().get_grid_consumed_energy(dt_from, dt_to)
return self.ok(wh)
# start of the program
# --------------------
if __name__ == '__main__':
_app_name = 'web_api'
if is_development_mode():
_app_name += '_dev'
config.load_app(_app_name)
loop = asyncio.get_event_loop()
server = WebAPIServer(config.get_addr('server.listen'))
server.run()
|
