Add select entity to Sensibo (#67741)

This commit is contained in:
G Johansson 2022-03-12 03:23:24 +01:00 committed by GitHub
parent dc31f420ed
commit c4b3e2b9cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 136 additions and 1 deletions

View File

@ -1017,6 +1017,7 @@ omit =
homeassistant/components/sensibo/diagnostics.py
homeassistant/components/sensibo/entity.py
homeassistant/components/sensibo/number.py
homeassistant/components/sensibo/select.py
homeassistant/components/serial/sensor.py
homeassistant/components/serial_pm/sensor.py
homeassistant/components/sesame/lock.py

View File

@ -12,7 +12,7 @@ LOGGER = logging.getLogger(__package__)
DEFAULT_SCAN_INTERVAL = 60
DOMAIN = "sensibo"
PLATFORMS = [Platform.CLIMATE, Platform.NUMBER]
PLATFORMS = [Platform.CLIMATE, Platform.NUMBER, Platform.SELECT]
ALL = ["all"]
DEFAULT_NAME = "Sensibo"
TIMEOUT = 8

View File

@ -91,6 +91,8 @@ class SensiboDataUpdateCoordinator(DataUpdateCoordinator):
running = ac_states.get("on")
fan_mode = ac_states.get("fanLevel")
swing_mode = ac_states.get("swing")
horizontal_swing_mode = ac_states.get("horizontalSwing")
light_mode = ac_states.get("light")
available = dev["connectionStatus"].get("isAlive", True)
capabilities = dev["remoteCapabilities"]
hvac_modes = list(capabilities["modes"])
@ -99,6 +101,8 @@ class SensiboDataUpdateCoordinator(DataUpdateCoordinator):
current_capabilities = capabilities["modes"][ac_states.get("mode")]
fan_modes = current_capabilities.get("fanLevels")
swing_modes = current_capabilities.get("swing")
horizontal_swing_modes = current_capabilities.get("horizontalSwing")
light_modes = current_capabilities.get("light")
temperature_unit_key = dev.get("temperatureUnit") or ac_states.get(
"temperatureUnit"
)
@ -123,6 +127,10 @@ class SensiboDataUpdateCoordinator(DataUpdateCoordinator):
full_features.add("swing")
if "fanLevels" in capabilities["modes"][mode]:
full_features.add("fanLevel")
if "horizontalSwing" in capabilities["modes"][mode]:
full_features.add("horizontalSwing")
if "light" in capabilities["modes"][mode]:
full_features.add("light")
state = hvac_mode if hvac_mode else "off"
@ -167,10 +175,14 @@ class SensiboDataUpdateCoordinator(DataUpdateCoordinator):
"on": running,
"fan_mode": fan_mode,
"swing_mode": swing_mode,
"horizontal_swing_mode": horizontal_swing_mode,
"light_mode": light_mode,
"available": available,
"hvac_modes": hvac_modes,
"fan_modes": fan_modes,
"swing_modes": swing_modes,
"horizontal_swing_modes": horizontal_swing_modes,
"light_modes": light_modes,
"temp_unit": temperature_unit_key,
"temp_list": temperatures_list,
"temp_step": temperature_step,

View File

@ -41,6 +41,11 @@ class SensiboBaseEntity(CoordinatorEntity):
suggested_area=device["name"],
)
@property
def device_data(self) -> dict[str, Any]:
"""Return data for device."""
return self.coordinator.data.parsed[self._device_id]
async def async_send_command(
self, command: str, params: dict[str, Any]
) -> dict[str, Any]:

View File

@ -0,0 +1,117 @@
"""Number platform for Sensibo integration."""
from __future__ import annotations
from dataclasses import dataclass
from homeassistant.components.select import SelectEntity, SelectEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN
from .coordinator import SensiboDataUpdateCoordinator
from .entity import SensiboBaseEntity
@dataclass
class SensiboSelectDescriptionMixin:
"""Mixin values for Sensibo entities."""
remote_key: str
remote_options: str
@dataclass
class SensiboSelectEntityDescription(
SelectEntityDescription, SensiboSelectDescriptionMixin
):
"""Class describing Sensibo Number entities."""
SELECT_TYPES = (
SensiboSelectEntityDescription(
key="horizontalSwing",
remote_key="horizontal_swing_mode",
remote_options="horizontal_swing_modes",
name="Horizontal Swing",
icon="mdi:air-conditioner",
),
SensiboSelectEntityDescription(
key="light",
remote_key="light_mode",
remote_options="light_modes",
name="Light",
icon="mdi:flashlight",
),
)
async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None:
"""Set up Sensibo number platform."""
coordinator: SensiboDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
async_add_entities(
SensiboSelect(coordinator, device_id, description)
for device_id, device_data in coordinator.data.parsed.items()
for description in SELECT_TYPES
if device_data["hvac_modes"] and description.key in device_data["full_features"]
)
class SensiboSelect(SensiboBaseEntity, SelectEntity):
"""Representation of a Sensibo Select."""
entity_description: SensiboSelectEntityDescription
def __init__(
self,
coordinator: SensiboDataUpdateCoordinator,
device_id: str,
entity_description: SensiboSelectEntityDescription,
) -> None:
"""Initiate Sensibo Select."""
super().__init__(coordinator, device_id)
self.entity_description = entity_description
self._attr_unique_id = f"{device_id}-{entity_description.key}"
self._attr_name = (
f"{coordinator.data.parsed[device_id]['name']} {entity_description.name}"
)
@property
def current_option(self) -> str | None:
"""Return the current selected option."""
return self.device_data[self.entity_description.remote_key]
@property
def options(self) -> list[str]:
"""Return possible options."""
return self.device_data[self.entity_description.remote_options] or []
async def async_select_option(self, option: str) -> None:
"""Set state to the selected option."""
if self.entity_description.key not in self.device_data["active_features"]:
raise HomeAssistantError(
f"Current mode {self.device_data['hvac_mode']} doesn't support setting {self.entity_description.name}"
)
params = {
"name": self.entity_description.key,
"value": option,
"ac_states": self.device_data["ac_states"],
"assumed_state": False,
}
result = await self.async_send_command("set_ac_state", params)
if result["result"]["status"] == "Success":
self.device_data[self.entity_description.remote_key] = option
self.async_write_ha_state()
return
failure = result["result"]["failureReason"]
raise HomeAssistantError(
f"Could not set state for device {self.name} due to reason {failure}"
)