53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
"""Support for Spa Client selects."""
|
|
|
|
from pybalboa import SpaClient, SpaControl
|
|
from pybalboa.enums import LowHighRange
|
|
|
|
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
|
|
from .entity import BalboaEntity
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
|
) -> None:
|
|
"""Set up the spa select entity."""
|
|
spa: SpaClient = hass.data[DOMAIN][entry.entry_id]
|
|
async_add_entities([BalboaTempRangeSelectEntity(spa.temperature_range)])
|
|
|
|
|
|
class BalboaTempRangeSelectEntity(BalboaEntity, SelectEntity):
|
|
"""Representation of a Temperature Range select."""
|
|
|
|
_attr_icon = "mdi:thermometer-lines"
|
|
_attr_name = "Temperature range"
|
|
_attr_unique_id = "temperature_range"
|
|
_attr_translation_key = "temperature_range"
|
|
_attr_options = [
|
|
LowHighRange.LOW.name.lower(),
|
|
LowHighRange.HIGH.name.lower(),
|
|
]
|
|
|
|
def __init__(self, control: SpaControl) -> None:
|
|
"""Initialise the select."""
|
|
super().__init__(control.client, "TempHiLow")
|
|
self._control = control
|
|
|
|
@property
|
|
def current_option(self) -> str | None:
|
|
"""Return current select option."""
|
|
if self._control.state == LowHighRange.HIGH:
|
|
return LowHighRange.HIGH.name.lower()
|
|
return LowHighRange.LOW.name.lower()
|
|
|
|
async def async_select_option(self, option: str) -> None:
|
|
"""Select temperature range high/low mode."""
|
|
if option == LowHighRange.HIGH.name.lower():
|
|
await self._client.set_temperature_range(LowHighRange.HIGH)
|
|
else:
|
|
await self._client.set_temperature_range(LowHighRange.LOW)
|