first commit

2024-01-11 15:10:51 +03:00
commit eddf07c652
602 changed files with 67154 additions and 0 deletions
@@ -0,0 +1,116 @@
+"""Support for Yandex Smart Home sensor."""
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+from . import CONF_INCLUDE
+from . import DATA_CONFIG
+from . import DOMAIN
+from . import YandexQuasar
+from homeassistant.components.sensor import SensorEntity
+from homeassistant.components.sensor import SensorEntityDescription
+from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT
+from homeassistant.const import PERCENTAGE
+from homeassistant.const import TEMP_CELSIUS
+
+_LOGGER = logging.getLogger(__name__)
+
+DEVICES = ["devices.types.humidifier"]
+
+SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
+    SensorEntityDescription(
+        key="temperature",
+        name="Temperature",
+        native_unit_of_measurement=TEMP_CELSIUS,
+        state_class=STATE_CLASS_MEASUREMENT,
+    ),
+    SensorEntityDescription(
+        key="humidity",
+        name="Humidity",
+        native_unit_of_measurement=PERCENTAGE,
+        state_class=STATE_CLASS_MEASUREMENT,
+    ),
+)
+
+SENSOR_KEYS: list[str] = [desc.key for desc in SENSOR_TYPES]
+
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    """Set up sensor from a config entry."""
+    include = hass.data[DOMAIN][DATA_CONFIG][CONF_INCLUDE]
+    quasar = hass.data[DOMAIN][entry.unique_id]
+
+    devices = []
+    for device in quasar.devices:
+        if device["name"] in include and device["type"] in DEVICES:
+            data = await quasar.get_device(device["id"])
+            for prop in data["properties"]:
+                for description in SENSOR_TYPES:
+                    if prop["parameters"]["instance"] == description.key:
+                        devices.append(
+                            YandexSensor(
+                                quasar,
+                                device,
+                                prop["parameters"]["name"],
+                                description,
+                            )
+                        )
+
+    async_add_entities(devices, True)
+
+
+# noinspection PyAbstractClass
+class YandexSensor(SensorEntity):
+    """Yandex Home sensor entity"""
+
+    _humidity = None
+    _temperature = None
+
+    def __init__(
+        self,
+        quasar: YandexQuasar,
+        device: dict,
+        name: str,
+        description: SensorEntityDescription,
+    ) -> None:
+        """Initialize entity."""
+        self.quasar = quasar
+        self.device = device
+        self.sensor_name = name
+        self.entity_description = description
+
+    @property
+    def unique_id(self):
+        """Return entity unique id."""
+        return f"{self.device['id'].replace('-', '')}: {self.entity_description.name}"
+
+    @property
+    def name(self):
+        """Return entity name."""
+        return f"{self.device['name']}: {self.sensor_name}"
+
+    @property
+    def humidity(self) -> int:
+        """Return current humidity."""
+        return self._humidity
+
+    @property
+    def temperature(self) -> int:
+        """Return current temperature."""
+        return self._temperature
+
+    async def async_update(self):
+        """Update the entity."""
+        data = await self.quasar.get_device(self.device["id"])
+        for prop in data["properties"]:
+            instance = prop["parameters"]["instance"]
+            if instance == "humidity":
+                self._humidity = prop["state"]["value"]
+            if instance == "temperature":
+                self._temperature = prop["state"]["value"]
+
+    @property
+    def native_value(self) -> Any:
+        """Return the native value of the sensor."""
+        return getattr(self, self.entity_description.key)