hass_flickerstrip/custom_components/flickerstrip/common.py

59 lines
1.5 KiB
Python
Raw Normal View History

2023-01-29 13:29:34 -05:00
from typing import Any
import flickerstrip_py as flstrp
from homeassistant.core import HomeAssistant
from homeassistant.components.light import (
ATTR_BRIGHTNESS,
LightEntity,
)
from .const import (
DEFAULT_PORT,
)
class Flickerstrip(LightEntity):
"""Flickerstrip class."""
def __init__(
self,
hass: HomeAssistant,
host: str,
port: int = DEFAULT_PORT,
):
self.hass = hass
self.host = host
self.port = port
self._flickerstrip = flstrp.Flickerstrip(host)
@property
def name(self) -> str:
"""Return the display name of this light."""
return self._name
@property
def brightness(self):
"""Return the brightness of the light."""
return self._flickerstrip.status.brightness
@property
def is_on(self) -> bool | None:
"""Return true if light is on."""
return self._flickerstrip.status.power
def turn_on(self, **kwargs: Any) -> None:
"""Instruct the light to turn on."""
brightness = kwargs.get(ATTR_BRIGHTNESS, 100)
self._flickerstrip.set_brightness(brightness)
self._flickerstrip.power_on()
def turn_off(self, **kwargs: Any) -> None:
"""Instruct the light to turn off."""
self._flickerstrip.power_off()
def update(self) -> None:
"""Fetch new state data for this light.
This is the only method that should fetch new data for Home Assistant.
"""
self._flickerstrip.force_update()