81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
"""Support for hive water heaters."""
|
|
from homeassistant.components.water_heater import (
|
|
STATE_ECO,
|
|
STATE_OFF,
|
|
STATE_ON,
|
|
SUPPORT_OPERATION_MODE,
|
|
WaterHeaterEntity,
|
|
)
|
|
from homeassistant.const import TEMP_CELSIUS
|
|
|
|
from . import DATA_HIVE, DOMAIN, HiveEntity, refresh_system
|
|
|
|
SUPPORT_FLAGS_HEATER = SUPPORT_OPERATION_MODE
|
|
|
|
HIVE_TO_HASS_STATE = {"SCHEDULE": STATE_ECO, "ON": STATE_ON, "OFF": STATE_OFF}
|
|
HASS_TO_HIVE_STATE = {STATE_ECO: "SCHEDULE", STATE_ON: "ON", STATE_OFF: "OFF"}
|
|
SUPPORT_WATER_HEATER = [STATE_ECO, STATE_ON, STATE_OFF]
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the Hive water heater devices."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
session = hass.data.get(DATA_HIVE)
|
|
devs = []
|
|
for dev in discovery_info:
|
|
devs.append(HiveWaterHeater(session, dev))
|
|
add_entities(devs)
|
|
|
|
|
|
class HiveWaterHeater(HiveEntity, WaterHeaterEntity):
|
|
"""Hive Water Heater Device."""
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return unique ID of entity."""
|
|
return self._unique_id
|
|
|
|
@property
|
|
def device_info(self):
|
|
"""Return device information."""
|
|
return {"identifiers": {(DOMAIN, self.unique_id)}, "name": self.name}
|
|
|
|
@property
|
|
def supported_features(self):
|
|
"""Return the list of supported features."""
|
|
return SUPPORT_FLAGS_HEATER
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the water heater."""
|
|
if self.node_name is None:
|
|
self.node_name = "Hot Water"
|
|
return self.node_name
|
|
|
|
@property
|
|
def temperature_unit(self):
|
|
"""Return the unit of measurement."""
|
|
return TEMP_CELSIUS
|
|
|
|
@property
|
|
def current_operation(self):
|
|
"""Return current operation."""
|
|
return HIVE_TO_HASS_STATE[self.session.hotwater.get_mode(self.node_id)]
|
|
|
|
@property
|
|
def operation_list(self):
|
|
"""List of available operation modes."""
|
|
return SUPPORT_WATER_HEATER
|
|
|
|
@refresh_system
|
|
def set_operation_mode(self, operation_mode):
|
|
"""Set operation mode."""
|
|
new_mode = HASS_TO_HIVE_STATE[operation_mode]
|
|
self.session.hotwater.set_mode(self.node_id, new_mode)
|
|
|
|
def update(self):
|
|
"""Update all Node data from Hive."""
|
|
self.session.core.update_data(self.node_id)
|