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
import paho.mqtt.client as mqtt
import logging
import re
from home.mqtt import MQTTBase
from home.config import config
from home.mqtt.message import Temperature
from home.api.types import TemperatureSensorLocation
from home.database import SensorsDatabase
logger = logging.getLogger(__name__)
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)
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/(\d+)/si7021/({variants})', msg.topic)
if not match:
return
home_id = int(match.group(1))
sensor = get_sensor_type(match.group(2))
packer = Temperature()
client_time, temp, rh = packer.unpack(msg.payload)
self.database.add_temperature(home_id, client_time, sensor,
temp=int(temp*100),
rh=int(rh*100))
except Exception as e:
logger.exception(str(e))
if __name__ == '__main__':
config.load('sensors_mqtt_receiver')
server = MQTTServer()
server.connect_and_loop()
|
