core/homeassistant/components/hive/water_heater.py

116 lines
3.3 KiB
Python

"""Support for hive water heaters."""
from homeassistant.const import TEMP_CELSIUS
from homeassistant.components.water_heater import (
STATE_ECO, STATE_ON, STATE_OFF, SUPPORT_OPERATION_MODE, WaterHeaterDevice)
from . import DATA_HIVE, DOMAIN
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 Wink water heater devices."""
if discovery_info is None:
return
if discovery_info["HA_DeviceType"] != "HotWater":
return
session = hass.data.get(DATA_HIVE)
water_heater = HiveWaterHeater(session, discovery_info)
add_entities([water_heater])
class HiveWaterHeater(WaterHeaterDevice):
"""Hive Water Heater Device."""
def __init__(self, hivesession, hivedevice):
"""Initialize the Water Heater device."""
self.node_id = hivedevice["Hive_NodeID"]
self.node_name = hivedevice["Hive_NodeName"]
self.device_type = hivedevice["HA_DeviceType"]
self.session = hivesession
self.data_updatesource = '{}.{}'.format(
self.device_type, self.node_id)
self._unique_id = '{}-{}'.format(self.node_id, self.device_type)
self._unit_of_measurement = TEMP_CELSIUS
@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
def handle_update(self, updatesource):
"""Handle the new update request."""
if '{}.{}'.format(self.device_type, self.node_id) not in updatesource:
self.schedule_update_ha_state()
@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 self._unit_of_measurement
@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
async def async_added_to_hass(self):
"""When entity is added to Home Assistant."""
await super().async_added_to_hass()
self.session.entities.append(self)
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)
for entity in self.session.entities:
entity.handle_update(self.data_updatesource)
def update(self):
"""Update all Node data from Hive."""
self.session.core.update_data(self.node_id)