diff --git a/smarttub/api.py b/smarttub/api.py index 490e8c7..bfe426c 100644 --- a/smarttub/api.py +++ b/smarttub/api.py @@ -465,6 +465,10 @@ def __init__(self, spa: Spa, **properties): self.intensity = properties["intensity"] self.mode = self.LightMode[properties["mode"]] + try: + self.cycleSpeed = properties["cycleSpeed"] + except KeyError as e: + logger.debug(f'Benign: {e}') self.properties = properties async def set_mode(self, mode: LightMode, intensity: int): @@ -480,7 +484,7 @@ async def turn_off(self): await self.set_mode(self.LightMode.OFF, 0) def __str__(self): - return f"" + return f"" class SpaReminder: diff --git a/tests/test_light.py b/tests/test_light.py index 55ef780..082dafb 100644 --- a/tests/test_light.py +++ b/tests/test_light.py @@ -14,6 +14,9 @@ def lights(mock_spa): "color": {"blue": 0, "green": 0, "red": 0, "white": 0}, "intensity": 0 if mode == SpaLight.LightMode.OFF else 50, "mode": mode.name, + "cycleSpeed": 0, + "exterior": False, + "irt": None, "zone": i + 1, }, ) diff --git a/tests/test_spa.py b/tests/test_spa.py index d241ac6..0d01bd7 100644 --- a/tests/test_spa.py +++ b/tests/test_spa.py @@ -381,6 +381,7 @@ async def test_get_lights(mock_api, spa): { "color": {"blue": 0, "green": 0, "red": 0, "white": 0}, "intensity": 0, + "cycleSpeed": 0, "mode": "OFF", "zone": 1, }