From cc0ac1e4baf74e835682eabb6a3a818904ba9606 Mon Sep 17 00:00:00 2001
From: Aarno Aukia <aarno.aukia@vshn.ch>
Date: Wed, 13 Dec 2023 20:06:27 +0100
Subject: [PATCH] add support for chromazon3 light setting, fixes #37

---
 controlmyspa.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/controlmyspa.py b/controlmyspa.py
index 0c7e8f2..b98fd38 100644
--- a/controlmyspa.py
+++ b/controlmyspa.py
@@ -440,3 +440,23 @@ def online(self):
         Get the spa online status
         """
         return self._info["online"]
+
+    def set_chromazon3(self, state=False):
+        """
+        Enable/disable Chromeazon3 lights
+        https://www.balboawatergroup.com/Chromazon3
+        :param state: False to turn off, True to turn on
+        """
+        response = requests.post(
+            "https://iot.controlmyspa.com/mobile/control/"
+            + self._info["_id"]
+            + "/tzl/setPower",
+            json={
+                "desiredState": ("ON" if state else "OFF"),
+            },
+            headers={"Authorization": "Bearer " + self._token["access_token"]},
+            timeout=10,
+        )
+        response.raise_for_status()
+        # update the local info
+        self._get_info()