1010
1111from .coordinator import OnectaRuntimeData
1212
13-
14- async def async_get_config_entry_diagnostics (hass : HomeAssistant , config_entry : ConfigEntry ) -> dict [str , Any ]:
15- """Return diagnostics for a config entry."""
13+ def get_entitities (hass : HomeAssistant , config_entry : ConfigEntry ):
1614 entity_registry = er .async_get (hass )
1715 entities_data : dict [str , dict [str , Any ]] = {}
1816
@@ -35,14 +33,18 @@ async def async_get_config_entry_diagnostics(hass: HomeAssistant, config_entry:
3533
3634 entities_data [entity_id ] = entity_info
3735
36+ return entities_data
37+
38+ async def async_get_config_entry_diagnostics (hass : HomeAssistant , config_entry : ConfigEntry ) -> dict [str , Any ]:
39+ """Return diagnostics for a config entry."""
3840 onecta_data : OnectaRuntimeData = config_entry .runtime_data
3941 daikin_api = onecta_data .daikin_api
4042 return {
4143 "json_data" : daikin_api .json_data ,
4244 "rate_limits" : daikin_api .rate_limits ,
4345 "options" : config_entry .options ,
4446 "oauth2_token_valid" : daikin_api .session .valid_token ,
45- "entities" : entities_data ,
47+ "entities" : get_entitities ( hass , config_entry ) ,
4648 }
4749
4850
@@ -58,4 +60,5 @@ async def async_get_device_diagnostics(hass: HomeAssistant, config_entry: Config
5860 data ["rate_limits" ] = daikin_api .rate_limits
5961 data ["options" ] = config_entry .options
6062 data ["oauth2_token_valid" ] = daikin_api .session .valid_token
63+ data ["entities" ] = get_entitities (hass , config_entry ),
6164 return data
0 commit comments