Check for attribute existence for HS220 support (home-assistant#21309)

* Check for attribute existence as smartplug does not have them (for HS220 support) * use getattr over hasattr and a separate check
sashayohan · Feb 23, 2019 · 492c3b2 · 492c3b2
1 parent 1eba90d
commit 492c3b2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/homeassistant/components/tplink/light.py b/homeassistant/components/tplink/light.py
@@ -222,11 +222,11 @@ def get_features(self):
 
         if self.smartbulb.is_dimmable:
             self._supported_features += SUPPORT_BRIGHTNESS
-        if self.smartbulb.is_variable_color_temp:
+        if getattr(self.smartbulb, 'is_variable_color_temp', False):
             self._supported_features += SUPPORT_COLOR_TEMP
             self._min_mireds = kelvin_to_mired(
                 self.smartbulb.valid_temperature_range[1])
             self._max_mireds = kelvin_to_mired(
                 self.smartbulb.valid_temperature_range[0])
-        if self.smartbulb.is_color:
+        if getattr(self.smartbulb, 'is_color', False):
             self._supported_features += SUPPORT_COLOR