mirror of
https://github.com/home-assistant/core.git
synced 2026-09-24 15:31:52 -05:00
120 lines
4.1 KiB
Python
120 lines
4.1 KiB
Python
"""Component providing Lights for UniFi Protect."""
|
|
|
|
import logging
|
|
from typing import Any, cast, override
|
|
|
|
from uiprotect.data import Light, ModelType, ProtectAdoptableDeviceModel
|
|
from uiprotect.data.devices import LightDeviceSettings
|
|
from uiprotect.data.public_devices import PublicLight
|
|
|
|
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")
|
|
# State comes from the public API; the base class primes the object and
|
|
# subscribes to the public devices websocket on this flag.
|
|
_ufp_uses_public = True
|
|
|
|
@callback
|
|
@override
|
|
def _async_update_device_from_protect(self, device: ProtectDeviceType) -> None:
|
|
super()._async_update_device_from_protect(device)
|
|
if (public := self._ufp_public_obj) is None:
|
|
return
|
|
light = cast(PublicLight, public)
|
|
self._attr_is_on = light.is_light_on
|
|
led_level = light.light_device_settings.led_level
|
|
self._attr_brightness = (
|
|
None if led_level is None else unifi_brightness_to_hass(led_level)
|
|
)
|
|
|
|
@async_ufp_instance_command
|
|
@override
|
|
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
|
|
@override
|
|
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
|
|
)
|