420 lines
12 KiB
Python
420 lines
12 KiB
Python
"""Provides device triggers for lutron caseta."""
|
|
from __future__ import annotations
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.automation import (
|
|
AutomationActionType,
|
|
AutomationTriggerInfo,
|
|
)
|
|
from homeassistant.components.device_automation import DEVICE_TRIGGER_BASE_SCHEMA
|
|
from homeassistant.components.device_automation.exceptions import (
|
|
InvalidDeviceAutomationConfig,
|
|
)
|
|
from homeassistant.components.homeassistant.triggers import event as event_trigger
|
|
from homeassistant.const import (
|
|
CONF_DEVICE_ID,
|
|
CONF_DOMAIN,
|
|
CONF_EVENT,
|
|
CONF_PLATFORM,
|
|
CONF_TYPE,
|
|
)
|
|
from homeassistant.core import CALLBACK_TYPE, HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.helpers.typing import ConfigType
|
|
|
|
from .const import (
|
|
ACTION_PRESS,
|
|
ACTION_RELEASE,
|
|
ATTR_ACTION,
|
|
ATTR_BUTTON_NUMBER,
|
|
ATTR_SERIAL,
|
|
CONF_SUBTYPE,
|
|
DOMAIN,
|
|
LUTRON_CASETA_BUTTON_EVENT,
|
|
)
|
|
from .models import LutronCasetaData
|
|
|
|
SUPPORTED_INPUTS_EVENTS_TYPES = [ACTION_PRESS, ACTION_RELEASE]
|
|
|
|
LUTRON_BUTTON_TRIGGER_SCHEMA = DEVICE_TRIGGER_BASE_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_TYPE): vol.In(SUPPORTED_INPUTS_EVENTS_TYPES),
|
|
}
|
|
)
|
|
|
|
|
|
PICO_2_BUTTON_BUTTON_TYPES_TO_LIP = {
|
|
"on": 2,
|
|
"off": 4,
|
|
}
|
|
PICO_2_BUTTON_BUTTON_TYPES_TO_LEAP = {
|
|
"on": 0,
|
|
"off": 2,
|
|
}
|
|
LEAP_TO_PICO_2_BUTTON_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_2_BUTTON_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_2_BUTTON_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(PICO_2_BUTTON_BUTTON_TYPES_TO_LIP),
|
|
}
|
|
)
|
|
|
|
|
|
PICO_2_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LIP = {
|
|
"on": 2,
|
|
"off": 4,
|
|
"raise": 5,
|
|
"lower": 6,
|
|
}
|
|
PICO_2_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LEAP = {
|
|
"on": 0,
|
|
"off": 2,
|
|
"raise": 3,
|
|
"lower": 4,
|
|
}
|
|
LEAP_TO_PICO_2_BUTTON_RAISE_LOWER_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_2_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_2_BUTTON_RAISE_LOWER_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(
|
|
PICO_2_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LIP
|
|
),
|
|
}
|
|
)
|
|
|
|
|
|
PICO_3_BUTTON_BUTTON_TYPES_TO_LIP = {
|
|
"on": 2,
|
|
"stop": 3,
|
|
"off": 4,
|
|
}
|
|
PICO_3_BUTTON_BUTTON_TYPES_TO_LEAP = {
|
|
"on": 0,
|
|
"stop": 1,
|
|
"off": 2,
|
|
}
|
|
LEAP_TO_PICO_3_BUTTON_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_3_BUTTON_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_3_BUTTON_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(PICO_3_BUTTON_BUTTON_TYPES_TO_LIP),
|
|
}
|
|
)
|
|
|
|
PICO_3_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LIP = {
|
|
"on": 2,
|
|
"stop": 3,
|
|
"off": 4,
|
|
"raise": 5,
|
|
"lower": 6,
|
|
}
|
|
PICO_3_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LEAP = {
|
|
"on": 0,
|
|
"stop": 1,
|
|
"off": 2,
|
|
"raise": 3,
|
|
"lower": 4,
|
|
}
|
|
LEAP_TO_PICO_3_BUTTON_RAISE_LOWER_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_3_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_3_BUTTON_RAISE_LOWER_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(
|
|
PICO_3_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LIP
|
|
),
|
|
}
|
|
)
|
|
|
|
PICO_4_BUTTON_BUTTON_TYPES_TO_LIP = {
|
|
"button_1": 8,
|
|
"button_2": 9,
|
|
"button_3": 10,
|
|
"button_4": 11,
|
|
}
|
|
PICO_4_BUTTON_BUTTON_TYPES_TO_LEAP = {
|
|
"button_1": 1,
|
|
"button_2": 2,
|
|
"button_3": 3,
|
|
"button_4": 4,
|
|
}
|
|
LEAP_TO_PICO_4_BUTTON_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_4_BUTTON_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_4_BUTTON_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(PICO_4_BUTTON_BUTTON_TYPES_TO_LIP),
|
|
}
|
|
)
|
|
|
|
|
|
PICO_4_BUTTON_ZONE_BUTTON_TYPES_TO_LIP = {
|
|
"on": 8,
|
|
"raise": 9,
|
|
"lower": 10,
|
|
"off": 11,
|
|
}
|
|
PICO_4_BUTTON_ZONE_BUTTON_TYPES_TO_LEAP = {
|
|
"on": 1,
|
|
"raise": 2,
|
|
"lower": 3,
|
|
"off": 4,
|
|
}
|
|
LEAP_TO_PICO_4_BUTTON_ZONE_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_4_BUTTON_ZONE_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_4_BUTTON_ZONE_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(PICO_4_BUTTON_ZONE_BUTTON_TYPES_TO_LIP),
|
|
}
|
|
)
|
|
|
|
|
|
PICO_4_BUTTON_SCENE_BUTTON_TYPES_TO_LIP = {
|
|
"button_1": 8,
|
|
"button_2": 9,
|
|
"button_3": 10,
|
|
"off": 11,
|
|
}
|
|
PICO_4_BUTTON_SCENE_BUTTON_TYPES_TO_LEAP = {
|
|
"button_1": 1,
|
|
"button_2": 2,
|
|
"button_3": 3,
|
|
"off": 4,
|
|
}
|
|
LEAP_TO_PICO_4_BUTTON_SCENE_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_4_BUTTON_SCENE_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_4_BUTTON_SCENE_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(PICO_4_BUTTON_SCENE_BUTTON_TYPES_TO_LIP),
|
|
}
|
|
)
|
|
|
|
|
|
PICO_4_BUTTON_2_GROUP_BUTTON_TYPES_TO_LIP = {
|
|
"group_1_button_1": 8,
|
|
"group_1_button_2": 9,
|
|
"group_2_button_1": 10,
|
|
"group_2_button_2": 11,
|
|
}
|
|
PICO_4_BUTTON_2_GROUP_BUTTON_TYPES_TO_LEAP = {
|
|
"group_1_button_1": 1,
|
|
"group_1_button_2": 2,
|
|
"group_2_button_1": 3,
|
|
"group_2_button_2": 4,
|
|
}
|
|
LEAP_TO_PICO_4_BUTTON_2_GROUP_BUTTON_TYPES = {
|
|
v: k for k, v in PICO_4_BUTTON_2_GROUP_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
PICO_4_BUTTON_2_GROUP_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(PICO_4_BUTTON_2_GROUP_BUTTON_TYPES_TO_LIP),
|
|
}
|
|
)
|
|
|
|
FOUR_GROUP_REMOTE_BUTTON_TYPES_TO_LIP = {
|
|
"open_all": 2,
|
|
"stop_all": 3,
|
|
"close_all": 4,
|
|
"raise_all": 5,
|
|
"lower_all": 6,
|
|
"open_1": 10,
|
|
"stop_1": 11,
|
|
"close_1": 12,
|
|
"raise_1": 13,
|
|
"lower_1": 14,
|
|
"open_2": 18,
|
|
"stop_2": 19,
|
|
"close_2": 20,
|
|
"raise_2": 21,
|
|
"lower_2": 22,
|
|
"open_3": 26,
|
|
"stop_3": 27,
|
|
"close_3": 28,
|
|
"raise_3": 29,
|
|
"lower_3": 30,
|
|
"open_4": 34,
|
|
"stop_4": 35,
|
|
"close_4": 36,
|
|
"raise_4": 37,
|
|
"lower_4": 38,
|
|
}
|
|
FOUR_GROUP_REMOTE_BUTTON_TYPES_TO_LEAP = {
|
|
"open_all": 0,
|
|
"stop_all": 1,
|
|
"close_all": 2,
|
|
"raise_all": 3,
|
|
"lower_all": 4,
|
|
"open_1": 5,
|
|
"stop_1": 6,
|
|
"close_1": 7,
|
|
"raise_1": 8,
|
|
"lower_1": 9,
|
|
"open_2": 10,
|
|
"stop_2": 11,
|
|
"close_2": 12,
|
|
"raise_2": 13,
|
|
"lower_2": 14,
|
|
"open_3": 15,
|
|
"stop_3": 16,
|
|
"close_3": 17,
|
|
"raise_3": 18,
|
|
"lower_3": 19,
|
|
"open_4": 20,
|
|
"stop_4": 21,
|
|
"close_4": 22,
|
|
"raise_4": 23,
|
|
"lower_4": 24,
|
|
}
|
|
LEAP_TO_FOUR_GROUP_REMOTE_BUTTON_TYPES = {
|
|
v: k for k, v in FOUR_GROUP_REMOTE_BUTTON_TYPES_TO_LEAP.items()
|
|
}
|
|
FOUR_GROUP_REMOTE_TRIGGER_SCHEMA = LUTRON_BUTTON_TRIGGER_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_SUBTYPE): vol.In(FOUR_GROUP_REMOTE_BUTTON_TYPES_TO_LIP),
|
|
}
|
|
)
|
|
|
|
DEVICE_TYPE_SCHEMA_MAP = {
|
|
"Pico2Button": PICO_2_BUTTON_TRIGGER_SCHEMA,
|
|
"Pico2ButtonRaiseLower": PICO_2_BUTTON_RAISE_LOWER_TRIGGER_SCHEMA,
|
|
"Pico3Button": PICO_3_BUTTON_TRIGGER_SCHEMA,
|
|
"Pico3ButtonRaiseLower": PICO_3_BUTTON_RAISE_LOWER_TRIGGER_SCHEMA,
|
|
"Pico4Button": PICO_4_BUTTON_TRIGGER_SCHEMA,
|
|
"Pico4ButtonScene": PICO_4_BUTTON_SCENE_TRIGGER_SCHEMA,
|
|
"Pico4ButtonZone": PICO_4_BUTTON_ZONE_TRIGGER_SCHEMA,
|
|
"Pico4Button2Group": PICO_4_BUTTON_2_GROUP_TRIGGER_SCHEMA,
|
|
"FourGroupRemote": FOUR_GROUP_REMOTE_TRIGGER_SCHEMA,
|
|
}
|
|
|
|
DEVICE_TYPE_SUBTYPE_MAP_TO_LIP = {
|
|
"Pico2Button": PICO_2_BUTTON_BUTTON_TYPES_TO_LIP,
|
|
"Pico2ButtonRaiseLower": PICO_2_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LIP,
|
|
"Pico3Button": PICO_3_BUTTON_BUTTON_TYPES_TO_LIP,
|
|
"Pico3ButtonRaiseLower": PICO_3_BUTTON_RAISE_LOWER_BUTTON_TYPES_TO_LIP,
|
|
"Pico4Button": PICO_4_BUTTON_BUTTON_TYPES_TO_LIP,
|
|
"Pico4ButtonScene": PICO_4_BUTTON_SCENE_BUTTON_TYPES_TO_LIP,
|
|
"Pico4ButtonZone": PICO_4_BUTTON_ZONE_BUTTON_TYPES_TO_LIP,
|
|
"Pico4Button2Group": PICO_4_BUTTON_2_GROUP_BUTTON_TYPES_TO_LIP,
|
|
"FourGroupRemote": FOUR_GROUP_REMOTE_BUTTON_TYPES_TO_LIP,
|
|
}
|
|
|
|
LEAP_TO_DEVICE_TYPE_SUBTYPE_MAP = {
|
|
"Pico2Button": LEAP_TO_PICO_2_BUTTON_BUTTON_TYPES,
|
|
"Pico2ButtonRaiseLower": LEAP_TO_PICO_2_BUTTON_RAISE_LOWER_BUTTON_TYPES,
|
|
"Pico3Button": LEAP_TO_PICO_3_BUTTON_BUTTON_TYPES,
|
|
"Pico3ButtonRaiseLower": LEAP_TO_PICO_3_BUTTON_RAISE_LOWER_BUTTON_TYPES,
|
|
"Pico4Button": LEAP_TO_PICO_4_BUTTON_BUTTON_TYPES,
|
|
"Pico4ButtonScene": LEAP_TO_PICO_4_BUTTON_SCENE_BUTTON_TYPES,
|
|
"Pico4ButtonZone": LEAP_TO_PICO_4_BUTTON_ZONE_BUTTON_TYPES,
|
|
"Pico4Button2Group": LEAP_TO_PICO_4_BUTTON_2_GROUP_BUTTON_TYPES,
|
|
"FourGroupRemote": LEAP_TO_FOUR_GROUP_REMOTE_BUTTON_TYPES,
|
|
}
|
|
|
|
TRIGGER_SCHEMA = vol.Any(
|
|
PICO_2_BUTTON_TRIGGER_SCHEMA,
|
|
PICO_3_BUTTON_RAISE_LOWER_TRIGGER_SCHEMA,
|
|
PICO_4_BUTTON_TRIGGER_SCHEMA,
|
|
PICO_4_BUTTON_SCENE_TRIGGER_SCHEMA,
|
|
PICO_4_BUTTON_ZONE_TRIGGER_SCHEMA,
|
|
PICO_4_BUTTON_2_GROUP_TRIGGER_SCHEMA,
|
|
FOUR_GROUP_REMOTE_TRIGGER_SCHEMA,
|
|
)
|
|
|
|
|
|
async def async_validate_trigger_config(
|
|
hass: HomeAssistant, config: ConfigType
|
|
) -> ConfigType:
|
|
"""Validate config."""
|
|
# if device is available verify parameters against device capabilities
|
|
device = get_button_device_by_dr_id(hass, config[CONF_DEVICE_ID])
|
|
|
|
if not device:
|
|
return config
|
|
|
|
if not (schema := DEVICE_TYPE_SCHEMA_MAP.get(device["type"])):
|
|
raise InvalidDeviceAutomationConfig(
|
|
f"Device type {device['type']} not supported: {config[CONF_DEVICE_ID]}"
|
|
)
|
|
|
|
return schema(config)
|
|
|
|
|
|
async def async_get_triggers(
|
|
hass: HomeAssistant, device_id: str
|
|
) -> list[dict[str, str]]:
|
|
"""List device triggers for lutron caseta devices."""
|
|
triggers = []
|
|
|
|
if not (device := get_button_device_by_dr_id(hass, device_id)):
|
|
raise InvalidDeviceAutomationConfig(f"Device not found: {device_id}")
|
|
|
|
valid_buttons = DEVICE_TYPE_SUBTYPE_MAP_TO_LIP.get(device["type"], {})
|
|
|
|
for trigger in SUPPORTED_INPUTS_EVENTS_TYPES:
|
|
for subtype in valid_buttons:
|
|
triggers.append(
|
|
{
|
|
CONF_PLATFORM: "device",
|
|
CONF_DEVICE_ID: device_id,
|
|
CONF_DOMAIN: DOMAIN,
|
|
CONF_TYPE: trigger,
|
|
CONF_SUBTYPE: subtype,
|
|
}
|
|
)
|
|
|
|
return triggers
|
|
|
|
|
|
def _device_model_to_type(model: str) -> str:
|
|
"""Convert a lutron_caseta device registry entry model to type."""
|
|
_, device_type = model.split(" ")
|
|
return device_type.replace("(", "").replace(")", "")
|
|
|
|
|
|
async def async_attach_trigger(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
action: AutomationActionType,
|
|
automation_info: AutomationTriggerInfo,
|
|
) -> CALLBACK_TYPE:
|
|
"""Attach a trigger."""
|
|
device_registry = dr.async_get(hass)
|
|
device = device_registry.async_get(config[CONF_DEVICE_ID])
|
|
device_type = _device_model_to_type(device.model)
|
|
_, serial = list(device.identifiers)[0]
|
|
schema = DEVICE_TYPE_SCHEMA_MAP.get(device_type)
|
|
valid_buttons = DEVICE_TYPE_SUBTYPE_MAP_TO_LIP.get(device_type)
|
|
config = schema(config)
|
|
event_config = {
|
|
event_trigger.CONF_PLATFORM: CONF_EVENT,
|
|
event_trigger.CONF_EVENT_TYPE: LUTRON_CASETA_BUTTON_EVENT,
|
|
event_trigger.CONF_EVENT_DATA: {
|
|
ATTR_SERIAL: serial,
|
|
ATTR_BUTTON_NUMBER: valid_buttons[config[CONF_SUBTYPE]],
|
|
ATTR_ACTION: config[CONF_TYPE],
|
|
},
|
|
}
|
|
event_config = event_trigger.TRIGGER_SCHEMA(event_config)
|
|
return await event_trigger.async_attach_trigger(
|
|
hass, event_config, action, automation_info, platform_type="device"
|
|
)
|
|
|
|
|
|
def get_button_device_by_dr_id(hass: HomeAssistant, device_id: str):
|
|
"""Get a lutron device for the given device id."""
|
|
if DOMAIN not in hass.data:
|
|
return None
|
|
|
|
for entry_id in hass.data[DOMAIN]:
|
|
data: LutronCasetaData = hass.data[DOMAIN][entry_id]
|
|
if device := data.button_devices.get(device_id):
|
|
return device
|
|
|
|
return None
|