Add Wiz config parsing logic

Author: mmynk

constants.py

@@ -12,4 +12,5 @@
 GREEN = 'green'
 BLUE = 'blue'
 
+NETWORK_NAME = 'wiz_a4f78f'
 IP = '192.168.31.170'

message.py

@@ -6,6 +6,9 @@ class Params:
     def __init__(self, **kwargs):
         self.params = kwargs
 
+    def __getitem__(self, key):
+        return self.params[key]
+
     def __setitem__(self, key, value):
         self.params[key] = value
 
@@ -16,6 +19,7 @@ def __str__(self):
 class Request:
     def __init__(self, method: str, params: Params = None):
         self.method = method
+        self.params = None
         if params:
             self.params = params.params
 
@@ -26,6 +30,16 @@ def __str__(self):
         return json.dumps(self.__dict__)
 
 
+class Config:
+    def __init__(self, state=True, scene=11, red=None, green=None, blue=None, brightness=100):
+        self.state = state
+        self.scene = scene
+        self.red = red
+        self.green = green
+        self.blue = blue
+        self.brightness = brightness
+
+
 if __name__ == '__main__':
     message = Request(GET_PILOT, Params(state=True))
     print(bytes(message).decode('utf-8'))

wiz.py

@@ -1,18 +1,24 @@
+import json
+
 import webcolors
 
-from udp import UDP
 from constants import *
-from message import Params, Request
+from message import Config, Params, Request
+from udp import UDP
 
 
 class Light:
     def __init__(self, ip: str):
         # TODO: get ip from network auto
         self.ip = ip
         self.udp = UDP(ip)
+        self.config = self.get_config()
 
     def send(self, message):
-        self.udp.call(message)
+        result = self.udp.call(message)
+        if message.params:
+            self.update_config(message.params)
+        return result
 
     def on(self):
         params = Params(state=True)
@@ -22,9 +28,14 @@ def on(self):
     def off(self):
         params = Params(state=False)
         message = self.message(params)
-        # print(message)
         self.send(message)
 
+    def switch(self):
+        if self.config.state:
+            self.off()
+        else:
+            self.on()
+
     def color(self, **kwargs):
         params = Params()
         if HEXCODE in kwargs:
@@ -41,7 +52,6 @@ def color(self, **kwargs):
             if GREEN in kwargs:
                 params['g'] = kwargs[GREEN]
         message = self.message(params)
-        print(params, message)
         self.send(message)
 
     def brightness(self, value: int):
@@ -61,7 +71,27 @@ def speed(self, value: int):
 
     def get_config(self):
         message = Request(GET_PILOT)
-        return self.udp.call(message)
+        result_json = self.send(message).decode('UTF-8')
+        result = json.loads(result_json)['result']
+        config = Config(state=result['state'], scene=result['sceneId'], brightness=result['dimming'])
+        if 'r' in result:
+            config.red = result['r']
+            config.green = result['g']
+            config.blue = result['b']
+        return config
+
+    def update_config(self, params):
+        for key, value in params.items():
+            if key == 'r':
+                self.config.red = params['r']
+            elif key == 'g':
+                self.config.green = params['g']
+            elif key == 'b':
+                self.config.blue = params['b']
+            elif key == 'dimming':
+                self.config.brightness = params['dimming']
+            elif key == 'state':
+                self.config.state = params['state']
 
     @staticmethod
     def message(params: Params):
@@ -71,3 +101,4 @@ def message(params: Params):
 if __name__ == '__main__':
     light = Light(IP)
     light.color(red=255)
+    light.switch()