mirror of
https://github.com/home-assistant/core.git
synced 2025-04-25 09:47:52 +00:00
bd0da03eb9
* Add number entity * Catch exceptions * Add test coverage * Add translation * Fix exception message * Simplify number.py * Remove dead code
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
"""Number platform for Palazzetti settings."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pypalazzetti.exceptions import CommunicationError, ValidationError
|
|
|
|
from homeassistant.components.number import NumberDeviceClass, NumberEntity
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from . import PalazzettiConfigEntry
|
|
from .const import DOMAIN
|
|
from .coordinator import PalazzettiDataUpdateCoordinator
|
|
from .entity import PalazzettiEntity
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: PalazzettiConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Set up Palazzetti number platform."""
|
|
async_add_entities([PalazzettiCombustionPowerEntity(config_entry.runtime_data)])
|
|
|
|
|
|
class PalazzettiCombustionPowerEntity(PalazzettiEntity, NumberEntity):
|
|
"""Representation of Palazzetti number entity for Combustion power."""
|
|
|
|
_attr_translation_key = "combustion_power"
|
|
_attr_device_class = NumberDeviceClass.POWER_FACTOR
|
|
_attr_native_min_value = 1
|
|
_attr_native_max_value = 5
|
|
_attr_native_step = 1
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: PalazzettiDataUpdateCoordinator,
|
|
) -> None:
|
|
"""Initialize the Palazzetti number entity."""
|
|
super().__init__(coordinator)
|
|
self._attr_unique_id = f"{coordinator.config_entry.unique_id}-combustion_power"
|
|
|
|
@property
|
|
def native_value(self) -> float:
|
|
"""Return the state of the setting entity."""
|
|
return self.coordinator.client.power_mode
|
|
|
|
async def async_set_native_value(self, value: float) -> None:
|
|
"""Update the setting."""
|
|
try:
|
|
await self.coordinator.client.set_power_mode(int(value))
|
|
except CommunicationError as err:
|
|
raise HomeAssistantError(
|
|
translation_domain=DOMAIN, translation_key="cannot_connect"
|
|
) from err
|
|
except ValidationError as err:
|
|
raise ServiceValidationError(
|
|
translation_domain=DOMAIN,
|
|
translation_key="invalid_combustion_power",
|
|
translation_placeholders={
|
|
"value": str(value),
|
|
},
|
|
) from err
|
|
|
|
await self.coordinator.async_request_refresh()
|