diff --git a/homeassistant/components/yamaha_musiccast/const.py b/homeassistant/components/yamaha_musiccast/const.py index 5384cc56694..470b3729c08 100644 --- a/homeassistant/components/yamaha_musiccast/const.py +++ b/homeassistant/components/yamaha_musiccast/const.py @@ -56,3 +56,14 @@ ENTITY_CATEGORY_MAPPING = { EntityType.DIAGNOSTIC: ENTITY_CATEGORY_DIAGNOSTIC, EntityType.SYSTEM: ENTITY_CATEGORY_SYSTEM, } + +DEVICE_CLASS_MAPPING = { + "DIMMER": "yamaha_musiccast__dimmer", + "zone_SLEEP": "yamaha_musiccast__zone_sleep", + "zone_TONE_CONTROL_mode": "yamaha_musiccast__zone_tone_control_mode", + "zone_SURR_DECODER_TYPE": "yamaha_musiccast__zone_surr_decoder_type", + "zone_EQUALIZER_mode": "yamaha_musiccast__zone_equalizer_mode", + "zone_LINK_AUDIO_QUALITY": "yamaha_musiccast__zone_link_audio_quality", + "zone_LINK_CONTROL": "yamaha_musiccast__zone_link_control", + "zone_LINK_AUDIO_DELAY": "yamaha_musiccast__zone_link_audio_delay", +} diff --git a/homeassistant/components/yamaha_musiccast/select.py b/homeassistant/components/yamaha_musiccast/select.py index 68be67eac10..d64682c8eb4 100644 --- a/homeassistant/components/yamaha_musiccast/select.py +++ b/homeassistant/components/yamaha_musiccast/select.py @@ -1,4 +1,5 @@ """The select entities for musiccast.""" +from __future__ import annotations from aiomusiccast.capabilities import OptionSetter @@ -8,6 +9,7 @@ from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from . import DOMAIN, MusicCastCapabilityEntity, MusicCastDataUpdateCoordinator +from .const import DEVICE_CLASS_MAPPING async def async_setup_entry( @@ -45,9 +47,9 @@ class SelectableCapapility(MusicCastCapabilityEntity, SelectEntity): await self.capability.set(value) @property - def device_class(self) -> str: - """Return the ID of the capability, to identify the entity for translations.""" - return f"{DOMAIN}__{self.capability.id.lower()}" + def device_class(self) -> str | None: + """Return the device class, to identify the entity for translations.""" + return DEVICE_CLASS_MAPPING.get(self.capability.id) @property def options(self):