Fix exception upon lock initialization on V2 SimpliSafe systems (#43705)

pull/43707/head
Aaron Bach 2020-11-27 13:09:37 -07:00 committed by GitHub
parent 0d5dc4aeee
commit fd6a2b078b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 7 deletions

View File

@ -17,13 +17,17 @@ ATTR_PIN_PAD_LOW_BATTERY = "pin_pad_low_battery"
async def async_setup_entry(hass, entry, async_add_entities):
"""Set up SimpliSafe locks based on a config entry."""
simplisafe = hass.data[DOMAIN][DATA_CLIENT][entry.entry_id]
async_add_entities(
[
SimpliSafeLock(simplisafe, system, lock)
for system in simplisafe.systems.values()
for lock in system.locks.values()
]
)
locks = []
for system in simplisafe.systems.values():
if system.version == 2:
LOGGER.info("Skipping lock setup for V2 system: %s", system.system_id)
continue
for lock in system.locks.values():
locks.append(SimpliSafeLock(simplisafe, system, lock))
async_add_entities(locks)
class SimpliSafeLock(SimpliSafeEntity, LockEntity):