Don't add RainMachine restriction switches if underlying data is missing (#80502)

This commit is contained in:
Aaron Bach 2022-10-18 02:40:49 -06:00 committed by GitHub
parent b09e95431c
commit cee8f2cabb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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)