108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
from builtins import staticmethod
|
|
|
|
from interruptingcow import timeout
|
|
from mqtt import MqttMessage
|
|
from workers.base import BaseWorker
|
|
|
|
REQUIREMENTS = ['python-eq3bt']
|
|
|
|
monitoredAttrs = ["low_battery", "valve_state", "target_temperature", "window_open", "locked"]
|
|
|
|
STATE_AWAY = 'away'
|
|
STATE_MANUAL = 'manual'
|
|
STATE_ON = 'on'
|
|
STATE_OFF = 'off'
|
|
|
|
class ThermostatWorker(BaseWorker):
|
|
class ModesMapper():
|
|
def __init__(self):
|
|
from eq3bt import Mode
|
|
|
|
self._mapped_modes = {
|
|
Mode.Closed: STATE_OFF,
|
|
Mode.Open: STATE_ON,
|
|
Mode.Auto: 'auto',
|
|
Mode.Manual: STATE_MANUAL,
|
|
Mode.Away: STATE_AWAY,
|
|
Mode.Boost: 'heat',
|
|
}
|
|
|
|
self._reverse_modes = {v: k for k, v in self._mapped_modes.items()}
|
|
|
|
def get_mapping(self, mode):
|
|
if mode < 0:
|
|
return None
|
|
return self._mapped_modes[mode]
|
|
|
|
def get_reverse_mapping(self, mode):
|
|
return self._reverse_modes[mode]
|
|
|
|
@staticmethod
|
|
def away_mode_on_off(mode):
|
|
if mode == STATE_AWAY:
|
|
return STATE_ON
|
|
else:
|
|
return STATE_OFF
|
|
|
|
@staticmethod
|
|
def on_off_to_mode(on_off):
|
|
if on_off == STATE_ON:
|
|
return STATE_AWAY
|
|
else:
|
|
return 'auto'
|
|
|
|
|
|
def _setup(self):
|
|
from eq3bt import Thermostat
|
|
|
|
for name, mac in self.devices.items():
|
|
self.devices[name] = Thermostat(mac)
|
|
|
|
self._modes_mapper = self.ModesMapper()
|
|
|
|
def status_update(self):
|
|
ret = []
|
|
for name, thermostat in self.devices.items():
|
|
try:
|
|
ret += self.update_device_state(name, thermostat)
|
|
except RuntimeError:
|
|
pass
|
|
|
|
return ret
|
|
|
|
def on_command(self, topic, value):
|
|
_, device_name, method, _ = topic.split('/')
|
|
thermostat = self.devices[device_name]
|
|
|
|
value = value.decode('utf-8')
|
|
|
|
if method == STATE_AWAY:
|
|
method = "mode"
|
|
value = self.ModesMapper.on_off_to_mode(value)
|
|
|
|
# It needs to be on separate if because first if can change method
|
|
if method == "mode":
|
|
value = self._modes_mapper.get_reverse_mapping(value)
|
|
elif method == "target_temperature":
|
|
value = float(value)
|
|
|
|
setattr(thermostat, method, value)
|
|
return self.update_device_state(device_name, thermostat)
|
|
|
|
@timeout(8.0)
|
|
def update_device_state(self, name, thermostat):
|
|
thermostat.update()
|
|
|
|
ret = []
|
|
for attr in monitoredAttrs:
|
|
ret.append(MqttMessage(topic=self.format_topic(name, attr), payload=getattr(thermostat, attr)))
|
|
|
|
ha_mode = self._modes_mapper.get_mapping(thermostat.mode)
|
|
ret.append(MqttMessage(topic=self.format_topic(name, 'mode'), payload=ha_mode))
|
|
ret.append(MqttMessage(topic=self.format_topic(name, STATE_AWAY), payload=self.ModesMapper.away_mode_on_off(ha_mode)))
|
|
|
|
return ret
|
|
|
|
def device_for(self, mac):
|
|
return eq3bt.Thermostat(mac)
|