From bc4e37a398021bacbd4a9f9af72db94151e8f48d Mon Sep 17 00:00:00 2001
From: Christian Colglazier <christian@cacolglazier.com>
Date: Sun, 25 Oct 2020 14:27:51 -0400
Subject: [PATCH] Home Assistant post requests

---
 scripts/homeassistant.py | 42 ++++++++++++++++++++++++++++++++++++----
 1 file changed, 38 insertions(+), 4 deletions(-)

diff --git a/scripts/homeassistant.py b/scripts/homeassistant.py
index cac6bcc..ab9592f 100644
--- a/scripts/homeassistant.py
+++ b/scripts/homeassistant.py
@@ -6,12 +6,16 @@ from requests import get, post
 import json
 
 class HomeAssistant(object):
+
+    # Initalizes Home Assistant API wrapper.
     def __init__(self, ip, token):
         self.url = 'http://{}:8123'.format(ip)
         self.headers = {
             'Authorization': 'Bearer {}'.format(token),
             'content-type': 'application/json',
         }
+
+    # Sends post requests.
     def postService(self, domain, service, data):
         response = post("{}/api/services/{}/{}".format(self.url,
                                                        domain,
@@ -19,10 +23,12 @@ class HomeAssistant(object):
                         headers=self.headers, data=json.dumps(data))
         response.raise_for_status()
         return response
-    def getRequest(self, domain, data=''):
+
+    # Sends get requests and turns requested data.
+    def getRequest(self, domain):
         response = get("{}/api/{}".format(self.url, domain),
-                        headers=self.headers, data=json.dumps(data))
-        return response.text
+                        headers=self.headers)
+        return json.loads(response.text)
 
     # Returns a message if the API is up and running.
     def getAPI(self):
@@ -95,7 +101,35 @@ class HomeAssistant(object):
     # Returns the data (image) from the specified camera entity_id.
     def getCameraProxy(self, entityId):
         return self.getRequest('camera_proxy/' + entityId)
-    
+
+    # Runs a Home Assistant scene.
     def runScene(self, entityId):
         data = {'entity_id': entityId}
         self.postService('scene', 'turn_on', data)
+
+    # Runs a Home Assistant script.
+    def runScript(self, entityId):
+        data = {'entity_id': entityId}
+        self.postService('script', 'turn_on', data)
+
+    # Sets the brightness level of a device.
+    def setLevel(self, entityId, level):
+        data = {'entity_id': entityId, 'brightness_pct': level}
+        self.postService('homeassistant', 'turn_on', data)
+
+    # Turns a device off.
+    def turnOn(self, entityId):
+        data = {'entity_id': entityId}
+        self.postService('homeassistant', 'turn_on', data)
+
+    # Turns a device on.
+    def turnOff(self, entityId):
+        data = {'entity_id': entityId}
+        self.postService('homeassistant', 'turn_off', data)
+
+    # Turns a device of the given power state.
+    def setOnOff(self, entityId, power):
+        if power:
+            self.turnOn(entityId)
+        else:
+            self.turnOff(entityId)