Use attributes in tikteck light (#76022)
This commit is contained in:
parent
a987cad973
commit
fa9d0b9ff7
1 changed files with 18 additions and 49 deletions
|
@ -55,82 +55,51 @@ def setup_platform(
|
|||
class TikteckLight(LightEntity):
|
||||
"""Representation of a Tikteck light."""
|
||||
|
||||
_attr_assumed_state = True
|
||||
_attr_color_mode = ColorMode.HS
|
||||
_attr_should_poll = False
|
||||
_attr_supported_color_modes = {ColorMode.HS}
|
||||
hs_color: tuple[float, float]
|
||||
brightness: int
|
||||
|
||||
def __init__(self, device):
|
||||
"""Initialize the light."""
|
||||
|
||||
self._name = device["name"]
|
||||
self._address = device["address"]
|
||||
self._password = device["password"]
|
||||
self._brightness = 255
|
||||
self._hs = [0, 0]
|
||||
self._state = False
|
||||
address = device["address"]
|
||||
self._attr_unique_id = address
|
||||
self._attr_name = device["name"]
|
||||
self._attr_brightness = 255
|
||||
self._attr_hs_color = [0, 0]
|
||||
self._attr_is_on = False
|
||||
self.is_valid = True
|
||||
self._bulb = tikteck.tikteck(self._address, "Smart Light", self._password)
|
||||
self._bulb = tikteck.tikteck(address, "Smart Light", device["password"])
|
||||
if self._bulb.connect() is False:
|
||||
self.is_valid = False
|
||||
_LOGGER.error("Failed to connect to bulb %s, %s", self._address, self._name)
|
||||
_LOGGER.error("Failed to connect to bulb %s, %s", address, self.name)
|
||||
|
||||
@property
|
||||
def unique_id(self):
|
||||
"""Return the ID of this light."""
|
||||
return self._address
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Return the name of the device if any."""
|
||||
return self._name
|
||||
|
||||
@property
|
||||
def is_on(self):
|
||||
"""Return true if device is on."""
|
||||
return self._state
|
||||
|
||||
@property
|
||||
def brightness(self):
|
||||
"""Return the brightness of this light between 0..255."""
|
||||
return self._brightness
|
||||
|
||||
@property
|
||||
def hs_color(self):
|
||||
"""Return the color property."""
|
||||
return self._hs
|
||||
|
||||
@property
|
||||
def should_poll(self):
|
||||
"""Return the polling state."""
|
||||
return False
|
||||
|
||||
@property
|
||||
def assumed_state(self):
|
||||
"""Return the assumed state."""
|
||||
return True
|
||||
|
||||
def set_state(self, red, green, blue, brightness):
|
||||
def set_state(self, red: int, green: int, blue: int, brightness: int) -> bool:
|
||||
"""Set the bulb state."""
|
||||
return self._bulb.set_state(red, green, blue, brightness)
|
||||
|
||||
def turn_on(self, **kwargs: Any) -> None:
|
||||
"""Turn the specified light on."""
|
||||
self._state = True
|
||||
self._attr_is_on = True
|
||||
|
||||
hs_color = kwargs.get(ATTR_HS_COLOR)
|
||||
brightness = kwargs.get(ATTR_BRIGHTNESS)
|
||||
|
||||
if hs_color is not None:
|
||||
self._hs = hs_color
|
||||
self._attr_hs_color = hs_color
|
||||
if brightness is not None:
|
||||
self._brightness = brightness
|
||||
self._attr_brightness = brightness
|
||||
|
||||
rgb = color_util.color_hs_to_RGB(*self._hs)
|
||||
rgb = color_util.color_hs_to_RGB(self.hs_color[0], self.hs_color[1])
|
||||
|
||||
self.set_state(rgb[0], rgb[1], rgb[2], self.brightness)
|
||||
self.schedule_update_ha_state()
|
||||
|
||||
def turn_off(self, **kwargs: Any) -> None:
|
||||
"""Turn the specified light off."""
|
||||
self._state = False
|
||||
self._attr_is_on = False
|
||||
self.set_state(0, 0, 0, 0)
|
||||
self.schedule_update_ha_state()
|
||||
|
|
Loading…
Add table
Reference in a new issue