Adding climate.velbus support

homeassistant/components/climate/velbus.py

@@ -0,0 +1,81 @@
+"""
+Velbus thermostat.
+
+For more details about this platform, please refer to the documentation
+https://home-assistant.io/components/climate.velbus/
+"""
+import logging
+
+from homeassistant.components.velbus import (
+    DOMAIN as VELBUS_DOMAIN, VelbusEntity)
+from homeassistant.components.climate import (
+    ClimateDevice, DOMAIN,
+    SUPPORT_TARGET_TEMPERATURE,
+    SUPPORT_OPERATION_MODE)
+from homeassistant.const import (
+    TEMP_CELSIUS, ATTR_TEMPERATURE)
+
+_LOGGER = logging.getLogger(__name__)
+
+DEPENDENCIES = ['velbus']
+
+SUPPORT_FLAGS = (SUPPORT_TARGET_TEMPERATURE | SUPPORT_OPERATION_MODE)
+OPERATION_LIST = ['comfort', 'day', 'night', 'safe']
+
+async def async_setup_platform(hass, config, async_add_entities,
+                               discovery_info=None):
+    """Set up the Velbus temp sensor platform."""
+    if discovery_info is None:
+        return
+    sensors = []
+    for sensor in discovery_info:
+        module = hass.data[VELBUS_DOMAIN].get_module(sensor[0])
+        channel = sensor[1]
+        sensors.append(VelbusClimate(module, channel))
+    async_add_entities(sensors)
+
+
+class VelbusClimate(VelbusEntity, ClimateDevice):
+    """Representation of a sensor."""
+
+    @property
+    def supported_features(self):
+        return SUPPORT_FLAGS
+
+    @property
+    def temperature_unit(self):
+        """Return the unit this state is expressed in."""
+        return self._module.get_unit(self._channel)
+
+    @property
+    def current_temperature(self):
+        """Return the current temperature."""
+        return self._module.get_state(self._channel)
+
+    @property
+    def current_operation(self):
+        """Return current operation ie. heat, cool, idle."""
+        return self._module.get_climate_mode()
+
+    @property
+    def operation_list(self):
+        """Return the list of available operation modes."""
+        return OPERATION_LIST
+
+    @property
+    def target_temperature(self):
+        """Return the temperature we try to reach."""
+        return self._module.get_climate_target()
+
+    def set_operation_mode(self, operation_mode):
+        """Set new target operation mode."""
+        self._module.set_mode(operation_mode)
+        self.schedule_update_ha_state()
+
+    def set_temperature(self, **kwargs):
+        """Set new target temperatures."""
+        if kwargs.get(ATTR_TEMPERATURE) is not None:
+            self._module.set_temp(kwargs.get(ATTR_TEMPERATURE))
+            self.schedule_update_ha_state()
+
+

homeassistant/components/velbus.py

@@ -12,7 +12,7 @@
 from homeassistant.helpers.discovery import load_platform
 from homeassistant.helpers.entity import Entity
 
-REQUIREMENTS = ['python-velbus==2.0.20']
+REQUIREMENTS = ['python-velbus==2.0.21']
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -47,6 +47,7 @@ def callback():
         modules = controller.get_modules()
         discovery_info = {
             'switch': [],
+            'climate': [],
             'binary_sensor': [],
             'sensor': []
         }
@@ -60,6 +61,8 @@ def callback():
                         ))
         load_platform(hass, 'switch', DOMAIN,
                       discovery_info['switch'], config)
+        load_platform(hass, 'climate', DOMAIN,
+                      discovery_info['climate'], config)
         load_platform(hass, 'binary_sensor', DOMAIN,
                       discovery_info['binary_sensor'], config)
         load_platform(hass, 'sensor', DOMAIN,

requirements_all.txt

@@ -1203,7 +1203,7 @@ python-telegram-bot==11.1.0
 python-twitch-client==0.6.0
 
 # homeassistant.components.velbus
-python-velbus==2.0.20
+python-velbus==2.0.21
 
 # homeassistant.components.media_player.vlc
 python-vlc==1.1.2