105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
"""Component providing Lights for UniFi Protect."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from pyunifiprotect.data import (
|
|
Light,
|
|
ModelType,
|
|
ProtectAdoptableDeviceModel,
|
|
ProtectModelWithId,
|
|
)
|
|
|
|
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from .const import DISPATCH_ADOPT, DOMAIN
|
|
from .data import ProtectData
|
|
from .entity import ProtectDeviceEntity
|
|
from .utils import async_dispatch_id as _ufpd
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Set up lights for UniFi Protect integration."""
|
|
data: ProtectData = hass.data[DOMAIN][entry.entry_id]
|
|
|
|
@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)])
|
|
|
|
entry.async_on_unload(
|
|
async_dispatcher_connect(hass, _ufpd(entry, DISPATCH_ADOPT), _add_new_device)
|
|
)
|
|
|
|
entities = []
|
|
for device in data.get_by_types({ModelType.LIGHT}):
|
|
if device.can_write(data.api.bootstrap.auth_user):
|
|
entities.append(ProtectLight(data, device))
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
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}
|
|
|
|
@callback
|
|
def _async_get_state_attrs(self) -> tuple[Any, ...]:
|
|
"""Retrieve data that goes into the current state of the entity.
|
|
|
|
Called before and after updating entity and state is only written if there
|
|
is a change.
|
|
"""
|
|
|
|
return (self._attr_available, self._attr_is_on, self._attr_brightness)
|
|
|
|
@callback
|
|
def _async_update_device_from_protect(self, device: ProtectModelWithId) -> 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 def async_turn_on(self, **kwargs: Any) -> None:
|
|
"""Turn the light on."""
|
|
hass_brightness = kwargs.get(ATTR_BRIGHTNESS, self.brightness)
|
|
unifi_brightness = hass_to_unifi_brightness(hass_brightness)
|
|
|
|
_LOGGER.debug("Turning on light with brightness %s", unifi_brightness)
|
|
await self.device.set_light(True, unifi_brightness)
|
|
|
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
|
"""Turn the light off."""
|
|
_LOGGER.debug("Turning off light")
|
|
await self.device.set_light(False)
|