Ensure initial_event_to_use works with error

This commit is contained in:
Aaron Bach
2020-02-03 15:29:17 -07:00
parent fb62c3d8df
commit 27cd418e16

View File

@@ -423,7 +423,7 @@ class SimpliSafe:
self._config_entry = config_entry self._config_entry = config_entry
self._emergency_refresh_token_used = False self._emergency_refresh_token_used = False
self._hass = hass self._hass = hass
self.inital_event_to_use = {} self.initial_event_to_use = {}
self.systems = None self.systems = None
self.websocket = SimpliSafeWebsocket(hass, api.websocket) self.websocket = SimpliSafeWebsocket(hass, api.websocket)
@@ -452,9 +452,9 @@ class SimpliSafe:
most_recent_event = await system.get_latest_event() most_recent_event = await system.get_latest_event()
except SimplipyError as err: except SimplipyError as err:
_LOGGER.error("Error while fetching initial event: %s", 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: 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() await self.async_update()
@@ -533,18 +533,18 @@ class SimpliSafeEntity(Entity):
self._serial = system.serial self._serial = system.serial
self._attrs = { 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" "info"
], ),
ATTR_LAST_EVENT_SENSOR_NAME: simplisafe.inital_event_to_use[ ATTR_LAST_EVENT_SENSOR_NAME: simplisafe.initial_event_to_use[
system.system_id system.system_id
]["sensorName"], ].get("sensorName"),
ATTR_LAST_EVENT_SENSOR_TYPE: simplisafe.inital_event_to_use[ ATTR_LAST_EVENT_SENSOR_TYPE: simplisafe.initial_event_to_use[
system.system_id system.system_id
]["sensorType"], ].get("sensorType"),
ATTR_LAST_EVENT_TIMESTAMP: simplisafe.inital_event_to_use[system.system_id][ ATTR_LAST_EVENT_TIMESTAMP: simplisafe.initial_event_to_use[
"eventTimestamp" system.system_id
], ].get("eventTimestamp"),
ATTR_SYSTEM_ID: system.system_id, ATTR_SYSTEM_ID: system.system_id,
} }