python

custom_components/yandex_station/number.py

@@ -0,0 +1,45 @@
+import logging
+
+from homeassistant.components.number import NumberEntity
+from homeassistant.const import UnitOfTemperature
+
+from .core.entity import YandexCustomEntity, extract_instance
+from .hass import hass_utils
+
+_LOGGER = logging.getLogger(__name__)
+
+INCLUDE_CAPABILITIES = ("devices.capabilities.range",)
+
+UNITS = {"unit.temperature.celsius": UnitOfTemperature.CELSIUS}
+
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    entities = []
+
+    for quasar, device, config in hass_utils.incluce_devices(hass, entry):
+        if instances := config.get("capabilities"):
+            for instance in device["capabilities"]:
+                if instance["type"] not in INCLUDE_CAPABILITIES:
+                    continue
+                if extract_instance(instance) in instances:
+                    entities.append(YandexCustomNumber(quasar, device, instance))
+
+    async_add_entities(entities)
+
+
+# noinspection PyAbstractClass
+class YandexCustomNumber(NumberEntity, YandexCustomEntity):
+    def internal_init(self, capabilities: dict, properties: dict):
+        if item := capabilities.get(self.instance):
+            if range_ := item.get("range"):
+                self._attr_native_max_value = range_["max"]
+                self._attr_native_min_value = range_["min"]
+                self._attr_native_step = range_["precision"]
+            self._attr_native_unit_of_measurement = UNITS.get(item["unit"])
+
+    def internal_update(self, capabilities: dict, properties: dict):
+        if self.instance in capabilities:
+            self._attr_native_value = capabilities[self.instance]
+
+    async def async_set_native_value(self, value: float) -> None:
+        await self.device_action(self.instance, value)