Add BleBox integration (#32664)

Co-Authored-By: J. Nick Koston <nick@koston.org>

homeassistant/components/blebox/cover.py

@@ -0,0 +1,97 @@
+"""BleBox cover entity."""
+
+from homeassistant.components.cover import (
+    ATTR_POSITION,
+    STATE_CLOSED,
+    STATE_CLOSING,
+    STATE_OPENING,
+    SUPPORT_CLOSE,
+    SUPPORT_OPEN,
+    SUPPORT_SET_POSITION,
+    SUPPORT_STOP,
+    CoverEntity,
+)
+
+from . import BleBoxEntity, create_blebox_entities
+from .const import (
+    BLEBOX_TO_HASS_COVER_STATES,
+    BLEBOX_TO_HASS_DEVICE_CLASSES,
+    DOMAIN,
+    PRODUCT,
+)
+
+
+async def async_setup_entry(hass, config_entry, async_add):
+    """Set up a BleBox entry."""
+
+    product = hass.data[DOMAIN][config_entry.entry_id][PRODUCT]
+    create_blebox_entities(product, async_add, BleBoxCoverEntity, "covers")
+    return True
+
+
+class BleBoxCoverEntity(BleBoxEntity, CoverEntity):
+    """Representation of a BleBox cover feature."""
+
+    @property
+    def state(self):
+        """Return the equivalent HA cover state."""
+        return BLEBOX_TO_HASS_COVER_STATES[self._feature.state]
+
+    @property
+    def device_class(self):
+        """Return the device class."""
+        return BLEBOX_TO_HASS_DEVICE_CLASSES[self._feature.device_class]
+
+    @property
+    def supported_features(self):
+        """Return the supported cover features."""
+        position = SUPPORT_SET_POSITION if self._feature.is_slider else 0
+        stop = SUPPORT_STOP if self._feature.has_stop else 0
+
+        return position | stop | SUPPORT_OPEN | SUPPORT_CLOSE
+
+    @property
+    def current_cover_position(self):
+        """Return the current cover position."""
+        position = self._feature.current
+        if position == -1:  # possible for shutterBox
+            return None
+
+        return None if position is None else 100 - position
+
+    @property
+    def is_opening(self):
+        """Return whether cover is opening."""
+        return self._is_state(STATE_OPENING)
+
+    @property
+    def is_closing(self):
+        """Return whether cover is closing."""
+        return self._is_state(STATE_CLOSING)
+
+    @property
+    def is_closed(self):
+        """Return whether cover is closed."""
+        return self._is_state(STATE_CLOSED)
+
+    async def async_open_cover(self, **kwargs):
+        """Open the cover position."""
+        await self._feature.async_open()
+
+    async def async_close_cover(self, **kwargs):
+        """Close the cover position."""
+        await self._feature.async_close()
+
+    async def async_set_cover_position(self, **kwargs):
+        """Set the cover position."""
+
+        position = kwargs[ATTR_POSITION]
+        await self._feature.async_set_position(100 - position)
+
+    async def async_stop_cover(self, **kwargs):
+        """Stop the cover."""
+        await self._feature.async_stop()
+
+    def _is_state(self, state_name):
+        value = self.state
+        return None if value is None else value == state_name