first commit

2024-01-11 15:10:51 +03:00
commit eddf07c652
602 changed files with 67154 additions and 0 deletions
@@ -0,0 +1,62 @@
+from homeassistant.components.cover import CoverEntity, ATTR_POSITION, \
+    ATTR_CURRENT_POSITION
+from homeassistant.const import STATE_CLOSING, STATE_OPENING
+from homeassistant.core import callback
+from homeassistant.helpers.restore_state import RestoreEntity
+
+from . import DOMAIN
+from .core.converters import Converter
+from .core.device import XDevice
+from .core.entity import XEntity
+from .core.gateway import XGateway
+
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    def setup(gateway: XGateway, device: XDevice, conv: Converter):
+        if conv.attr in device.entities:
+            entity: XEntity = device.entities[conv.attr]
+            entity.gw = gateway
+        else:
+            entity = XiaomiCover(gateway, device, conv)
+        async_add_entities([entity])
+
+    gw: XGateway = hass.data[DOMAIN][config_entry.entry_id]
+    gw.add_setup(__name__, setup)
+
+
+# noinspection PyAbstractClass
+class XiaomiCover(XEntity, CoverEntity, RestoreEntity):
+    _attr_current_cover_position = 0
+    _attr_is_closed = None
+
+    @callback
+    def async_set_state(self, data: dict):
+        if 'run_state' in data:
+            self._attr_state = data["run_state"]
+            self._attr_is_opening = self._attr_state == STATE_OPENING
+            self._attr_is_closing = self._attr_state == STATE_CLOSING
+        if 'position' in data:
+            self._attr_current_cover_position = data['position']
+            # https://github.com/AlexxIT/XiaomiGateway3/issues/771
+            self._attr_is_closed = self._attr_current_cover_position <= 2
+
+    @callback
+    def async_restore_last_state(self, state: str, attrs: dict):
+        if not state:
+            return
+        self.async_set_state({
+            "run_state": state,
+            "position": attrs[ATTR_CURRENT_POSITION]
+        })
+
+    async def async_open_cover(self, **kwargs):
+        await self.device_send({self.attr: "open"})
+
+    async def async_close_cover(self, **kwargs):
+        await self.device_send({self.attr: "close"})
+
+    async def async_stop_cover(self, **kwargs):
+        await self.device_send({self.attr: "stop"})
+
+    async def async_set_cover_position(self, **kwargs):
+        await self.device_send({"position": kwargs[ATTR_POSITION]})