mirror of
https://github.com/home-assistant/core.git
synced 2026-09-08 06:31:35 +01:00
88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
"""Button platform for LED Infrared integration."""
|
|
|
|
from typing import override
|
|
|
|
from homeassistant.components.button import ButtonEntity
|
|
from homeassistant.components.infrared import InfraredEmitterConsumerEntity
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .const import CONF_DEVICE_TYPE, CONF_INFRARED_ENTITY_ID, LEDIrDeviceType
|
|
from .entity import LEDIrBaseEntity
|
|
|
|
PARALLEL_UPDATES = 1
|
|
|
|
SUPPORTED_BUTTONS = {
|
|
LEDIrDeviceType.GENERIC_24_KEY: ["brightness_up", "brightness_down"],
|
|
LEDIrDeviceType.GENERIC_13_KEY: ["brightness_up", "brightness_down", "timer"],
|
|
LEDIrDeviceType.GENERIC_40_KEY: [
|
|
"brightness_up",
|
|
"brightness_down",
|
|
"white_brightness_up",
|
|
"white_brightness_down",
|
|
"white_on",
|
|
"white_off",
|
|
"white_brightness_25",
|
|
"white_brightness_50",
|
|
"white_brightness_75",
|
|
"white_brightness_100",
|
|
"quick",
|
|
"slow",
|
|
],
|
|
LEDIrDeviceType.GENERIC_44_KEY: [
|
|
"brightness_up",
|
|
"brightness_down",
|
|
"red_up",
|
|
"green_up",
|
|
"blue_up",
|
|
"red_down",
|
|
"green_down",
|
|
"blue_down",
|
|
"quick",
|
|
"slow",
|
|
],
|
|
}
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: ConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up platform from config entry."""
|
|
if not (infrared_entity_id := entry.data.get(CONF_INFRARED_ENTITY_ID)):
|
|
return
|
|
|
|
async_add_entities(
|
|
[
|
|
LEDIrButtonEntity(
|
|
entry, entry.data[CONF_DEVICE_TYPE], infrared_entity_id, key
|
|
)
|
|
for key in SUPPORTED_BUTTONS.get(entry.data[CONF_DEVICE_TYPE], [])
|
|
]
|
|
)
|
|
|
|
|
|
class LEDIrButtonEntity(LEDIrBaseEntity, InfraredEmitterConsumerEntity, ButtonEntity):
|
|
"""Represents a LED Infrared button entity."""
|
|
|
|
def __init__(
|
|
self,
|
|
entry: ConfigEntry,
|
|
device_type: LEDIrDeviceType,
|
|
infrared_entity_id: str,
|
|
key: str,
|
|
) -> None:
|
|
"""Initialize the entity."""
|
|
super().__init__(entry, device_type)
|
|
self._infrared_emitter_entity_id = infrared_entity_id
|
|
self._attr_unique_id = f"{entry.entry_id}_{key}"
|
|
self._key = key
|
|
self._attr_translation_key = key
|
|
|
|
@override
|
|
async def async_press(self) -> None:
|
|
"""Press the button."""
|
|
await self._send_command(self._codes[self._key.upper()].to_command())
|