mirror of
https://github.com/home-assistant/core.git
synced 2025-12-20 02:48:57 +00:00
112 lines
3.8 KiB
Python
112 lines
3.8 KiB
Python
"""Component providing Lights for UniFi Protect."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from uiprotect.data import Light, ModelType, ProtectAdoptableDeviceModel
|
|
from uiprotect.data.devices import LightDeviceSettings
|
|
|
|
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from .data import ProtectDeviceType, UFPConfigEntry
|
|
from .entity import ProtectDeviceEntity
|
|
from .utils import async_ufp_instance_command
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
PARALLEL_UPDATES = 0
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: UFPConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up lights for UniFi Protect integration."""
|
|
data = entry.runtime_data
|
|
|
|
@callback
|
|
def _add_new_device(device: ProtectAdoptableDeviceModel) -> None:
|
|
if device.model is ModelType.LIGHT and device.can_write(
|
|
data.api.bootstrap.auth_user
|
|
):
|
|
async_add_entities([ProtectLight(data, device)])
|
|
|
|
data.async_subscribe_adopt(_add_new_device)
|
|
async_add_entities(
|
|
ProtectLight(data, device)
|
|
for device in data.get_by_types({ModelType.LIGHT})
|
|
if device.can_write(data.api.bootstrap.auth_user)
|
|
)
|
|
|
|
|
|
def unifi_brightness_to_hass(value: int) -> int:
|
|
"""Convert unifi brightness 1..6 to hass format 0..255."""
|
|
return min(255, round((value / 6) * 255))
|
|
|
|
|
|
def hass_to_unifi_brightness(value: int) -> int:
|
|
"""Convert hass brightness 0..255 to unifi 1..6 scale."""
|
|
return max(1, round((value / 255) * 6))
|
|
|
|
|
|
class ProtectLight(ProtectDeviceEntity, LightEntity):
|
|
"""A Ubiquiti UniFi Protect Light Entity."""
|
|
|
|
device: Light
|
|
|
|
_attr_icon = "mdi:spotlight-beam"
|
|
_attr_color_mode = ColorMode.BRIGHTNESS
|
|
_attr_supported_color_modes = {ColorMode.BRIGHTNESS}
|
|
_state_attrs = ("_attr_available", "_attr_is_on", "_attr_brightness")
|
|
|
|
@callback
|
|
def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None:
|
|
super()._async_update_device_from_protect(device)
|
|
updated_device = self.device
|
|
self._attr_is_on = updated_device.is_light_on
|
|
self._attr_brightness = unifi_brightness_to_hass(
|
|
updated_device.light_device_settings.led_level
|
|
)
|
|
|
|
@async_ufp_instance_command
|
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn the light on."""
|
|
brightness = kwargs.get(ATTR_BRIGHTNESS)
|
|
led_level: int | None = None
|
|
if brightness is not None:
|
|
led_level = hass_to_unifi_brightness(brightness)
|
|
_LOGGER.debug(
|
|
"Turning on light with brightness %s (led_level=%s)",
|
|
brightness,
|
|
led_level,
|
|
)
|
|
else:
|
|
_LOGGER.debug("Turning on light")
|
|
|
|
await self.device.api.update_light_public(
|
|
self.device.id,
|
|
is_light_force_enabled=True,
|
|
light_device_settings=(
|
|
LightDeviceSettings(
|
|
is_indicator_enabled=self.device.light_device_settings.is_indicator_enabled,
|
|
led_level=led_level,
|
|
pir_duration=self.device.light_device_settings.pir_duration,
|
|
pir_sensitivity=self.device.light_device_settings.pir_sensitivity,
|
|
)
|
|
if led_level is not None
|
|
else None
|
|
),
|
|
)
|
|
|
|
@async_ufp_instance_command
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn the light off."""
|
|
_LOGGER.debug("Turning off light")
|
|
await self.device.api.update_light_public(
|
|
self.device.id, is_light_force_enabled=False
|
|
)
|