Allow SimpliSafe startup to retry on failure (#50211)

* Allow SimpliSafe startup to retry on failure

* Update __init__.py

* Black

Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>
This commit is contained in:
Aaron Bach 2021-05-06 20:03:35 -06:00 committed by GitHub
parent 5ec09eab42
commit fec02c88af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -213,11 +213,14 @@ async def async_setup_entry(hass, config_entry): # noqa: C901
_async_save_refresh_token(hass, config_entry, api.refresh_token)
simplisafe = hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id] = SimpliSafe(
hass, api, config_entry
)
await simplisafe.async_init()
simplisafe = SimpliSafe(hass, api, config_entry)
try:
await simplisafe.async_init()
except SimplipyError as err:
raise ConfigEntryNotReady from err
hass.data[DOMAIN][DATA_CLIENT][config_entry.entry_id] = simplisafe
hass.config_entries.async_setup_platforms(config_entry, PLATFORMS)
@callback