59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
"""Entity representing a Sonos number control."""
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.components.number import NumberEntity
|
|
from homeassistant.const import ENTITY_CATEGORY_CONFIG
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
|
|
from .const import SONOS_CREATE_LEVELS
|
|
from .entity import SonosEntity
|
|
from .helpers import soco_error
|
|
from .speaker import SonosSpeaker
|
|
|
|
LEVEL_TYPES = ("bass", "treble")
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Set up the Sonos number platform from a config entry."""
|
|
|
|
@callback
|
|
def _async_create_entities(speaker: SonosSpeaker) -> None:
|
|
entities = []
|
|
for level_type in LEVEL_TYPES:
|
|
entities.append(SonosLevelEntity(speaker, level_type))
|
|
async_add_entities(entities)
|
|
|
|
config_entry.async_on_unload(
|
|
async_dispatcher_connect(hass, SONOS_CREATE_LEVELS, _async_create_entities)
|
|
)
|
|
|
|
|
|
class SonosLevelEntity(SonosEntity, NumberEntity):
|
|
"""Representation of a Sonos level entity."""
|
|
|
|
_attr_entity_category = ENTITY_CATEGORY_CONFIG
|
|
_attr_min_value = -10
|
|
_attr_max_value = 10
|
|
|
|
def __init__(self, speaker: SonosSpeaker, level_type: str) -> None:
|
|
"""Initialize the level entity."""
|
|
super().__init__(speaker)
|
|
self._attr_unique_id = f"{self.soco.uid}-{level_type}"
|
|
self._attr_name = f"{self.speaker.zone_name} {level_type.capitalize()}"
|
|
self.level_type = level_type
|
|
|
|
async def _async_poll(self) -> None:
|
|
"""Poll the value if subscriptions are not working."""
|
|
# Handled by SonosSpeaker
|
|
|
|
@soco_error()
|
|
def set_value(self, value: float) -> None:
|
|
"""Set a new value."""
|
|
setattr(self.soco, self.level_type, value)
|
|
|
|
@property
|
|
def value(self) -> float:
|
|
"""Return the current value."""
|
|
return getattr(self.speaker, self.level_type)
|