light/hyperion: use RGB, clean code

pull/544/head
MakeMeASandwich 2015-10-20 17:30:23 +02:00
parent c5a7e3abd1
commit 2e9ee28637
1 changed files with 41 additions and 45 deletions

View File

@ -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)