Huge restructure in progress

custom_components/browser_mod/light.py

@@ -0,0 +1,52 @@
+import logging
+from datetime import datetime
+
+from homeassistant.const import STATE_UNAVAILABLE, STATE_ON, STATE_OFF
+from homeassistant.components.light import Light, SUPPORT_BRIGHTNESS
+
+from .helpers import setup_platform, BrowserModEntity
+
+PLATFORM = 'light'
+
+async def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
+    return setup_platform(hass, config, async_add_devices, PLATFORM, BrowserModLight)
+
+class BrowserModLight(Light, BrowserModEntity):
+    domain = PLATFORM
+
+    def __init__(self, hass, connection, deviceID, alias=None):
+        super().__init__(hass, connection, deviceID, alias)
+
+    def updated(self):
+        self.last_seen = datetime.now()
+        self.schedule_update_ha_state()
+
+    @property
+    def state(self):
+        if not self.connection.connection:
+            return STATE_UNAVAILABLE
+        if self.data.get('blackout', False):
+            return STATE_OFF
+        return STATE_ON
+
+    @property
+    def is_on(self):
+        return not self.data.get('blackout', False)
+
+    @property
+    def device_state_attributes(self):
+        return {
+                "type": "browser_mod",
+                }
+
+    @property
+    def supported_features(self):
+        if self.data.get('brightness', False):
+            return SUPPORT_BRIGHTNESS
+        return 0
+
+    def turn_on(self, **kwargs):
+        self.connection.send("no-blackout", **kwargs)
+
+    def turn_off(self, **kwargs):
+        self.connection.send("blackout")