From aa69abaf522ed70d3142f056e924bc0566667732 Mon Sep 17 00:00:00 2001
From: "E. Almqvist" <elalmqvist@gmail.com>
Date: Tue, 4 Aug 2020 10:47:10 +0200
Subject: [PATCH] Switch light function

---
 hue_controller.py | 39 +++++++++++++++++++++++++++++++++------
 1 file changed, 33 insertions(+), 6 deletions(-)

diff --git a/hue_controller.py b/hue_controller.py
index fd61d6c..e00f7f3 100644
--- a/hue_controller.py
+++ b/hue_controller.py
@@ -3,16 +3,26 @@ import json # API uses JSON
 import asyncio # ASync stuff
 import config # Configuration for the controller (/config.py <- change this file)
 
+LIGHTS = {} # dictionary of all the lights
+
+
 loop = asyncio.get_event_loop() # ASync loop
 
 def genUrl(params: str):
     return "http://" + config.address + "/api/" + config.username + params
 
+boolStr = {
+    True: "true",
+    False: "false"
+}
+
+StrBool = { # because doing something else is too expensive 
+    "true": True,
+    "false": False
+}
+
 def boolToString(v: bool): # To fix the dumb python syntax
-    if(v):
-        return "true"
-    else:
-        return "false"
+    return boolStr[v]
 
 class APIrequest:
     # Get Req
@@ -46,6 +56,7 @@ class APIrequest:
 
 
 class controller:
+
     async def getLights():
         return await APIrequest.get("/lights")
 
@@ -53,9 +64,25 @@ class controller:
         return await APIrequest.get( "/lights/" + str(index) )
 
     async def toggleLight(index: int=1, isOn: bool=True):
-        await APIrequest.put( "/lights/1/state", '{"on":' + boolToString(isOn) + '}' )
+        await APIrequest.put( "/lights/" + str(index) + "/state", '{"on":' + boolToString(isOn) + '}' )
+
+    def switchLight( index: int=1 ):
+        key = LIGHTS.get(str(index))
+        if(key):
+            if( key.get(state) ):
+                loop.run_until_complete( controller.toggleLight( index, boolToString(StrBool[LIGHTS[str(index)]["state"]["on"]]) ) )
+        else:
+            print("Error: Light index out of range")
+
+    def init():
+        jsonLights = loop.run_until_complete(APIrequest.get("/lights"))
+        global LIGHTS
+        LIGHTS = json.loads(jsonLights.text)
+        print(LIGHTS)
 
 def testReq():
-    loop.run_until_complete( controller.toggleLight(1, True) ) # try to turn on/off a light
+    controller.init()
+    controller.switchLight( 1 )
+    # loop.run_until_complete( controller.toggleLight(1, True) ) # try to turn on/off a light
 
     loop.close()