From 3dec59b721c51ea1f4478906b06d6bdfe915e544 Mon Sep 17 00:00:00 2001
From: filmgarage <30559371+filmgarage@users.noreply.github.com>
Date: Mon, 8 Jun 2020 00:03:17 +0200
Subject: [PATCH] Update light.py

Included proposed changes as described in the HA blog: https://developers.home-assistant.io/blog/2020/05/14/entity-class-names
Light is deprecated and replaced by LightEntity.
No idea if what I did makes sense, but might be a starting point to fix this error
Should fix issue #31
---
 custom_components/dmx/light.py | 40 +++++++++++++++++++++++-----------
 1 file changed, 27 insertions(+), 13 deletions(-)

diff --git a/custom_components/dmx/light.py b/custom_components/dmx/light.py
index f9ca2e3..741d4f7 100644
--- a/custom_components/dmx/light.py
+++ b/custom_components/dmx/light.py
@@ -12,18 +12,32 @@
 
 from homeassistant.const import (CONF_DEVICES, CONF_HOST, CONF_NAME, CONF_PORT,
                                  CONF_TYPE, STATE_ON, STATE_OFF)
-from homeassistant.components.light import (ATTR_BRIGHTNESS,
-                                            ATTR_HS_COLOR,
-                                            ATTR_TRANSITION,
-                                            ATTR_WHITE_VALUE,
-                                            ATTR_COLOR_TEMP,
-                                            Light,
-                                            PLATFORM_SCHEMA,
-                                            SUPPORT_BRIGHTNESS,
-                                            SUPPORT_COLOR,
-                                            SUPPORT_WHITE_VALUE,
-                                            SUPPORT_TRANSITION,
-                                            SUPPORT_COLOR_TEMP)
+try:
+  from homeassistant.components.light import (ATTR_BRIGHTNESS,
+                                              ATTR_HS_COLOR,
+                                              ATTR_TRANSITION,
+                                              ATTR_WHITE_VALUE,
+                                              ATTR_COLOR_TEMP,
+                                              LightEntity,
+                                              PLATFORM_SCHEMA,
+                                              SUPPORT_BRIGHTNESS,
+                                              SUPPORT_COLOR,
+                                              SUPPORT_WHITE_VALUE,
+                                              SUPPORT_TRANSITION,
+                                              SUPPORT_COLOR_TEMP)
+except ImportError:
+  from homeassistant.components.light import (ATTR_BRIGHTNESS,
+                                              ATTR_HS_COLOR,
+                                              ATTR_TRANSITION,
+                                              ATTR_WHITE_VALUE,
+                                              ATTR_COLOR_TEMP,
+                                              Light as LightEntity,
+                                              PLATFORM_SCHEMA,
+                                              SUPPORT_BRIGHTNESS,
+                                              SUPPORT_COLOR,
+                                              SUPPORT_WHITE_VALUE,
+                                              SUPPORT_TRANSITION,
+                                              SUPPORT_COLOR_TEMP)
 from homeassistant.util.color import color_rgb_to_rgbw
 import homeassistant.helpers.config_validation as cv
 import homeassistant.util.color as color_util
@@ -167,7 +181,7 @@ def async_setup_platform(hass, config, async_add_devices, discovery_info=None):
     return True
 
 
-class DMXLight(Light):
+class DMXLight(LightEntity):
     """Representation of a DMX Art-Net light."""
 
     def __init__(self, light, dmx_gateway, send_immediately, default_type):