Skip to content

Commit

Permalink
Add light entity for lights
Browse files Browse the repository at this point in the history
  • Loading branch information
Andre0512 committed Jun 30, 2023
1 parent 1d83162 commit 97637ef
Show file tree
Hide file tree
Showing 5 changed files with 113 additions and 17 deletions.
1 change: 1 addition & 0 deletions custom_components/hon/const.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
"binary_sensor",
"climate",
"fan",
"light",
]

HON_HVAC_MODE = {
Expand Down
5 changes: 3 additions & 2 deletions custom_components/hon/fan.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
109 changes: 109 additions & 0 deletions custom_components/hon/light.py
Original file line number Diff line number Diff line change
@@ -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()
12 changes: 0 additions & 12 deletions custom_components/hon/number.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,13 +162,6 @@ class HonNumberEntityDescription(NumberEntityDescription):
translation_key="freezer_temp_sel",
),
),
"HO": (
HonNumberEntityDescription(
key="startProgram.lightStatus",
name="Light status",
icon="mdi:lightbulb",
),
),
"AP": (
HonNumberEntityDescription(
key="settings.aromaTimeOn",
Expand All @@ -182,11 +175,6 @@ class HonNumberEntityDescription(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",
Expand Down
3 changes: 0 additions & 3 deletions custom_components/hon/switch.py
Original file line number Diff line number Diff line change
Expand Up @@ -329,9 +329,6 @@ class HonConfigSwitchEntityDescription(SwitchEntityDescription):
icon="mdi:palm-tree",
translation_key="holiday_mode",
),
HonSwitchEntityDescription(
key="lightStatus", name="Light", icon="mdi:lightbulb"
),
),
"AP": (
HonSwitchEntityDescription(
Expand Down

0 comments on commit 97637ef

Please sign in to comment.