jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add decorator typing [limitlessled] (#107557)

Browse Source
This commit is contained in:
Marc Mueller 2024-01-12 11:30:07 +01:00 committed by GitHub
parent 668fc442e9
commit c9befe8700
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
homeassistant/components/limitlessled/light.py
View File

@ -1,8 +1,9 @@
"""Support for LimitlessLED bulbs.""" """Support for LimitlessLED bulbs."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Callable
import logging import logging
from typing import Any from typing import Any, Concatenate, ParamSpec, TypeVar, cast
from limitlessled import Color from limitlessled import Color
from limitlessled.bridge import Bridge from limitlessled.bridge import Bridge
@ -38,6 +39,9 @@ from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util.color import color_hs_to_RGB, color_temperature_mired_to_kelvin from homeassistant.util.color import color_hs_to_RGB, color_temperature_mired_to_kelvin
_LimitlessLEDGroupT = TypeVar("_LimitlessLEDGroupT", bound="LimitlessLEDGroup")
_P = ParamSpec("_P")
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
CONF_BRIDGES = "bridges" CONF_BRIDGES = "bridges"
@ -171,16 +175,25 @@ def setup_platform(
add_entities(lights) add_entities(lights)
def state(new_state): def state(
new_state: bool,
) -> Callable[
[Callable[Concatenate[_LimitlessLEDGroupT, int, Pipeline, _P], Any]],
Callable[Concatenate[_LimitlessLEDGroupT, _P], None],
]:
"""State decorator. """State decorator.
Specify True (turn on) or False (turn off). Specify True (turn on) or False (turn off).
""" """
def decorator(function): def decorator(
function: Callable[Concatenate[_LimitlessLEDGroupT, int, Pipeline, _P], Any],
) -> Callable[Concatenate[_LimitlessLEDGroupT, _P], None]:
"""Set up the decorator function.""" """Set up the decorator function."""
def wrapper(self: LimitlessLEDGroup, **kwargs: Any) -> None: def wrapper(
self: _LimitlessLEDGroupT, *args: _P.args, **kwargs: _P.kwargs
) -> None:
"""Wrap a group state change.""" """Wrap a group state change."""
pipeline = Pipeline() pipeline = Pipeline()
transition_time = DEFAULT_TRANSITION transition_time = DEFAULT_TRANSITION
@ -189,9 +202,9 @@ def state(new_state):
self._attr_effect = None # pylint: disable=protected-access self._attr_effect = None # pylint: disable=protected-access
# Set transition time. # Set transition time.
if ATTR_TRANSITION in kwargs: if ATTR_TRANSITION in kwargs:
transition_time = int(kwargs[ATTR_TRANSITION]) transition_time = int(cast(float, kwargs[ATTR_TRANSITION]))
# Do group type-specific work. # Do group type-specific work.
function(self, transition_time, pipeline, **kwargs) function(self, transition_time, pipeline, *args, **kwargs)
# Update state. # Update state.
self._attr_is_on = new_state # pylint: disable=protected-access self._attr_is_on = new_state # pylint: disable=protected-access
self.group.enqueue(pipeline) self.group.enqueue(pipeline)