Fix deprecation warnings in the log

Closes #12
sebr · May 21, 2020 · 816afdd · 816afdd
1 parent 9c8ce77
commit 816afdd
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/custom_components/bhyve/binary_sensor.py b/custom_components/bhyve/binary_sensor.py
@@ -1,7 +1,13 @@
 """Support for Orbit BHyve binary sensors ()."""
 import logging
 
-from homeassistant.components.binary_sensor import BinarySensorDevice
+try:
+    from homeassistant.components.binary_sensor import BinarySensorEntity
+except ImportError:
+    from homeassistant.components.binary_sensor import (
+        BinarySensorDevice as BinarySensorEntity,
+    )
+
 from homeassistant.helpers.event import async_call_later
 
 from . import BHyveEntity
@@ -39,7 +45,7 @@ async def async_setup_platform(hass, config, async_add_entities, _discovery_info
     async_add_entities(binary_sensors, True)
 
 
-class BHyveRainDelayBinarySensor(BHyveEntity, BinarySensorDevice):
+class BHyveRainDelayBinarySensor(BHyveEntity, BinarySensorEntity):
     """Define a BHyve binary sensor."""
 
     def _setup(self, device):

diff --git a/custom_components/bhyve/switch.py b/custom_components/bhyve/switch.py
@@ -3,7 +3,15 @@
 import logging
 
 from datetime import timedelta
-from homeassistant.components.switch import DEVICE_CLASS_SWITCH, SwitchDevice
+
+try:
+    from homeassistant.components.switch import DEVICE_CLASS_SWITCH, SwitchEntity
+except ImportError:
+    from homeassistant.components.switch import (
+        DEVICE_CLASS_SWITCH,
+        SwitchDevice as SwitchEntity,
+    )
+
 from homeassistant.util import dt
 
 from . import BHyveEntity
@@ -49,7 +57,7 @@ async def async_setup_platform(hass, config, async_add_entities, _discovery_info
     async_add_entities(switches, True)
 
 
-class BHyveZoneSwitch(BHyveEntity, SwitchDevice):
+class BHyveZoneSwitch(BHyveEntity, SwitchEntity):
     """Define a BHyve switch."""
 
     def __init__(self, hass, bhyve, device, zone, name, programs, icon):