Add house_name to device import filters #522

AlexxIT · Oct 13, 2024 · 05012e2 · 05012e2
1 parent bc59ca2
commit 05012e2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/custom_components/yandex_station/core/yandex_quasar.py b/custom_components/yandex_station/core/yandex_quasar.py
@@ -159,7 +159,9 @@ async def init(self):
         for house in resp["households"]:
             if "sharing_info" in house:
                 continue
-            self.devices += house["all"]
+            self.devices.extend(
+                {**device, "house_name": house["name"]} for device in house["all"]
+            )
 
         await self.load_scenarios()
 

diff --git a/custom_components/yandex_station/hass/hass_utils.py b/custom_components/yandex_station/hass/hass_utils.py
@@ -15,7 +15,7 @@
 from ..vacuum import INCLUDE_TYPES as VACUUM
 from ..water_heater import INCLUDE_TYPES as WATER_HEATER
 
-INCLUDE_KEYS = ("id", "name", "type", "room_name", "skill_id")
+INCLUDE_KEYS = ("id", "name", "type", "room_name", "skill_id", "house_name")
 
 INCLUDE_TYPES_UNKNOWN = (
     "devices.types.camera",