from __future__ import annotations import logging from abc import abstractmethod from typing import Any from .device import ColorMode, SwitchbotDevice _LOGGER = logging.getLogger(__name__) import asyncio import time from ..models import SwitchBotAdvertisement class SwitchbotBaseLight(SwitchbotDevice): """Representation of a Switchbot light.""" def __init__(self, *args: Any, **kwargs: Any) -> None: """Switchbot bulb constructor.""" super().__init__(*args, **kwargs) self._state: dict[str, Any] = {} @property def on(self) -> bool | None: """Return if bulb is on.""" return self.is_on() @property def rgb(self) -> tuple[int, int, int] | None: """Return the current rgb value.""" if "r" not in self._state or "g" not in self._state or "b" not in self._state: return None return self._state["r"], self._state["g"], self._state["b"] @property def color_temp(self) -> int | None: """Return the current color temp value.""" return self._state.get("cw") or self.min_temp @property def brightness(self) -> int | None: """Return the current brightness value.""" return self._get_adv_value("brightness") or 0 @property def color_mode(self) -> ColorMode: """Return the current color mode.""" return ColorMode(self._get_adv_value("color_mode") or 0) @property def min_temp(self) -> int: """Return minimum color temp.""" return 2700 @property def max_temp(self) -> int: """Return maximum color temp.""" return 6500 def is_on(self) -> bool | None: """Return bulb state from cache.""" return self._get_adv_value("isOn") @abstractmethod async def turn_on(self) -> bool: """Turn device on.""" @abstractmethod async def turn_off(self) -> bool: """Turn device off.""" @abstractmethod async def set_brightness(self, brightness: int) -> bool: """Set brightness.""" @abstractmethod async def set_color_temp(self, brightness: int, color_temp: int) -> bool: """Set color temp.""" @abstractmethod async def set_rgb(self, brightness: int, r: int, g: int, b: int) -> bool: """Set rgb.""" def poll_needed(self, last_poll_time: float | None) -> bool: """Return if poll is needed.""" return False async def update(self) -> None: """Update device data.""" self._last_full_update = time.monotonic() class SwitchbotSequenceBaseLight(SwitchbotBaseLight): """Representation of a Switchbot light.""" def update_from_advertisement(self, advertisement: SwitchBotAdvertisement) -> None: """Update device data from advertisement.""" current_state = self._get_adv_value("sequence_number") super().update_from_advertisement(advertisement) new_state = self._get_adv_value("sequence_number") _LOGGER.debug( "%s: update advertisement: %s (seq before: %s) (seq after: %s)", self.name, advertisement, current_state, new_state, ) if current_state != new_state: asyncio.ensure_future(self.update())