From 53216883f840dba9cd3fd5c04278cccc9ac0b48d Mon Sep 17 00:00:00 2001 From: "David F. Mulcahey" Date: Wed, 13 Apr 2022 17:02:57 -0400 Subject: [PATCH] ZHA diagnostics fixes (#70000) --- homeassistant/components/zha/diagnostics.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/zha/diagnostics.py b/homeassistant/components/zha/diagnostics.py index 1a775723b0b..ec8dab241de 100644 --- a/homeassistant/components/zha/diagnostics.py +++ b/homeassistant/components/zha/diagnostics.py @@ -29,6 +29,7 @@ KEYS_TO_REDACT = { CONF_UNIQUE_ID, "network_key", CONF_NWK_EXTENDED_PAN_ID, + "partner_ieee", } @@ -50,7 +51,7 @@ async def async_get_config_entry_diagnostics( hass: HomeAssistant, config_entry: ConfigEntry ) -> dict: """Return diagnostics for a config entry.""" - config: dict = hass.data[DATA_ZHA][DATA_ZHA_CONFIG] + config: dict = hass.data[DATA_ZHA].get(DATA_ZHA_CONFIG, {}) gateway: ZHAGateway = hass.data[DATA_ZHA][DATA_ZHA_GATEWAY] return async_redact_data( {