From 97637ef244e90fa6a75d17f5017823970c10e33d Mon Sep 17 00:00:00 2001 From: Andre Basche Date: Fri, 30 Jun 2023 19:36:36 +0200 Subject: [PATCH] Add light entity for lights --- custom_components/hon/const.py | 1 + custom_components/hon/fan.py | 5 +- custom_components/hon/light.py | 109 ++++++++++++++++++++++++++++++++ custom_components/hon/number.py | 12 ---- custom_components/hon/switch.py | 3 - 5 files changed, 113 insertions(+), 17 deletions(-) create mode 100644 custom_components/hon/light.py diff --git a/custom_components/hon/const.py b/custom_components/hon/const.py index 48f7fba..d43308d 100644 --- a/custom_components/hon/const.py +++ b/custom_components/hon/const.py @@ -18,6 +18,7 @@ PLATFORMS = [ "binary_sensor", "climate", "fan", + "light", ] HON_HVAC_MODE = { diff --git a/custom_components/hon/fan.py b/custom_components/hon/fan.py index f4ce402..35f3536 100644 --- a/custom_components/hon/fan.py +++ b/custom_components/hon/fan.py @@ -44,8 +44,9 @@ async def async_setup_entry(hass, entry: ConfigEntry, async_add_entities) -> Non for device in hass.data[DOMAIN][entry.unique_id].appliances: for description in FANS.get(device.appliance_type, []): if isinstance(description, HonFanEntityDescription): - if description.key not in device.available_settings or device.get( - description.key.split(".")[-1] is None + if ( + description.key not in device.available_settings + or device.get(description.key.split(".")[-1]) is None ): continue entity = HonFanEntity(hass, entry, device, description) diff --git a/custom_components/hon/light.py b/custom_components/hon/light.py new file mode 100644 index 0000000..d6be0ae --- /dev/null +++ b/custom_components/hon/light.py @@ -0,0 +1,109 @@ +import logging +from typing import Any + +from homeassistant.components.light import ( + LightEntityDescription, + LightEntity, + ColorMode, + ATTR_BRIGHTNESS, +) +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import callback +from pyhon.appliance import HonAppliance +from pyhon.parameter.range import HonParameterRange + +from .const import DOMAIN +from .hon import HonEntity + +_LOGGER = logging.getLogger(__name__) + + +FANS = { + "WC": (LightEntityDescription(key="settings.lightStatus", name="Light"),), + "HO": ( + LightEntityDescription( + key="startProgram.lightStatus", + name="Light status", + ), + ), + "AP": (LightEntityDescription(key="settings.lightStatus", name="Light status"),), +} + + +async def async_setup_entry(hass, entry: ConfigEntry, async_add_entities) -> None: + entities = [] + for device in hass.data[DOMAIN][entry.unique_id].appliances: + for description in FANS.get(device.appliance_type, []): + if ( + description.key not in device.available_settings + or device.get(description.key.split(".")[-1]) is None + ): + continue + entity = HonLightEntity(hass, entry, device, description) + await entity.coordinator.async_config_entry_first_refresh() + entities.append(entity) + async_add_entities(entities) + + +class HonLightEntity(HonEntity, LightEntity): + entity_description: LightEntityDescription + + def __init__(self, hass, entry, device: HonAppliance, description) -> None: + light: HonParameterRange = device.settings.get(description.key) + self._light_range = (light.min, light.max) + self._attr_supported_color_modes: set[ColorMode] = set() + if len(light.values) == 2: + self._attr_supported_color_modes.add(ColorMode.ONOFF) + else: + self._attr_supported_color_modes.add(ColorMode.BRIGHTNESS) + self._command, self._parameter = description.key.split(".") + super().__init__(hass, entry, device, description) + self._handle_coordinator_update(update=False) + + @property + def is_on(self) -> bool: + """Return true if light is on.""" + light = self._device.settings.get(self.entity_description.key) + return light.value != light.min + + async def async_turn_on(self, **kwargs: Any) -> None: + """Turn on or control the light.""" + light: HonParameterRange = self._device.settings.get( + self.entity_description.key + ) + if ColorMode.BRIGHTNESS in self._attr_supported_color_modes: + percent = int(100 / 255 * kwargs.get(ATTR_BRIGHTNESS, 128)) + light.value = round(light.max / 100 * percent) + if light.value == light.min: + self._attr_is_on = False + self._attr_brightness = self.brightness + else: + light.value = light.max + await self._device.commands[self._command].send() + self.async_write_ha_state() + + async def async_turn_off(self, **kwargs: Any) -> None: + """Instruct the light to turn off.""" + light: HonParameterRange = self._device.settings.get( + self.entity_description.key + ) + light.value = light.min + await self._device.commands[self._command].send() + self.async_write_ha_state() + + @property + def brightness(self) -> int | None: + """Return the brightness of the light.""" + light: HonParameterRange = self._device.settings.get( + self.entity_description.key + ) + if light.value == light.min: + return None + return int(255 / light.max * light.value) + + @callback + def _handle_coordinator_update(self, update=True) -> None: + self._attr_is_on = self.is_on + self._attr_brightness = self.brightness + if update: + self.async_write_ha_state() diff --git a/custom_components/hon/number.py b/custom_components/hon/number.py index 3bbad56..60081e9 100644 --- a/custom_components/hon/number.py +++ b/custom_components/hon/number.py @@ -162,13 +162,6 @@ NUMBERS: dict[str, tuple[NumberEntityDescription, ...]] = { translation_key="freezer_temp_sel", ), ), - "HO": ( - HonNumberEntityDescription( - key="startProgram.lightStatus", - name="Light status", - icon="mdi:lightbulb", - ), - ), "AP": ( HonNumberEntityDescription( key="settings.aromaTimeOn", @@ -182,11 +175,6 @@ NUMBERS: dict[str, tuple[NumberEntityDescription, ...]] = { icon="mdi:thermometer", native_unit_of_measurement=UnitOfTime.SECONDS, ), - HonNumberEntityDescription( - key="settings.lightStatus", - name="Light status", - icon="mdi:lightbulb", - ), HonNumberEntityDescription( key="settings.pollenLevel", name="Pollen Level", diff --git a/custom_components/hon/switch.py b/custom_components/hon/switch.py index a77aacd..3b8e059 100644 --- a/custom_components/hon/switch.py +++ b/custom_components/hon/switch.py @@ -329,9 +329,6 @@ SWITCHES: dict[str, tuple[HonSwitchEntityDescription, ...]] = { icon="mdi:palm-tree", translation_key="holiday_mode", ), - HonSwitchEntityDescription( - key="lightStatus", name="Light", icon="mdi:lightbulb" - ), ), "AP": ( HonSwitchEntityDescription(