From cee8f2cabb3a01ac794cac6bcb7e18cc06b349e9 Mon Sep 17 00:00:00 2001 From: Aaron Bach Date: Tue, 18 Oct 2022 02:40:49 -0600 Subject: [PATCH] Don't add RainMachine restriction switches if underlying data is missing (#80502) --- homeassistant/components/rainmachine/switch.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/rainmachine/switch.py b/homeassistant/components/rainmachine/switch.py index 56ac814e2eb..db560c3c64c 100644 --- a/homeassistant/components/rainmachine/switch.py +++ b/homeassistant/components/rainmachine/switch.py @@ -34,7 +34,7 @@ from .model import ( RainMachineEntityDescriptionMixinDataKey, RainMachineEntityDescriptionMixinUid, ) -from .util import RUN_STATE_MAP +from .util import RUN_STATE_MAP, key_exists ATTR_AREA = "area" ATTR_CS_ON = "cs_on" @@ -237,6 +237,8 @@ async def async_setup_entry( # Add switches to control restrictions: for description in RESTRICTIONS_SWITCH_DESCRIPTIONS: + if not key_exists(coordinator.data, description.data_key): + continue entities.append(RainMachineRestrictionSwitch(entry, data, description)) async_add_entities(entities)