evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use TriggerActionType [core, d-h] (#76804)

Browse source
This commit is contained in:
Marc Mueller 2022-08-15 17:39:14 +02:00 • committed by GitHub
parent 1557a7c36d
commit e39672078d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 29 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

13
homeassistant/components/device_automation/toggle_entity.py
View file

@ -3,10 +3,6 @@ from __future__ import annotations
import voluptuous as vol
from homeassistant.components.automation import (
AutomationActionType,
AutomationTriggerInfo,
)
from homeassistant.components.homeassistant.triggers import state as state_trigger
from homeassistant.const import (
ATTR_ENTITY_ID,
@ -23,6 +19,7 @@ from homeassistant.helpers import (
config_validation as cv,
entity_registry as er,
)
from homeassistant.helpers.trigger import TriggerActionType, TriggerInfo
from homeassistant.helpers.typing import ConfigType, TemplateVarsType
from . import DEVICE_TRIGGER_BASE_SCHEMA, entity
@ -155,12 +152,12 @@ def async_condition_from_config(
async def async_attach_trigger(
hass: HomeAssistant,
config: ConfigType,
action: AutomationActionType,
automation_info: AutomationTriggerInfo,
action: TriggerActionType,
trigger_info: TriggerInfo,
) -> CALLBACK_TYPE:
"""Listen for state changes based on configuration."""
if config[CONF_TYPE] not in [CONF_TURNED_ON, CONF_TURNED_OFF]:
return await entity.async_attach_trigger(hass, config, action, automation_info)
return await entity.async_attach_trigger(hass, config, action, trigger_info)
if config[CONF_TYPE] == CONF_TURNED_ON:
to_state = "on"
@ -176,7 +173,7 @@ async def async_attach_trigger(
state_config = await state_trigger.async_validate_trigger_config(hass, state_config)
return await state_trigger.async_attach_trigger(
hass, state_config, action, automation_info, platform_type="device"
hass, state_config, action, trigger_info, platform_type="device"
)