Fix #269 Adjust method to get list of devices to current ZHA code

Author: mdeweerd

custom_components/zha_toolkit/ota.py

@@ -44,7 +44,7 @@ async def download_koenkk_ota(listener, ota_dir):
     # Get manufacturers
     manfs = {}
     for info in [
-        device.zha_device_info for device in listener.devices.values()
+        device.zha_device_info for device in u.get_zha_devices(listener)
     ]:
         manfs[info["manufacturer_code"]] = True
 
@@ -111,7 +111,7 @@ async def download_sonoff_ota(listener, ota_dir):
     # Get manufacturers
     manfs = {}
     for info in [
-        device.zha_device_info for device in listener.devices.values()
+        device.zha_device_info for device in u.get_zha_devices(listener)
     ]:
         manfs[info["manufacturer_code"]] = True
 

custom_components/zha_toolkit/utils.py

@@ -58,7 +58,7 @@
     MANIFEST: dict[str, str | list[str]] = {}
 
 
-def get_zha_gateway(hass: HomeAssistant) -> ZHAGateway:
+def get_zha_gateway(hass: HomeAssistant) -> ZHAGateway | zha_helpers.ZHAGatewayProxy:
     """Get the ZHA gateway object."""
     if parse_version(HA_VERSION) >= parse_version("2024.8"):
         return zha_helpers.get_zha_gateway(hass)
@@ -67,6 +67,15 @@ def get_zha_gateway(hass: HomeAssistant) -> ZHAGateway:
     return zha.gateway
 
 
+def get_zha_devices(listener: ZHAGateway | zha_helpers.ZHAGatewayProxy):
+   devices = getattr(listener, "device_proxies", None)
+   if devices is None:
+       # Old method
+       devices = getattr(listener, "devices", None)
+   if devices is not None:
+       return devices.values()
+
+
 def get_zha_gateway_hass(
     hass: HomeAssistant,
 ) -> ZHAGateway | zha_helpers.ZHAGatewayProxy:

custom_components/zha_toolkit/zha.py

@@ -45,7 +45,7 @@ async def zha_devices(
         #  'signature'
         #  'endpoints'
 
-    devices = [device.zha_device_info for device in listener.devices.values()]
+    devices = [device.zha_device_info for device in listener.device_proxies.values()]
 
     if ieee is not None:
         ieee = str(ieee)