sascha/badnest
1
0
Fork 0
mirror of https://github.com/USA-RedDragon/badnest.git synced 2025-01-18 19:20:44 +00:00
Code Issues Releases Wiki Activity

maintain backwards compatibility

Browse Source
This commit is contained in:
Guy Badman 2020-05-21 07:11:13 +01:00
parent 4160f9c894
commit 76f0b306b8
No known key found for this signature in database
GPG Key ID: 0259B610971EA10F
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
custom_components/badnest/water_heater.py
View File

@ -11,7 +11,6 @@ from homeassistant.const import (
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
) )
from homeassistant.components.water_heater import ( from homeassistant.components.water_heater import (
WaterHeaterDevice,
SCAN_INTERVAL, SCAN_INTERVAL,
STATE_OFF, STATE_OFF,
STATE_ON, STATE_ON,
@ -22,6 +21,11 @@ from homeassistant.components.water_heater import (
ATTR_OPERATION_MODE, ATTR_OPERATION_MODE,
ATTR_OPERATION_LIST, ATTR_OPERATION_LIST,
) )
try:
from homeassistant.components.water_heater import WaterHeaterEntity
except ImportError:
from homeassistant.components.water_heater import WaterHeaterDevice as WaterHeaterEntity
from .const import ( from .const import (
DOMAIN, DOMAIN,
) )