diff --git a/homeassistant/components/mill/__init__.py b/homeassistant/components/mill/__init__.py index 11199e126cf67d570451e48dd35aa1645c992c21..116b3ef0341d1de74e07dc0ad3d4d594dcff741a 100644 --- a/homeassistant/components/mill/__init__.py +++ b/homeassistant/components/mill/__init__.py @@ -16,7 +16,7 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession from .const import CLOUD, CONNECTION_TYPE, DOMAIN, LOCAL from .coordinator import MillDataUpdateCoordinator -PLATFORMS = [Platform.CLIMATE, Platform.SENSOR] +PLATFORMS = [Platform.CLIMATE, Platform.NUMBER, Platform.SENSOR] async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: diff --git a/homeassistant/components/mill/number.py b/homeassistant/components/mill/number.py new file mode 100644 index 0000000000000000000000000000000000000000..e2425cfba5b24527bac6355b1cfdee8f0c408e72 --- /dev/null +++ b/homeassistant/components/mill/number.py @@ -0,0 +1,83 @@ +"""Support for mill wifi-enabled home heaters.""" + +from __future__ import annotations + +from mill import MillDevice + +from homeassistant.components.number import NumberDeviceClass, NumberEntity +from homeassistant.config_entries import ConfigEntry +from homeassistant.const import CONF_USERNAME, UnitOfPower +from homeassistant.core import HomeAssistant, callback +from homeassistant.helpers.device_registry import DeviceInfo +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.update_coordinator import CoordinatorEntity + +from .const import CLOUD, CONNECTION_TYPE, DOMAIN, MANUFACTURER +from .coordinator import MillDataUpdateCoordinator + + +async def async_setup_entry( + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: + """Set up the Mill Number.""" + if entry.data.get(CONNECTION_TYPE) == CLOUD: + mill_data_coordinator: MillDataUpdateCoordinator = hass.data[DOMAIN][CLOUD][ + entry.data[CONF_USERNAME] + ] + + async_add_entities( + MillNumber(mill_data_coordinator, mill_device) + for mill_device in mill_data_coordinator.data.values() + ) + + +class MillNumber(CoordinatorEntity[MillDataUpdateCoordinator], NumberEntity): + """Representation of a Mill number device.""" + + _attr_device_class = NumberDeviceClass.POWER + _attr_has_entity_name = True + _attr_native_max_value = 2000 + _attr_native_min_value = 0 + _attr_native_step = 1 + _attr_native_unit_of_measurement = UnitOfPower.WATT + + def __init__( + self, + coordinator: MillDataUpdateCoordinator, + mill_device: MillDevice, + ) -> None: + """Initialize the number.""" + super().__init__(coordinator) + + self._id = mill_device.device_id + self._available = False + self._attr_unique_id = f"{mill_device.device_id}_max_heating_power" + self._attr_device_info = DeviceInfo( + identifiers={(DOMAIN, mill_device.device_id)}, + name=mill_device.name, + manufacturer=MANUFACTURER, + model=mill_device.model, + ) + self._update_attr(mill_device) + + @callback + def _handle_coordinator_update(self) -> None: + """Handle updated data from the coordinator.""" + self._update_attr(self.coordinator.data[self._id]) + self.async_write_ha_state() + + @callback + def _update_attr(self, device: MillDevice) -> None: + self._attr_native_value = device.data["deviceSettings"]["reported"].get( + "max_heater_power" + ) + self._available = device.available and self._attr_native_value is not None + + @property + def available(self) -> bool: + """Return True if entity is available.""" + return super().available and self._available + + async def async_set_native_value(self, value: float) -> None: + """Set new value.""" + await self.coordinator.mill_data_connection.max_heating_power(self._id, value) diff --git a/tests/components/mill/test_init.py b/tests/components/mill/test_init.py index bf0026b8c6cb2bb35acc5d937c401fa32f071e51..a47e6422bf85b3461bd8db823ca55fd6f3ec725f 100644 --- a/tests/components/mill/test_init.py +++ b/tests/components/mill/test_init.py @@ -147,5 +147,5 @@ async def test_unload_entry(hass: HomeAssistant) -> None: assert await hass.config_entries.async_unload(entry.entry_id) - assert unload_entry.call_count == 2 + assert unload_entry.call_count == 3 assert entry.entry_id not in hass.data[mill.DOMAIN]