python

custom_components/yandex_station/text.py

@@ -0,0 +1,32 @@
+import re
+
+from homeassistant.components.text import TextEntity
+
+from .core.entity import YandexEntity
+from .hass import hass_utils
+
+INCLUDE_TYPES = (
+    "devices.types.smart_speaker.yandex.station.orion",
+)
+
+
+async def async_setup_entry(hass, entry, async_add_entities):
+    async_add_entities(
+        YandexText(quasar, device, config)
+        for quasar, device, config in hass_utils.incluce_devices(hass, entry)
+        if device["type"] in INCLUDE_TYPES
+    )
+
+
+class YandexText(TextEntity, YandexEntity):
+    def internal_init(self, capabilities: dict, properties: dict):
+        pass
+
+    def internal_update(self, capabilities: dict, properties: dict):
+        if "led_array" in capabilities:
+            array: list[int] = capabilities["led_array"]
+            self._attr_native_value = ",".join(str(i) for i in array)
+
+    async def async_set_value(self, value: str) -> None:
+        array: list[int] = [int(i) for i in re.findall(r"\d+", value)]
+        await self.device_action("led_array", array)