call blutrv only for gw gen3

home-assistant-libs · Dec 17, 2024 · 13501dc · 13501dc
1 parent ed115f4
commit 13501dc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/aioshelly/const.py b/aioshelly/const.py
@@ -929,6 +929,8 @@ class ShellyDevice:
     MODEL_VINTAGE_V2,
 }
 
+BLU_TRV_IDENTIFIER = "blutrv"
+
 
 class UndefinedType(Enum):
     """Singleton type for use with not set sentinel values."""

diff --git a/aioshelly/rpc_device/device.py b/aioshelly/rpc_device/device.py
@@ -18,11 +18,13 @@
     process_ip_or_options,
 )
 from ..const import (
+    BLU_TRV_IDENTIFIER,
     CONNECT_ERRORS,
     DEVICE_INIT_TIMEOUT,
     DEVICE_IO_TIMEOUT,
     DEVICE_POLL_TIMEOUT,
     FIRMWARE_PATTERN,
+    MODEL_BLU_GATEWAY_GEN3,
     NOTIFY_WS_CLOSED,
     VIRTUAL_COMPONENTS,
     VIRTUAL_COMPONENTS_MIN_FIRMWARE,
@@ -545,13 +547,16 @@ def _parse_dynamic_components(self, components: dict[str, Any]) -> None:
 
     async def _retrieve_blutrv_components(self, components: dict[str, Any]) -> None:
         """Retrieve BLU TRV components."""
+        if self.model != MODEL_BLU_GATEWAY_GEN3:
+            return
+
         if TYPE_CHECKING:
             assert self._config
             assert self._status
 
         for component in components.get("components", []):
             _key = component["key"].split(":")
-            if _key[0] == "blutrv":
+            if _key[0] == BLU_TRV_IDENTIFIER:
                 calls = [
                     ("BluTrv.GetRemoteConfig", {"id": _key[1]}),
                     ("BluTrv.GetRemoteStatus", {"id": _key[1]}),