Binary sensor platform for the Balboa Spa (#60409)

homeassistant/components/balboa/binary_sensor.py

@@ -0,0 +1,61 @@
+"""Support for Balboa Spa binary sensors."""
+from homeassistant.components.binary_sensor import (
+    DEVICE_CLASS_MOVING,
+    BinarySensorEntity,
+)
+
+from .const import CIRC_PUMP, DOMAIN, FILTER
+from .entity import BalboaEntity
+
+FILTER_STATES = [
+    [False, False],  # self.FILTER_OFF
+    [True, False],  # self.FILTER_1
+    [False, True],  # self.FILTER_2
+    [True, True],  # self.FILTER_1_2
+]
+
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    """Set up the spa's binary sensors."""
+    spa = hass.data[DOMAIN][entry.entry_id]
+    entities = [
+        BalboaSpaFilter(hass, entry, spa, FILTER, index) for index in range(1, 3)
+    ]
+    if spa.have_circ_pump():
+        entities.append(BalboaSpaCircPump(hass, entry, spa, CIRC_PUMP))
+
+    async_add_entities(entities)
+
+
+class BalboaSpaBinarySensor(BalboaEntity, BinarySensorEntity):
+    """Representation of a Balboa Spa binary sensor entity."""
+
+    _attr_device_class = DEVICE_CLASS_MOVING
+
+
+class BalboaSpaCircPump(BalboaSpaBinarySensor):
+    """Representation of a Balboa Spa circulation pump."""
+
+    @property
+    def is_on(self) -> bool:
+        """Return true if the filter is on."""
+        return self._client.get_circ_pump()
+
+    @property
+    def icon(self):
+        """Return the icon to use in the frontend."""
+        return "mdi:water-pump" if self.is_on else "mdi:water-pump-off"
+
+
+class BalboaSpaFilter(BalboaSpaBinarySensor):
+    """Representation of a Balboa Spa Filter."""
+
+    @property
+    def is_on(self) -> bool:
+        """Return true if the filter is on."""
+        return FILTER_STATES[self._client.get_filtermode()][self._num - 1]
+
+    @property
+    def icon(self):
+        """Return the icon to use in the frontend."""
+        return "mdi:sync" if self.is_on else "mdi:sync-off"