hOn/custom_components/hon/light.py

110 lines
3.8 KiB
Python
Raw Normal View History

2023-06-30 17:36:36 +00:00
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__)
2023-06-30 17:40:30 +00:00
LIGHTS = {
2023-06-30 17:36:36 +00:00
"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:
2023-06-30 17:40:30 +00:00
for description in LIGHTS.get(device.appliance_type, []):
2023-06-30 17:36:36 +00:00
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()