Upload files to "custom_components/aguaiot"

custom_components/aguaiot/switch.py

@@ -0,0 +1,86 @@
+import logging
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+)
+from homeassistant.components.switch import SwitchEntity
+from homeassistant.helpers.entity import DeviceInfo
+from .const import SWITCHES, DOMAIN
+from .aguaiot import AguaIOTError
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    coordinator = config_entry.runtime_data
+    agua = coordinator.agua
+
+    switches = []
+    for device in agua.devices:
+        for switch in SWITCHES:
+            if switch.key in device.registers and device.get_register_enabled(
+                switch.key
+            ):
+                switches.append(AguaIOTHeatingSwitch(coordinator, device, switch))
+
+    async_add_entities(switches, True)
+
+
+class AguaIOTHeatingSwitch(CoordinatorEntity, SwitchEntity):
+    """Switch entity"""
+
+    _attr_has_entity_name = True
+
+    def __init__(self, coordinator, device, description):
+        """Initialize the thermostat."""
+        super().__init__(coordinator)
+        self._device = device
+        self.entity_description = description
+
+    @property
+    def unique_id(self):
+        """Return a unique ID."""
+        return f"{self._device.id_device}_{self.entity_description.key}"
+
+    @property
+    def name(self):
+        """Return the name of the device, if any."""
+        return f"{self._device.name} {self.entity_description.name}"
+
+    @property
+    def device_info(self):
+        """Return the device info."""
+        return DeviceInfo(
+            identifiers={(DOMAIN, self._device.id_device)},
+            name=self._device.name,
+            manufacturer="Micronova",
+            model=self._device.name_product,
+        )
+
+    @property
+    def is_on(self):
+        """Return the state of the sensor."""
+        return bool(self._device.get_register_value(self.entity_description.key))
+
+    async def async_turn_off(self):
+        """Turn device off."""
+        try:
+            await self._device.set_register_value(self.entity_description.key, 0)
+            await self.coordinator.async_request_refresh()
+        except AguaIOTError as err:
+            _LOGGER.error(
+                "Failed to turn off '%s', error: %s",
+                f"{self._device.name} {self.entity_description.name}",
+                err,
+            )
+
+    async def async_turn_on(self):
+        """Turn device on."""
+        try:
+            await self._device.set_register_value(self.entity_description.key, 1)
+            await self.coordinator.async_request_refresh()
+        except AguaIOTError as err:
+            _LOGGER.error(
+                "Failed to turn on '%s', error: %s",
+                f"{self._device.name} {self.entity_description.name}",
+                err,
+            )