114 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Python
		
	
	
			
		
		
	
	
			114 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Python
		
	
	
"""Support for Axis lights."""
 | 
						|
from typing import Any
 | 
						|
 | 
						|
from axis.models.event import Event, EventOperation, EventTopic
 | 
						|
 | 
						|
from homeassistant.components.light import ATTR_BRIGHTNESS, ColorMode, LightEntity
 | 
						|
from homeassistant.config_entries import ConfigEntry
 | 
						|
from homeassistant.core import HomeAssistant, callback
 | 
						|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 | 
						|
 | 
						|
from .const import DOMAIN as AXIS_DOMAIN
 | 
						|
from .device import AxisNetworkDevice
 | 
						|
from .entity import AxisEventEntity
 | 
						|
 | 
						|
 | 
						|
async def async_setup_entry(
 | 
						|
    hass: HomeAssistant,
 | 
						|
    config_entry: ConfigEntry,
 | 
						|
    async_add_entities: AddEntitiesCallback,
 | 
						|
) -> None:
 | 
						|
    """Set up a Axis light."""
 | 
						|
    device: AxisNetworkDevice = hass.data[AXIS_DOMAIN][config_entry.entry_id]
 | 
						|
 | 
						|
    if (
 | 
						|
        device.api.vapix.light_control is None
 | 
						|
        or len(device.api.vapix.light_control) == 0
 | 
						|
    ):
 | 
						|
        return
 | 
						|
 | 
						|
    @callback
 | 
						|
    def async_create_entity(event: Event) -> None:
 | 
						|
        """Create Axis light entity."""
 | 
						|
        async_add_entities([AxisLight(event, device)])
 | 
						|
 | 
						|
    device.api.event.subscribe(
 | 
						|
        async_create_entity,
 | 
						|
        topic_filter=EventTopic.LIGHT_STATUS,
 | 
						|
        operation_filter=EventOperation.INITIALIZED,
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class AxisLight(AxisEventEntity, LightEntity):
 | 
						|
    """Representation of a light Axis event."""
 | 
						|
 | 
						|
    _attr_should_poll = True
 | 
						|
 | 
						|
    def __init__(self, event: Event, device: AxisNetworkDevice) -> None:
 | 
						|
        """Initialize the Axis light."""
 | 
						|
        super().__init__(event, device)
 | 
						|
 | 
						|
        self._light_id = f"led{event.id}"
 | 
						|
 | 
						|
        self.current_intensity = 0
 | 
						|
        self.max_intensity = 0
 | 
						|
 | 
						|
        light_type = device.api.vapix.light_control[self._light_id].light_type
 | 
						|
        self._attr_name = f"{light_type} {self._event_type} {event.id}"
 | 
						|
        self._attr_is_on = event.is_tripped
 | 
						|
 | 
						|
        self._attr_supported_color_modes = {ColorMode.BRIGHTNESS}
 | 
						|
        self._attr_color_mode = ColorMode.BRIGHTNESS
 | 
						|
 | 
						|
    async def async_added_to_hass(self) -> None:
 | 
						|
        """Subscribe lights events."""
 | 
						|
        await super().async_added_to_hass()
 | 
						|
 | 
						|
        current_intensity = (
 | 
						|
            await self.device.api.vapix.light_control.get_current_intensity(
 | 
						|
                self._light_id
 | 
						|
            )
 | 
						|
        )
 | 
						|
        self.current_intensity = current_intensity["data"]["intensity"]
 | 
						|
 | 
						|
        max_intensity = await self.device.api.vapix.light_control.get_valid_intensity(
 | 
						|
            self._light_id
 | 
						|
        )
 | 
						|
        self.max_intensity = max_intensity["data"]["ranges"][0]["high"]
 | 
						|
 | 
						|
    @callback
 | 
						|
    def async_event_callback(self, event: Event) -> None:
 | 
						|
        """Update light state."""
 | 
						|
        self._attr_is_on = event.is_tripped
 | 
						|
        self.async_write_ha_state()
 | 
						|
 | 
						|
    @property
 | 
						|
    def brightness(self) -> int:
 | 
						|
        """Return the brightness of this light between 0..255."""
 | 
						|
        return int((self.current_intensity / self.max_intensity) * 255)
 | 
						|
 | 
						|
    async def async_turn_on(self, **kwargs: Any) -> None:
 | 
						|
        """Turn on light."""
 | 
						|
        if not self.is_on:
 | 
						|
            await self.device.api.vapix.light_control.activate_light(self._light_id)
 | 
						|
 | 
						|
        if ATTR_BRIGHTNESS in kwargs:
 | 
						|
            intensity = int((kwargs[ATTR_BRIGHTNESS] / 255) * self.max_intensity)
 | 
						|
            await self.device.api.vapix.light_control.set_manual_intensity(
 | 
						|
                self._light_id, intensity
 | 
						|
            )
 | 
						|
 | 
						|
    async def async_turn_off(self, **kwargs: Any) -> None:
 | 
						|
        """Turn off light."""
 | 
						|
        if self.is_on:
 | 
						|
            await self.device.api.vapix.light_control.deactivate_light(self._light_id)
 | 
						|
 | 
						|
    async def async_update(self) -> None:
 | 
						|
        """Update brightness."""
 | 
						|
        current_intensity = (
 | 
						|
            await self.device.api.vapix.light_control.get_current_intensity(
 | 
						|
                self._light_id
 | 
						|
            )
 | 
						|
        )
 | 
						|
        self.current_intensity = current_intensity["data"]["intensity"]
 |