Guard both Shelly 2 & Shelly 2.5 in roller mode (#40086)

Co-authored-by: Maciej Bieniek <bieniu@users.noreply.github.com>
This commit is contained in:
Shay Levy 2020-09-15 17:29:24 +03:00 committed by GitHub
parent b116e58620
commit 4fbd4957bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -14,7 +14,10 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
wrapper = hass.data[DOMAIN][config_entry.entry_id]
# In roller mode the relay blocks exist but do not contain required info
if wrapper.model == "SHSW-25" and wrapper.device.settings["mode"] != "relay":
if (
wrapper.model in ["SHSW-21", "SHSW-25"]
and wrapper.device.settings["mode"] != "relay"
):
return
relay_blocks = [block for block in wrapper.device.blocks if block.type == "relay"]