mirror of
https://github.com/python-kasa/python-kasa.git
synced 2024-12-23 11:43:34 +00:00
7f24408c32
Fixes issue with unpaired devices reporting light effect as `softAP` reported in https://github.com/python-kasa/python-kasa/pull/972. I don't think we need to handle that effect properly so just reports as off.
146 lines
4.3 KiB
Python
146 lines
4.3 KiB
Python
"""Module for light effects."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from ...interfaces.lighteffect import LightEffect as LightEffectInterface
|
|
from ..effects import EFFECT_MAPPING, EFFECT_NAMES
|
|
from ..smartmodule import SmartModule
|
|
|
|
if TYPE_CHECKING:
|
|
from ..smartdevice import SmartDevice
|
|
|
|
|
|
class LightStripEffect(SmartModule, LightEffectInterface):
|
|
"""Implementation of dynamic light effects."""
|
|
|
|
REQUIRED_COMPONENT = "light_strip_lighting_effect"
|
|
|
|
def __init__(self, device: SmartDevice, module: str):
|
|
super().__init__(device, module)
|
|
effect_list = [self.LIGHT_EFFECTS_OFF]
|
|
effect_list.extend(EFFECT_NAMES)
|
|
self._effect_list = effect_list
|
|
|
|
@property
|
|
def name(self) -> str:
|
|
"""Name of the module.
|
|
|
|
By default smart modules are keyed in the module mapping by class name.
|
|
The name is overriden here as this module implements the same common interface
|
|
as the bulb light_effect and the assumption is a device only supports one
|
|
or the other.
|
|
|
|
"""
|
|
return "LightEffect"
|
|
|
|
@property
|
|
def effect(self) -> str:
|
|
"""Return effect state.
|
|
|
|
Example:
|
|
{'brightness': 50,
|
|
'custom': 0,
|
|
'enable': 0,
|
|
'id': '',
|
|
'name': ''}
|
|
"""
|
|
eff = self.data["lighting_effect"]
|
|
name = eff["name"]
|
|
# When devices are unpaired effect name is softAP which is not in our list
|
|
if eff["enable"] and name in self._effect_list:
|
|
return name
|
|
return self.LIGHT_EFFECTS_OFF
|
|
|
|
@property
|
|
def effect_list(self) -> list[str]:
|
|
"""Return built-in effects list.
|
|
|
|
Example:
|
|
['Aurora', 'Bubbling Cauldron', ...]
|
|
"""
|
|
return self._effect_list
|
|
|
|
async def set_effect(
|
|
self,
|
|
effect: str,
|
|
*,
|
|
brightness: int | None = None,
|
|
transition: int | None = None,
|
|
) -> None:
|
|
"""Set an effect on the device.
|
|
|
|
If brightness or transition is defined,
|
|
its value will be used instead of the effect-specific default.
|
|
|
|
See :meth:`effect_list` for available effects,
|
|
or use :meth:`set_custom_effect` for custom effects.
|
|
|
|
:param str effect: The effect to set
|
|
:param int brightness: The wanted brightness
|
|
:param int transition: The wanted transition time
|
|
"""
|
|
if effect == self.LIGHT_EFFECTS_OFF:
|
|
effect_dict = dict(self.data["lighting_effect"])
|
|
effect_dict["enable"] = 0
|
|
elif effect not in EFFECT_MAPPING:
|
|
raise ValueError(f"The effect {effect} is not a built in effect.")
|
|
else:
|
|
effect_dict = EFFECT_MAPPING[effect]
|
|
|
|
if brightness is not None:
|
|
effect_dict["brightness"] = brightness
|
|
if transition is not None:
|
|
effect_dict["transition"] = transition
|
|
|
|
await self.set_custom_effect(effect_dict)
|
|
|
|
async def set_custom_effect(
|
|
self,
|
|
effect_dict: dict,
|
|
) -> None:
|
|
"""Set a custom effect on the device.
|
|
|
|
:param str effect_dict: The custom effect dict to set
|
|
"""
|
|
return await self.call(
|
|
"set_lighting_effect",
|
|
effect_dict,
|
|
)
|
|
|
|
@property
|
|
def has_custom_effects(self) -> bool:
|
|
"""Return True if the device supports setting custom effects."""
|
|
return True
|
|
|
|
def query(self):
|
|
"""Return the base query."""
|
|
return {}
|
|
|
|
@property # type: ignore
|
|
def _deprecated_effect(self) -> dict:
|
|
"""Return effect state.
|
|
|
|
Example:
|
|
{'brightness': 50,
|
|
'custom': 0,
|
|
'enable': 0,
|
|
'id': '',
|
|
'name': ''}
|
|
"""
|
|
# LightEffectModule returns the current effect name
|
|
# so return the dict here for backwards compatibility
|
|
return self.data["lighting_effect"]
|
|
|
|
@property # type: ignore
|
|
def _deprecated_effect_list(self) -> list[str] | None:
|
|
"""Return built-in effects list.
|
|
|
|
Example:
|
|
['Aurora', 'Bubbling Cauldron', ...]
|
|
"""
|
|
# LightEffectModule returns effect names along with a LIGHT_EFFECTS_OFF value
|
|
# so return the original effect names here for backwards compatibility
|
|
return EFFECT_NAMES
|