Use entity class attributes for blinkt (#52893)

* Use entity class attributes for blinkt

* tweak

* tweak

* remove redundant properties
This commit is contained in:
Robert Hillis 2021-07-18 17:21:40 -04:00 committed by GitHub
parent 73976d2a26
commit c5fe01a466
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -41,77 +41,43 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
class BlinktLight(LightEntity):
"""Representation of a Blinkt! Light."""
_attr_supported_features = SUPPORT_BLINKT
_attr_should_poll = False
_attr_assumed_state = True
def __init__(self, blinkt, name, index):
"""Initialize a Blinkt Light.
Default brightness and white color.
"""
self._blinkt = blinkt
self._name = f"{name}_{index}"
self._attr_name = f"{name}_{index}"
self._index = index
self._is_on = False
self._brightness = 255
self._hs_color = [0, 0]
@property
def name(self):
"""Return the display name of this light."""
return self._name
@property
def brightness(self):
"""Read back the brightness of the light.
Returns integer in the range of 1-255.
"""
return self._brightness
@property
def hs_color(self):
"""Read back the color of the light."""
return self._hs_color
@property
def supported_features(self):
"""Flag supported features."""
return SUPPORT_BLINKT
@property
def is_on(self):
"""Return true if light is on."""
return self._is_on
@property
def should_poll(self):
"""Return if we should poll this device."""
return False
@property
def assumed_state(self) -> bool:
"""Return True if unable to access real state of the entity."""
return True
self._attr_is_on = False
self._attr_brightness = 255
self._attr_hs_color = [0, 0]
def turn_on(self, **kwargs):
"""Instruct the light to turn on and set correct brightness & color."""
if ATTR_HS_COLOR in kwargs:
self._hs_color = kwargs[ATTR_HS_COLOR]
self._attr_hs_color = kwargs[ATTR_HS_COLOR]
if ATTR_BRIGHTNESS in kwargs:
self._brightness = kwargs[ATTR_BRIGHTNESS]
self._attr_brightness = kwargs[ATTR_BRIGHTNESS]
percent_bright = self._brightness / 255
rgb_color = color_util.color_hs_to_RGB(*self._hs_color)
percent_bright = self.brightness / 255
rgb_color = color_util.color_hs_to_RGB(*self.hs_color)
self._blinkt.set_pixel(
self._index, rgb_color[0], rgb_color[1], rgb_color[2], percent_bright
)
self._blinkt.show()
self._is_on = True
self._attr_is_on = True
self.schedule_update_ha_state()
def turn_off(self, **kwargs):
"""Instruct the light to turn off."""
self._blinkt.set_pixel(self._index, 0, 0, 0, 0)
self._blinkt.show()
self._is_on = False
self._attr_is_on = False
self.schedule_update_ha_state()