Custom thermostat

homeassistant/components/thermostat/__init__.py

@@ -70,7 +70,7 @@ def setup(hass, config):
     @util.Throttle(MIN_TIME_BETWEEN_SCANS)
     def update_state(now):
         """ Update thermostat state. """
-        logger.info("Updating nest state")
+        logger.info("Updating thermostat state")
 
         for thermostat in thermostats.values():
             thermostat.update_ha_state(hass, True)
@@ -81,6 +81,7 @@ def setup(hass, config):
 
     def thermostat_service(service):
         """ Handles calls to the services. """
+
         # Convert the entity ids to valid light ids
         target_thermostats = [thermostats[entity_id] for entity_id
                               in extract_entity_ids(hass, service)
@@ -105,7 +106,7 @@ def setup(hass, config):
                 return
 
             for thermostat in target_thermostats:
-                thermostat.nest.set_temperature(temperature)
+                thermostat.set_temperature(temperature)
 
         for thermostat in target_thermostats:
             thermostat.update_ha_state(hass, True)

homeassistant/components/thermostat/heat_control.py

@@ -0,0 +1,166 @@
+"""
+Adds support for a heat control.
+
+Config:
+
+[thermostat]
+platform=heat_control
+
+name = Name of thermostat
+
+heater = entity_id for heater switch,
+         must be a toggle device
+
+target_sensor = entity_id for temperature sensor,
+                target_sensor.state must be temperature
+
+time_temp = start_time-end_time:target_temp,
+
+min_temp = minimum temperature, used when away mode is
+           active or no other temperature specified.
+
+Example:
+[thermostat]
+platform=heat_control
+name = Stue
+heater = switch.Ovn_stue
+target_sensor = tellstick_sensor.Stue_temperature
+time_temp = 0700-0745:17,1500-1850:20
+min_temp = 10
+
+
+"""
+
+import logging
+import datetime
+import homeassistant.components as core
+
+from homeassistant.components.thermostat import ThermostatDevice
+from homeassistant.const import TEMP_CELCIUS, STATE_ON, STATE_OFF
+
+TOL_TEMP = 0.3
+
+
+# pylint: disable=unused-argument
+def get_devices(hass, config):
+    """ Gets thermostats. """
+    logger = logging.getLogger(__name__)
+
+    return [HeatControl(hass, config, logger)]
+
+
+# pylint: disable=too-many-instance-attributes
+class HeatControl(ThermostatDevice):
+    """ Represents a HeatControl within Home Assistant. """
+
+    def __init__(self, hass, config, logger):
+
+        self.logger = logger
+        self.hass = hass
+        self.heater_entity_id = config.get("heater")
+
+        self.name_device = config.get("name")
+        self.target_sensor_entity_id = config.get("target_sensor")
+
+        self.time_temp = []
+        for time_temp in list(config.get("time_temp").split(",")):
+            time, temp = time_temp.split(':')
+            time_start, time_end = time.split('-')
+            start_time = datetime.datetime.time(datetime.datetime.
+                                                strptime(time_start, '%H%M'))
+            end_time = datetime.datetime.time(datetime.datetime.
+                                              strptime(time_end, '%H%M'))
+            self.time_temp.append((start_time, end_time, float(temp)))
+
+        self.min_temp = float(config.get("min_temp"))
+
+        self._manual_sat_temp = None
+        self._heater_manual_changed = True
+
+        hass.states.track_change(self.heater_entity_id,
+                                 self._heater_turned_on,
+                                 STATE_OFF, STATE_ON)
+        hass.states.track_change(self.heater_entity_id,
+                                 self._heater_turned_off,
+                                 STATE_ON, STATE_OFF)
+
+    @property
+    def name(self):
+        """ Returns the name. """
+        return self.name_device
+
+    @property
+    def unit_of_measurement(self):
+        """ Returns the unit of measurement. """
+        return TEMP_CELCIUS
+
+    @property
+    def current_temperature(self):
+        """ Returns the current temperature. """
+        target_sensor = self.hass.states.get(self.target_sensor_entity_id)
+        if target_sensor:
+            return float(target_sensor.state)
+        else:
+            return None
+
+    @property
+    def target_temperature(self):
+        """ Returns the temperature we try to reach. """
+        if self._manual_sat_temp:
+            return self._manual_sat_temp
+        else:
+            now = datetime.datetime.time(datetime.datetime.now())
+            for (start_time, end_time, temp) in self.time_temp:
+                if start_time < now and end_time > now:
+                    return temp
+            return self.min_temp
+
+    def set_temperature(self, temperature):
+        """ Set new target temperature """
+        if temperature is None:
+            self._manual_sat_temp = None
+        else:
+            self._manual_sat_temp = float(temperature)
+
+    def update(self):
+        """ Update current thermostat """
+        heater = self.hass.states.get(self.heater_entity_id)
+        if heater is None:
+            self.logger.error("No heater available")
+            return
+
+        current_temperature = self.current_temperature
+        if current_temperature is None:
+            self.logger.error("No temperature available")
+            return
+
+        if (current_temperature - self.target_temperature) > \
+                TOL_TEMP and heater.state is STATE_ON:
+            self._heater_manual_changed = False
+            core.turn_off(self.hass, self.heater_entity_id)
+        elif (self.target_temperature - self.current_temperature) > TOL_TEMP \
+                and heater.state is STATE_OFF:
+            self._heater_manual_changed = False
+            core.turn_on(self.hass, self.heater_entity_id)
+
+    def _heater_turned_on(self, entity_id, old_state, new_state):
+        """ heater is turned on"""
+        if not self._heater_manual_changed:
+            pass
+        else:
+            self.set_temperature(100)
+
+        self._heater_manual_changed = True
+
+    def _heater_turned_off(self, entity_id, old_state, new_state):
+        """ heater is turned off"""
+        if self._heater_manual_changed:
+            self.set_temperature(None)
+
+    def turn_away_mode_on(self):
+        """ Turns away mode on. """
+        self.set_temperature(None)
+
+    def turn_away_mode_off(self):
+        """ Turns away mode off. """
+        self.set_temperature(self.min_temp)