170 lines
4.9 KiB
Python
170 lines
4.9 KiB
Python
"""Allows to configure a switch using GPIO."""
|
|
import json
|
|
import logging
|
|
from typing import Callable
|
|
|
|
# pylint: disable=R0801
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_NAME, STATE_ON, STATE_OFF
|
|
from homeassistant.helpers.entity import EntityCategory
|
|
from homeassistant.core import HomeAssistant, callback
|
|
import homeassistant.helpers.config_validation as cv
|
|
import voluptuous as vol
|
|
|
|
from .common import HASPToggleEntity
|
|
from .const import CONF_HWID, CONF_RELAYS, CONF_TOPIC
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
HASP_RELAY_SCHEMA = vol.Schema(
|
|
{
|
|
vol.Required("state"): cv.boolean,
|
|
}
|
|
)
|
|
|
|
|
|
# pylint: disable=R0801, W0613
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: Callable
|
|
):
|
|
"""Set up Plate Relays as switch based on a config entry."""
|
|
|
|
async_add_entities(
|
|
[
|
|
HASPSwitch(
|
|
entry.data[CONF_NAME],
|
|
entry.data[CONF_HWID],
|
|
entry.data[CONF_TOPIC],
|
|
gpio,
|
|
)
|
|
for gpio in entry.data[CONF_RELAYS]
|
|
]
|
|
+ [
|
|
HASPAntiBurn(
|
|
entry.data[CONF_NAME],
|
|
entry.data[CONF_HWID],
|
|
entry.data[CONF_TOPIC],
|
|
)
|
|
]
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
class HASPSwitch(HASPToggleEntity):
|
|
"""Representation of an openHASP relay."""
|
|
|
|
def __init__(self, name, hwid, topic, gpio):
|
|
"""Initialize the relay."""
|
|
super().__init__(name, hwid, topic, gpio)
|
|
self._attr_name = f"{name} switch {self._gpio}"
|
|
|
|
async def refresh(self):
|
|
"""Sync local state back to plate."""
|
|
if self._state is None:
|
|
# Don't do anything before we know the state
|
|
return
|
|
|
|
await self.hass.components.mqtt.async_publish(
|
|
self.hass,
|
|
f"{self._topic}/command/output{self._gpio}",
|
|
json.dumps(HASP_RELAY_SCHEMA({"state": int(self._state)})),
|
|
qos=0,
|
|
retain=False,
|
|
)
|
|
self.async_write_ha_state()
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Run when entity about to be added."""
|
|
await super().async_added_to_hass()
|
|
|
|
@callback
|
|
async def relay_state_message_received(msg):
|
|
"""Process State."""
|
|
|
|
try:
|
|
self._available = True
|
|
message = HASP_RELAY_SCHEMA(json.loads(msg.payload))
|
|
_LOGGER.debug("%s state = %s", self.name, message)
|
|
|
|
self._state = message["state"]
|
|
self.async_write_ha_state()
|
|
|
|
except vol.error.Invalid as err:
|
|
_LOGGER.error(err)
|
|
|
|
self._subscriptions.append(
|
|
await self.hass.components.mqtt.async_subscribe(
|
|
f"{self._topic}/state/output{self._gpio}", relay_state_message_received
|
|
)
|
|
)
|
|
|
|
await self.hass.components.mqtt.async_publish(
|
|
self.hass,
|
|
f"{self._topic}/command/output{self._gpio}",
|
|
"",
|
|
qos=0,
|
|
retain=False,
|
|
)
|
|
|
|
|
|
class HASPAntiBurn(HASPToggleEntity):
|
|
"""Configuration switch of an openHASP antiburn feature."""
|
|
|
|
_attr_entity_category = EntityCategory.CONFIG
|
|
_attr_icon = "mdi:progress-wrench"
|
|
|
|
def __init__(self, name, hwid, topic):
|
|
"""Initialize the protection."""
|
|
super().__init__(name, hwid, topic, None)
|
|
self._attr_name = f"{self._name} antiburn"
|
|
|
|
async def refresh(self):
|
|
"""Sync local state back to plate."""
|
|
await self.hass.components.mqtt.async_publish(
|
|
self.hass,
|
|
f"{self._topic}/command/antiburn",
|
|
int(self._state),
|
|
# json.dumps(HASP_RELAY_SCHEMA({"state": int(self._state)})),
|
|
qos=0,
|
|
retain=False,
|
|
)
|
|
self.async_write_ha_state()
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Run when entity about to be added."""
|
|
await super().async_added_to_hass()
|
|
|
|
@callback
|
|
async def antiburn_state_message_received(msg):
|
|
"""Process State."""
|
|
|
|
try:
|
|
self._available = True
|
|
message = HASP_RELAY_SCHEMA(json.loads(msg.payload))
|
|
_LOGGER.debug("%s state = %s", self.name, message)
|
|
|
|
self._state = message["state"]
|
|
self.async_write_ha_state()
|
|
|
|
except vol.error.Invalid as err:
|
|
_LOGGER.error(err)
|
|
|
|
self._subscriptions.append(
|
|
await self.hass.components.mqtt.async_subscribe(
|
|
f"{self._topic}/state/antiburn", antiburn_state_message_received
|
|
)
|
|
)
|
|
|
|
self._state = False
|
|
self._available = True
|
|
|
|
await self.hass.components.mqtt.async_publish(
|
|
self.hass,
|
|
f"{self._topic}/command/antiburn",
|
|
int(self._state),
|
|
qos=0,
|
|
retain=False,
|
|
)
|