bump

custom_components/kittybediq/light.py

@@ -0,0 +1,66 @@
+"""Support for SleepIQ outlet lights."""
+
+import logging
+from typing import Any, override
+
+from asyncsleepiq import SleepIQBed, SleepIQLight
+
+from homeassistant.components.light import ColorMode, LightEntity
+from homeassistant.core import HomeAssistant, callback
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import SleepIQConfigEntry, SleepIQDataUpdateCoordinator
+from .entity import SleepIQBedEntity
+
+_LOGGER = logging.getLogger(__name__)
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: SleepIQConfigEntry,
+    async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+    """Set up the SleepIQ bed lights."""
+    data = entry.runtime_data
+    async_add_entities(
+        SleepIQLightEntity(data.data_coordinator, bed, light)
+        for bed in data.client.beds.values()
+        for light in bed.foundation.lights
+    )
+
+
+class SleepIQLightEntity(SleepIQBedEntity[SleepIQDataUpdateCoordinator], LightEntity):
+    """Representation of a light."""
+
+    _attr_color_mode = ColorMode.ONOFF
+    _attr_supported_color_modes = {ColorMode.ONOFF}
+
+    def __init__(
+        self,
+        coordinator: SleepIQDataUpdateCoordinator,
+        bed: SleepIQBed,
+        light: SleepIQLight,
+    ) -> None:
+        """Initialize the light."""
+        self.light = light
+        super().__init__(coordinator, bed)
+        self._attr_name = f"SleepNumber {bed.name} Light {light.outlet_id}"
+        self._attr_unique_id = f"{bed.id}-light-{light.outlet_id}"  # pylint: disable=home-assistant-entity-unique-id-redundant-platform
+
+    @override
+    async def async_turn_on(self, **kwargs: Any) -> None:
+        """Turn on light."""
+        await self.light.turn_on()
+        self._handle_coordinator_update()
+
+    @override
+    async def async_turn_off(self, **kwargs: Any) -> None:
+        """Turn off light."""
+        await self.light.turn_off()
+        self._handle_coordinator_update()
+
+    @callback
+    @override
+    def _async_update_attrs(self) -> None:
+        """Update light attributes."""
+        self._attr_is_on = self.light.is_on