From c267326891b18ac67d35ef1e3117efba726a9c3a Mon Sep 17 00:00:00 2001
From: Patrik <slasktratt@gammelgravlingen.com>
Date: Thu, 13 Apr 2017 19:04:42 +0200
Subject: [PATCH] Added initial support for IKEA Tradfri Gateway (#7074)

* Added initial support for IKEA Tradfri Gateway

* Pinned requirement

* Fixed lint-errors

* Added file to .coveragerc

* Trying to fix commit

* Fixed requirements_all again

* Minor reorder of code

* Minor reorder of code

* Made the changes suggested by @balloob

* Made the changes suggested by @balloob and removed debug

* Update tradfri.py
---
 .coveragerc                               |   1 +
 homeassistant/components/light/tradfri.py | 100 ++++++++++++++++++++++
 requirements_all.txt                      |   3 +
 3 files changed, 104 insertions(+)
 create mode 100644 homeassistant/components/light/tradfri.py

diff --git a/.coveragerc b/.coveragerc
index b2f6a21c84b..2d40ca8b7b3 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -232,6 +232,7 @@ omit =
     homeassistant/components/light/limitlessled.py
     homeassistant/components/light/osramlightify.py
     homeassistant/components/light/tikteck.py
+    homeassistant/components/light/tradfri.py
     homeassistant/components/light/x10.py
     homeassistant/components/light/yeelight.py
     homeassistant/components/light/yeelightsunflower.py
diff --git a/homeassistant/components/light/tradfri.py b/homeassistant/components/light/tradfri.py
new file mode 100644
index 00000000000..1b2e42b5b99
--- /dev/null
+++ b/homeassistant/components/light/tradfri.py
@@ -0,0 +1,100 @@
+"""Support for the IKEA Tradfri platform."""
+
+import logging
+
+
+import voluptuous as vol
+
+# Import the device class from the component that you want to support
+from homeassistant.components.light import ATTR_BRIGHTNESS, \
+    SUPPORT_BRIGHTNESS, Light, PLATFORM_SCHEMA
+from homeassistant.const import CONF_HOST, CONF_API_KEY
+import homeassistant.helpers.config_validation as cv
+
+
+SUPPORTED_FEATURES = (SUPPORT_BRIGHTNESS)
+
+# Home Assistant depends on 3rd party packages for API specific code.
+REQUIREMENTS = ['pytradfri==0.4']
+
+_LOGGER = logging.getLogger(__name__)
+
+# Validation of the user's configuration
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
+    vol.Required(CONF_HOST): cv.string,
+    vol.Optional(CONF_API_KEY): cv.string,
+})
+
+
+def setup_platform(hass, config, add_devices, discovery_info=None):
+    """Setup the IKEA Tradfri Light platform."""
+    import pytradfri
+
+    # Assign configuration variables.
+    # The configuration check takes care they are present.
+    host = config.get(CONF_HOST)
+    securitycode = config.get(CONF_API_KEY)
+
+    api = pytradfri.coap_cli.api_factory(host, securitycode)
+
+    gateway = pytradfri.gateway.Gateway(api)
+    devices = gateway.get_devices()
+    lights = [dev for dev in devices if dev.has_light_control]
+
+    _LOGGER.debug("IKEA Tradfri Hub | init | Initialization Process Complete")
+
+    add_devices(IKEATradfri(light) for light in lights)
+    _LOGGER.debug("IKEA Tradfri Hub | get_lights | All Lights Loaded")
+
+
+class IKEATradfri(Light):
+    """The platform class required by hass."""
+
+    def __init__(self, light):
+        """Initialize a Light."""
+        self._light = light
+
+        # Caching of LightControl and light object
+        self._light_control = light.light_control
+        self._light_data = light.light_control.lights[0]
+        self._name = light.name
+        self._brightness = None
+
+    @property
+    def supported_features(self):
+        """Flag supported features."""
+        return SUPPORTED_FEATURES
+
+    @property
+    def name(self):
+        """Return the display name of this light."""
+        return self._name
+
+    @property
+    def is_on(self):
+        """Return true if light is on."""
+        return self._light_data.state
+
+    @property
+    def brightness(self):
+        """Brightness of the light (an integer in the range 1-255)."""
+        return self._light_data.dimmer
+
+    def turn_off(self, **kwargs):
+        """Instruct the light to turn off."""
+        return self._light_control.set_state(False)
+
+    def turn_on(self, **kwargs):
+        """Instruct the light to turn on."""
+        if ATTR_BRIGHTNESS in kwargs:
+            self._light.light_control.set_dimmer(kwargs.get(ATTR_BRIGHTNESS))
+        else:
+            self._light.light_control.set_state(True)
+
+    def update(self):
+        """Fetch new state data for this light.
+
+        This is the only method that should fetch new data for Home Assistant.
+        """
+        self._light.update()
+        self._brightness = self._light_data.dimmer
diff --git a/requirements_all.txt b/requirements_all.txt
index defefb66aee..2820d9ad6c5 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -652,6 +652,9 @@ python-wink==1.2.3
 # homeassistant.components.device_tracker.trackr
 pytrackr==0.0.5
 
+# homeassistant.components.light.tradfri
+pytradfri==0.4
+
 # homeassistant.components.device_tracker.unifi
 pyunifi==2.0