select.py

Go to the documentation of this file.

"""The select entities for musiccast."""
 
from __future__ import annotations
 
from aiomusiccast.capabilities import OptionSetter
 
from homeassistant.components.select import SelectEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .const import DOMAIN, TRANSLATION_KEY_MAPPING
from .coordinator import MusicCastDataUpdateCoordinator
from .entity import MusicCastCapabilityEntity
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: ConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up MusicCast select entities based on a config entry."""
    coordinator: MusicCastDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
 
    select_entities = [
        SelectableCapability(coordinator, capability)
        for capability in coordinator.data.capabilities
        if isinstance(capability, OptionSetter)
    ]
 
    select_entities.extend(
        SelectableCapability(coordinator, capability, zone)
        for zone, data in coordinator.data.zones.items()
        for capability in data.capabilities
        if isinstance(capability, OptionSetter)
    )
 
    async_add_entities(select_entities)
 
 
class SelectableCapability(MusicCastCapabilityEntity, SelectEntity):
    """Representation of a MusicCast Select entity."""
 
    capability: OptionSetter
 
    def __init__(
        self,
        coordinator: MusicCastDataUpdateCoordinator,
        capability: OptionSetter,
        zone_id: str | None = None,
    ) -> None:
        """Initialize the MusicCast Select entity."""
        MusicCastCapabilityEntity.__init__(self, coordinator, capability, zone_id)
        self._attr_options_attr_options = list(capability.options.values())
        self._attr_translation_key_attr_translation_key = TRANSLATION_KEY_MAPPING.get(capability.id)
 
    async def async_select_option(self, option: str) -> None:
        """Select the given option."""
        value = {val: key for key, val in self.capabilitycapability.options.items()}[option]
        await self.capabilitycapability.set(value)
        self._attr_translation_key_attr_translation_key = TRANSLATION_KEY_MAPPING.get(self.capabilitycapability.id)
 
    @property
    def current_option(self) -> str | None:
        """Return the currently selected option."""
        return self.capabilitycapability.options.get(self.capabilitycapability.current)