first commit

2024-01-11 15:10:51 +03:00
commit eddf07c652
602 changed files with 67154 additions and 0 deletions
@@ -0,0 +1,55 @@
+from homeassistant.components.number import NumberEntity
+from homeassistant.const import MAJOR_VERSION, MINOR_VERSION
+from homeassistant.core import callback
+
+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 = XiaomiNumber(gateway, device, conv)
+        async_add_entities([entity])
+
+    gw: XGateway = hass.data[DOMAIN][config_entry.entry_id]
+    gw.add_setup(__name__, setup)
+
+
+# noinspection PyAbstractClass
+class XiaomiNumber(XEntity, NumberEntity):
+    _attr_value: float = None
+
+    def __init__(self, gateway: 'XGateway', device: XDevice, conv: Converter):
+        super().__init__(gateway, device, conv)
+
+        if hasattr(conv, "min"):
+            self._attr_min_value = conv.min
+        if hasattr(conv, "max"):
+            self._attr_max_value = conv.max
+
+    @callback
+    def async_set_state(self, data: dict):
+        if self.attr in data:
+            self._attr_value = data[self.attr]
+
+    @callback
+    def async_restore_last_state(self, state: float, attrs: dict):
+        self._attr_value = state
+
+    async def async_update(self):
+        await self.device_read(self.subscribed_attrs)
+
+    # backward compatibility fix
+    if (MAJOR_VERSION, MINOR_VERSION) >= (2022, 8):
+        async def async_set_native_value(self, value: float) -> None:
+            await self.device_send({self.attr: value})
+    else:
+        async def async_set_value(self, value: float) -> None:
+            await self.device_send({self.attr: value})