jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-08 05:47:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

guard against invalid trigger and action scenarios (#32512)

Browse Source
This commit is contained in:
David F. Mulcahey 2020-03-05 15:52:09 -05:00 committed by GitHub
parent ae0ea0f088
commit b5022f5bcb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/zha/device_action.py
View File

@ -56,7 +56,10 @@ async def async_call_action_from_config(
async def async_get_actions(hass: HomeAssistant, device_id: str) -> List[dict]: async def async_get_actions(hass: HomeAssistant, device_id: str) -> List[dict]:
"""List device actions.""" """List device actions."""
zha_device = await async_get_zha_device(hass, device_id) try:
zha_device = await async_get_zha_device(hass, device_id)
except (KeyError, AttributeError):
return []
cluster_channels = [ cluster_channels = [
ch.name ch.name
for pool in zha_device.channels.pools for pool in zha_device.channels.pools
@ -81,7 +84,10 @@ async def _execute_service_based_action(
) -> None: ) -> None:
action_type = config[CONF_TYPE] action_type = config[CONF_TYPE]
service_name = SERVICE_NAMES[action_type] service_name = SERVICE_NAMES[action_type]
zha_device = await async_get_zha_device(hass, config[CONF_DEVICE_ID]) try:
zha_device = await async_get_zha_device(hass, config[CONF_DEVICE_ID])
except (KeyError, AttributeError):
return
service_data = {ATTR_IEEE: str(zha_device.ieee)} service_data = {ATTR_IEEE: str(zha_device.ieee)}

11
homeassistant/components/zha/device_trigger.py
View File

@ -27,7 +27,10 @@ async def async_validate_trigger_config(hass, config):
if "zha" in hass.config.components: if "zha" in hass.config.components:
trigger = (config[CONF_TYPE], config[CONF_SUBTYPE]) trigger = (config[CONF_TYPE], config[CONF_SUBTYPE])
zha_device = await async_get_zha_device(hass, config[CONF_DEVICE_ID]) try:
zha_device = await async_get_zha_device(hass, config[CONF_DEVICE_ID])
except (KeyError, AttributeError):
raise InvalidDeviceAutomationConfig
if ( if (
zha_device.device_automation_triggers is None zha_device.device_automation_triggers is None
or trigger not in zha_device.device_automation_triggers or trigger not in zha_device.device_automation_triggers
@ -40,8 +43,10 @@ async def async_validate_trigger_config(hass, config):
async def async_attach_trigger(hass, config, action, automation_info): async def async_attach_trigger(hass, config, action, automation_info):
"""Listen for state changes based on configuration.""" """Listen for state changes based on configuration."""
trigger = (config[CONF_TYPE], config[CONF_SUBTYPE]) trigger = (config[CONF_TYPE], config[CONF_SUBTYPE])
zha_device = await async_get_zha_device(hass, config[CONF_DEVICE_ID]) try:
zha_device = await async_get_zha_device(hass, config[CONF_DEVICE_ID])
except (KeyError, AttributeError):
return None
trigger = zha_device.device_automation_triggers[trigger] trigger = zha_device.device_automation_triggers[trigger]
event_config = { event_config = {