home_emulation/devices/shelly.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
""" Communication (MQTT)
    shelly
        +- relay
        |      +- 0 ["on" / "off"]              <- status 
        |      |  +- command ["on"/ "off"]      <- command
        |      |  +- energy [numeric]           <- status
        |      +- 1 ["on" / "off"]              <- status
        |         +- command ["on"/ "off"]      <- command
        |         +- energy [numeric]           <- status
        +- input
        |      +- 0 [0 / 1]                     <- status
        |      +- 1 [0 / 1]                     <- status
        +- input_event
        |      +- 0                             <- status
        |      +- 1                             <- status
        +- logpush
        |      +- 0 [0 / 1]                     <- status
        |      +- 1 [0 / 1]                     <- status
        +- temperature [numeric] °C             <- status
        +- temperature_f [numeric] F            <- status
        +- overtemperature [0 / 1]              <- status
        +- id                                   <- status
        +- model                                <- status
        +- mac                                  <- status
        +- ip                                   <- status
        +- new_fw                               <- status
        +- fw_ver                               <- status
"""

from devices.base import base
import json


class sw_plain(base):
    """A shelly device with switching functionality

    Args:
        mqtt_client (mqtt.mqtt_client): A MQTT Client instance
        topic (str): the base topic for this device
        kwargs (**dict): cd_r0=list of devices connected to relay/0
    """
    PROPERTIES = [
        "relay/0",
    ]
    def __init__(self, mqtt_client, topic, **kwargs):
        super().__init__(mqtt_client, topic, **kwargs)
        if getattr(self, 'cd_r0', None) is None:
            self.cd_r0 = []
        self["state"] = "off"
        #
        self.mqtt_client.add_callback(self.topic + '/relay/0/command', self.__rx_set__)

    def __rx_set__(self, client, userdata, message):
        data = message.payload.decode('utf-8')
        key = message.topic.split('/')[-3] + '/' + message.topic.split('/')[-2]
        self.logger.info("Received set data for %s: %s", key, repr(data))
        self.__set__(key, data)
        self.send_device_status(key)
        if key == "relay/0":
            for d in self.cd_r0:
                d.set_state(data.lower() == "on")

    def send_device_status(self, key):
        data = self[key]
        self.logger.info("Sending status for %s: %s", key, repr(data))
        self.mqtt_client.send(self.topic + '/' + key, data)