From fd6a2b078b744b4018113702506a7262fe4d5f50 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Fri, 27 Nov 2020 13:09:37 -0700 Subject: [PATCH] Fix exception upon lock initialization on V2 SimpliSafe systems (#43705) --- homeassistant/components/simplisafe/lock.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/homeassistant/components/simplisafe/lock.py b/homeassistant/components/simplisafe/lock.py index 000b1cd9abb..08ffb82d24f 100644 --- a/homeassistant/components/simplisafe/lock.py +++ b/homeassistant/components/simplisafe/lock.py @@ -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):