133 lines
4.6 KiB
Python
133 lines
4.6 KiB
Python
"""Number platform for Sensibo integration."""
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
|
|
import async_timeout
|
|
|
|
from homeassistant.components.number import NumberEntity, NumberEntityDescription
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
|
|
from homeassistant.helpers.entity import DeviceInfo, EntityCategory
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .const import DOMAIN, LOGGER, SENSIBO_ERRORS, TIMEOUT
|
|
from .coordinator import SensiboDataUpdateCoordinator
|
|
|
|
|
|
@dataclass
|
|
class SensiboEntityDescriptionMixin:
|
|
"""Mixin values for Sensibo entities."""
|
|
|
|
remote_key: str
|
|
|
|
|
|
@dataclass
|
|
class SensiboNumberEntityDescription(
|
|
NumberEntityDescription, SensiboEntityDescriptionMixin
|
|
):
|
|
"""Class describing Sensibo Number entities."""
|
|
|
|
|
|
NUMBER_TYPES = (
|
|
SensiboNumberEntityDescription(
|
|
key="calibration_temp",
|
|
remote_key="temperature",
|
|
name="Temperature calibration",
|
|
icon="mdi:thermometer",
|
|
entity_category=EntityCategory.CONFIG,
|
|
entity_registry_enabled_default=False,
|
|
min_value=-10,
|
|
max_value=10,
|
|
step=0.1,
|
|
),
|
|
SensiboNumberEntityDescription(
|
|
key="calibration_hum",
|
|
remote_key="humidity",
|
|
name="Humidity calibration",
|
|
icon="mdi:water",
|
|
entity_category=EntityCategory.CONFIG,
|
|
entity_registry_enabled_default=False,
|
|
min_value=-10,
|
|
max_value=10,
|
|
step=0.1,
|
|
),
|
|
)
|
|
|
|
|
|
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(
|
|
SensiboNumber(coordinator, device_id, description)
|
|
for device_id, device_data in coordinator.data.items()
|
|
for description in NUMBER_TYPES
|
|
if device_data["hvac_modes"] and device_data["temp"]
|
|
)
|
|
|
|
|
|
class SensiboNumber(CoordinatorEntity, NumberEntity):
|
|
"""Representation of a Sensibo numbers."""
|
|
|
|
coordinator: SensiboDataUpdateCoordinator
|
|
entity_description: SensiboNumberEntityDescription
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: SensiboDataUpdateCoordinator,
|
|
device_id: str,
|
|
entity_description: SensiboNumberEntityDescription,
|
|
) -> None:
|
|
"""Initiate Sensibo Number."""
|
|
super().__init__(coordinator)
|
|
self.entity_description = entity_description
|
|
self._device_id = device_id
|
|
self._client = coordinator.client
|
|
self._attr_unique_id = f"{device_id}-{entity_description.key}"
|
|
self._attr_name = (
|
|
f"{coordinator.data[device_id]['name']} {entity_description.name}"
|
|
)
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, coordinator.data[device_id]["id"])},
|
|
name=coordinator.data[device_id]["name"],
|
|
connections={(CONNECTION_NETWORK_MAC, coordinator.data[device_id]["mac"])},
|
|
manufacturer="Sensibo",
|
|
configuration_url="https://home.sensibo.com/",
|
|
model=coordinator.data[device_id]["model"],
|
|
sw_version=coordinator.data[device_id]["fw_ver"],
|
|
hw_version=coordinator.data[device_id]["fw_type"],
|
|
suggested_area=coordinator.data[device_id]["name"],
|
|
)
|
|
|
|
@property
|
|
def value(self) -> float:
|
|
"""Return the value from coordinator data."""
|
|
return self.coordinator.data[self._device_id][self.entity_description.key]
|
|
|
|
async def async_set_value(self, value: float) -> None:
|
|
"""Set value for calibration."""
|
|
data = {self.entity_description.remote_key: value}
|
|
try:
|
|
async with async_timeout.timeout(TIMEOUT):
|
|
result = await self._client.async_set_calibration(
|
|
self._device_id,
|
|
data,
|
|
)
|
|
except SENSIBO_ERRORS as err:
|
|
raise HomeAssistantError(
|
|
f"Failed to set calibration for device {self.name} to Sensibo servers: {err}"
|
|
) from err
|
|
LOGGER.debug("Result: %s", result)
|
|
if result["status"] == "success":
|
|
self.coordinator.data[self._device_id][self.entity_description.key] = value
|
|
self.async_write_ha_state()
|
|
return
|
|
raise HomeAssistantError(f"Could not set calibration for device {self.name}")
|