organize

2026-06-17 23:11:29 -04:00
parent 7c1350ac01
commit d856a4ea40
6 changed files with 172 additions and 169 deletions
@@ -0,0 +1,81 @@
+import requests
+import random
+import string
+import json
+
+GOVEE_API_BASE_URL = "https://openapi.api.govee.com/router/api/v1/"
+
+
+class GoveeApi:
+    key = ""
+    headers = {}
+
+    def __init__(self, key):
+        self.key = key
+        self.headers = {"Govee-API-Key": key}
+
+    def __get_random_string(self):
+        characters = string.ascii_letters + string.digits
+        result_str = "".join(random.choices(characters, k=4))
+        return result_str
+
+    def get_diy_scenes(self, sku, device):
+        payload = {
+            "requestId": self.__get_random_string(),
+            "payload": {"sku": sku, "device": device},
+        }
+
+        res = requests.post(
+            GOVEE_API_BASE_URL + "device/diy-scenes", json=payload, headers=self.headers
+        )
+        res.raise_for_status()
+
+        print("[GOVEE] DIY scene fetch: " + json.dumps(res.json(), indent=4))
+
+        return res.ok
+
+    def set_diy_scene(self, sku, device):
+        payload = {
+            "requestId": self.__get_random_string(),
+            "payload": {
+                "sku": sku,
+                "device": device,
+                "capability": {
+                    "type": "device.capabilities.dynamic_scene",
+                    "instance": "diyScene",
+                    "value": 22757907,
+                },
+            },
+        }
+
+        res = requests.post(
+            GOVEE_API_BASE_URL + "device/control", headers=self.headers, json=payload
+        )
+        res.raise_for_status()
+
+        print("[GOVEE] Set DIY scene: " + json.dumps(res.json(), indent=4))
+
+        return res.ok
+
+    def set_to_original_color(self, sku, device):
+        payload = {
+            "requestId": self.__get_random_string(),
+            "payload": {
+                "sku": sku,
+                "device": device,
+                "capability": {
+                    "type": "devices.capabilities.color_setting",
+                    "instance": "colorRgb",
+                    "value": 16711680,
+                },
+            },
+        }
+
+        res = requests.post(
+            GOVEE_API_BASE_URL + "device/control", json=payload, headers=self.headers
+        )
+        res.raise_for_status()
+
+        print("[GOVEE] Set to original: " + json.dumps(res.json(), indent=4))
+
+        return res.ok