Re-add supported features. (#158)

home-assistant-libs · Mar 22, 2018 · a8ab326 · a8ab326
1 parent 177ee74
commit a8ab326
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/pytradfri/device.py b/pytradfri/device.py
@@ -146,6 +146,18 @@ class LightControl:
     def __init__(self, device):
         self._device = device
 
+        self.can_set_temp = None
+        self.can_set_color = None
+
+        if 'WS' in self._device.device_info.model_number:
+            self.can_set_temp = True
+
+        if 'CWS' in self._device.device_info.model_number:
+            self.can_set_color = True
+
+        self.min_mireds = RANGE_MIREDS[0]
+        self.max_mireds = RANGE_MIREDS[1]
+
     @property
     def raw(self):
         """Return raw data that it represents."""

diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
 with open(path.join(here, 'README.md'), encoding='utf-8') as f:
     long_description = f.read()
 
-VERSION = "5.4.0"
+VERSION = "5.4.1"
 DOWNLOAD_URL = \
     'https://github.com/ggravlingen/pytradfri/archive/{}.zip'.format(VERSION)