From 27cd418e16e3013b119ecbf4450c8ff4a469aaf8 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Mon, 3 Feb 2020 15:29:17 -0700 Subject: [PATCH] Ensure initial_event_to_use works with error --- .../components/simplisafe/__init__.py | 24 +++++++++---------- 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/homeassistant/components/simplisafe/__init__.py b/homeassistant/components/simplisafe/__init__.py index 6adaf4bd7bf..7047c09bdaa 100644 --- a/homeassistant/components/simplisafe/__init__.py +++ b/homeassistant/components/simplisafe/__init__.py @@ -423,7 +423,7 @@ class SimpliSafe: self._config_entry = config_entry self._emergency_refresh_token_used = False self._hass = hass - self.inital_event_to_use = {} + self.initial_event_to_use = {} self.systems = None self.websocket = SimpliSafeWebsocket(hass, api.websocket) @@ -452,9 +452,9 @@ class SimpliSafe: most_recent_event = await system.get_latest_event() except SimplipyError as err: _LOGGER.error("Error while fetching initial event: %s", err) - self.inital_event_to_use[system.system_id] = None + self.initial_event_to_use[system.system_id] = {} else: - self.inital_event_to_use[system.system_id] = most_recent_event + self.initial_event_to_use[system.system_id] = most_recent_event await self.async_update() @@ -533,18 +533,18 @@ class SimpliSafeEntity(Entity): self._serial = system.serial self._attrs = { - ATTR_LAST_EVENT_INFO: simplisafe.inital_event_to_use[system.system_id][ + ATTR_LAST_EVENT_INFO: simplisafe.initial_event_to_use[system.system_id].get( "info" - ], - ATTR_LAST_EVENT_SENSOR_NAME: simplisafe.inital_event_to_use[ + ), + ATTR_LAST_EVENT_SENSOR_NAME: simplisafe.initial_event_to_use[ system.system_id - ]["sensorName"], - ATTR_LAST_EVENT_SENSOR_TYPE: simplisafe.inital_event_to_use[ + ].get("sensorName"), + ATTR_LAST_EVENT_SENSOR_TYPE: simplisafe.initial_event_to_use[ system.system_id - ]["sensorType"], - ATTR_LAST_EVENT_TIMESTAMP: simplisafe.inital_event_to_use[system.system_id][ - "eventTimestamp" - ], + ].get("sensorType"), + ATTR_LAST_EVENT_TIMESTAMP: simplisafe.initial_event_to_use[ + system.system_id + ].get("eventTimestamp"), ATTR_SYSTEM_ID: system.system_id, }