From 2e9ee2863734fa3e1cec6e450c0888a8e3f3887f Mon Sep 17 00:00:00 2001
From: MakeMeASandwich <makemeasandwich@users.noreply.github.com>
Date: Tue, 20 Oct 2015 17:30:23 +0200
Subject: [PATCH] light/hyperion: use RGB, clean code

---
 homeassistant/components/light/hyperion.py | 86 +++++++++++-----------
 1 file changed, 41 insertions(+), 45 deletions(-)

diff --git a/homeassistant/components/light/hyperion.py b/homeassistant/components/light/hyperion.py
index ddd98661155..eed30f8b2d0 100644
--- a/homeassistant/components/light/hyperion.py
+++ b/homeassistant/components/light/hyperion.py
@@ -20,10 +20,7 @@ import socket
 import json
 
 from homeassistant.const import CONF_HOST
-from homeassistant.components.light import (Light, ATTR_XY_COLOR,
-                                            ATTR_BRIGHTNESS)
-from homeassistant.util.color import (color_RGB_to_xy,
-                                      color_xy_brightness_to_RGB)
+from homeassistant.components.light import (Light, ATTR_RGB_COLOR)
 
 _LOGGER = logging.getLogger(__name__)
 REQUIREMENTS = []
@@ -33,7 +30,11 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
     """ Sets up a Hyperion server remote """
     host = config.get(CONF_HOST, None)
     port = config.get("port", 19444)
-    add_devices_callback([Hyperion(host, port)])
+    device = Hyperion(host, port)
+    if device.setup():
+        add_devices_callback([device])
+    else:
+        return False
 
 
 class Hyperion(Light):
@@ -42,76 +43,74 @@ class Hyperion(Light):
     def __init__(self, host, port):
         self._host = host
         self._port = port
-        self._name = "unknown"
-        self._is_available = False
-        self._xy_color = color_RGB_to_xy(0, 0, 0)
-        self._brightness = 255
+        self._name = host
+        self._is_available = True
+        self._rgb_color = [255, 255, 255]
 
     @property
     def name(self):
-        """ Get the hostname of the server. """
+        """ Return the hostname of the server. """
         return self._name
 
     @property
-    def color_xy(self):
-        """ XY color value. """
-        return self._xy_color
-
-    @property
-    def brightness(self):
-        """ Brightness. """
-        return self._brightness
+    def rgb_color(self):
+        """ Last RGB color value set. """
+        return self._rgb_color
 
     @property
     def is_on(self):
-        """ True if device is on. """
-        self.check_remote()
+        """ True if the device is online. """
         return self._is_available
 
     def turn_on(self, **kwargs):
         """ Turn the lights on. """
         if self._is_available:
-            if ATTR_XY_COLOR in kwargs:
-                self._xy_color = kwargs[ATTR_XY_COLOR]
-            if ATTR_BRIGHTNESS in kwargs:
-                self._brightness = kwargs[ATTR_BRIGHTNESS]
-            self.update_remote()
+            if ATTR_RGB_COLOR in kwargs:
+                self._rgb_color = kwargs[ATTR_RGB_COLOR]
+
+            self.json_request({"command": "color", "priority": 128,
+                               "color": self._rgb_color})
 
     def turn_off(self, **kwargs):
         """ Disconnect the remote. """
         self.json_request({"command": "clearall"})
 
-    def check_remote(self):
-        """ Ping the remote and gets the hostname. """
+    def update(self):
+        """ Ping the remote. """
+        # just see if the remote port is open
+        self._is_available = self.json_request()
+
+    def setup(self):
+        """ Get the hostname of the remote. """
         response = self.json_request({"command": "serverinfo"})
         if response:
             self._name = response["info"]["hostname"]
+            return True
 
-    def update_remote(self):
-        """ Set the remote's lights. """
-        rgb = color_xy_brightness_to_RGB(self._xy_color[0], self._xy_color[1],
-                                         self._brightness)
-        rgb = [int(c) for c in rgb]
-        self.json_request(
-            {"command": "color", "priority": 128, "color": rgb})
+        return False
 
-    def json_request(self, request, wait_for_response=False):
+    def json_request(self, request=None, wait_for_response=False):
         """ Communicate with the json server. """
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.settimeout(5)
+
         try:
-            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-            sock.settimeout(5)
             sock.connect((self._host, self._port))
         except OSError:
-            self._is_available = False
-            return None
+            return False
+
+        if not request:
+            # no communication needed, simple presence detection returns True
+            return True
 
         sock.send(bytearray(json.dumps(request) + "\n", "utf-8"))
-
         try:
             buf = sock.recv(4096)
         except socket.timeout:
-            return None
+            # something is wrong, assume it's offline
+            return False
 
+        # read until a newline or timeout
         buffering = True
         while buffering:
             if "\n" in str(buf, "utf-8"):
@@ -129,7 +128,4 @@ class Hyperion(Light):
                     buf += more
 
         sock.close()
-
-        j = json.loads(response)
-        self._is_available = True
-        return j
+        return json.loads(response)