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-16 13:33:46 +00:00
committed by Paulus Schoutsen
co-authored by Maciej Bieniek
parent 5d518b5365
commit b856b0e15d
+4 -1
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"]