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
|
#!/usr/bin/env python3
import paho.mqtt.client as mqtt
import re
from home.mqtt import MQTTBase
from home.config import config
from home.mqtt.payload.sensors import Temperature
from home.api.types import TemperatureSensorLocation
from home.database import SensorsDatabase
def get_sensor_type(sensor: str) -> TemperatureSensorLocation:
for item in TemperatureSensorLocation:
if sensor == item.name.lower():
return item
raise ValueError(f'unexpected sensor value: {sensor}')
class MQTTServer(MQTTBase):
def __init__(self):
super().__init__(clean_session=False)
self.database = SensorsDatabase()
def on_connect(self, client: mqtt.Client, userdata, flags, rc):
super().on_connect(client, userdata, flags, rc)
self._logger.info("subscribing to home/#")
client.subscribe('home/#', qos=1)
def on_message(self, client: mqtt.Client, userdata, msg):
try:
variants = '|'.join([s.name.lower() for s in TemperatureSensorLocation])
match = re.match(rf'(?:home|hk)/(\d+)/si7021/({variants})', msg.topic)
if not match:
return
# FIXME string home_id must be supported
home_id = int(match.group(1))
sensor = get_sensor_type(match.group(2))
payload = Temperature.unpack(msg.payload)
self.database.add_temperature(home_id, payload.time, sensor,
temp=int(payload.temp*100),
rh=int(payload.rh*100))
except Exception as e:
self._logger.exception(str(e))
if __name__ == '__main__':
config.load('sensors_mqtt_receiver')
server = MQTTServer()
server.connect_and_loop()
|
