mirror of
https://github.com/home-assistant/core.git
synced 2026-05-29 19:57:40 +01:00
Paulus Schoutsen
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""Button platform for Marantz IR integration.
|
|
|
|
Only commands that aren't already exposed by the media player live here:
|
|
speaker A/B, source-direct toggle, and loudness toggle.
|
|
"""
|
|
|
|
from dataclasses import dataclass
|
|
|
|
from infrared_protocols.codes.marantz.audio import MarantzAudioCode
|
|
|
|
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
|
|
from homeassistant.const import CONF_MODEL
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from . import MarantzIrConfigEntry
|
|
from .const import CONF_INFRARED_EMITTER_ENTITY_ID, MODELS
|
|
from .entity import MarantzIrEntity
|
|
|
|
PARALLEL_UPDATES = 1
|
|
|
|
|
|
@dataclass(frozen=True, kw_only=True)
|
|
class MarantzIrButtonEntityDescription(ButtonEntityDescription):
|
|
"""Describes Marantz IR button entity."""
|
|
|
|
command_code: MarantzAudioCode
|
|
|
|
|
|
BUTTON_DESCRIPTIONS: tuple[MarantzIrButtonEntityDescription, ...] = (
|
|
MarantzIrButtonEntityDescription(
|
|
key="speaker_ab",
|
|
translation_key="speaker_ab",
|
|
command_code=MarantzAudioCode.SPEAKER_AB,
|
|
),
|
|
MarantzIrButtonEntityDescription(
|
|
key="source_direct",
|
|
translation_key="source_direct",
|
|
command_code=MarantzAudioCode.SOURCE_DIRECT,
|
|
),
|
|
MarantzIrButtonEntityDescription(
|
|
key="loudness",
|
|
translation_key="loudness",
|
|
command_code=MarantzAudioCode.LOUDNESS,
|
|
),
|
|
)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: MarantzIrConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up Marantz IR buttons from config entry."""
|
|
infrared_entity_id = entry.data[CONF_INFRARED_EMITTER_ENTITY_ID]
|
|
model_codes = MODELS[entry.data[CONF_MODEL]].codes
|
|
async_add_entities(
|
|
MarantzIrButton(entry, infrared_entity_id, description)
|
|
for description in BUTTON_DESCRIPTIONS
|
|
if description.command_code in model_codes
|
|
)
|
|
|
|
|
|
class MarantzIrButton(MarantzIrEntity, ButtonEntity):
|
|
"""Marantz IR button entity."""
|
|
|
|
entity_description: MarantzIrButtonEntityDescription
|
|
|
|
def __init__(
|
|
self,
|
|
entry: MarantzIrConfigEntry,
|
|
infrared_entity_id: str,
|
|
description: MarantzIrButtonEntityDescription,
|
|
) -> None:
|
|
"""Initialize Marantz IR button."""
|
|
super().__init__(entry, infrared_entity_id, unique_id_suffix=description.key)
|
|
self.entity_description = description
|
|
|
|
async def async_press(self) -> None:
|
|
"""Press the button."""
|
|
await self._send_marantz_command(self.entity_description.command_code)
|