From 3fe895c18f0ead0dd4af3b4648e88f8f14adcb6e Mon Sep 17 00:00:00 2001
From: Maikel Punie <maikel.punie@gmail.com>
Date: Fri, 2 Nov 2018 14:43:17 +0100
Subject: [PATCH]  Adding climate.velbus support  (#18100)

* Adding climate.velbus support

* Fix version

* fixed houndci-bot

* More fixes

* Fix typos and ordering
---
 homeassistant/components/climate/velbus.py | 81 ++++++++++++++++++++++
 homeassistant/components/velbus.py         |  5 +-
 requirements_all.txt                       |  2 +-
 3 files changed, 86 insertions(+), 2 deletions(-)
 create mode 100644 homeassistant/components/climate/velbus.py

diff --git a/homeassistant/components/climate/velbus.py b/homeassistant/components/climate/velbus.py
new file mode 100644
index 00000000000..ab8542541c8
--- /dev/null
+++ b/homeassistant/components/climate/velbus.py
@@ -0,0 +1,81 @@
+"""
+Support for 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.climate import (
+    SUPPORT_OPERATION_MODE, SUPPORT_TARGET_TEMPERATURE, ClimateDevice)
+from homeassistant.components.velbus import (
+    DOMAIN as VELBUS_DOMAIN, VelbusEntity)
+from homeassistant.const import ATTR_TEMPERATURE
+
+_LOGGER = logging.getLogger(__name__)
+
+DEPENDENCIES = ['velbus']
+
+OPERATION_LIST = ['comfort', 'day', 'night', 'safe']
+
+SUPPORT_FLAGS = (SUPPORT_TARGET_TEMPERATURE | SUPPORT_OPERATION_MODE)
+
+
+async def async_setup_platform(
+        hass, config, async_add_entities, discovery_info=None):
+    """Set up the Velbus thermostat 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 Velbus thermostat."""
+
+    @property
+    def supported_features(self):
+        """Return the list off supported features."""
+        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()
diff --git a/homeassistant/components/velbus.py b/homeassistant/components/velbus.py
index 2304054c404..294061527f8 100644
--- a/homeassistant/components/velbus.py
+++ b/homeassistant/components/velbus.py
@@ -12,7 +12,7 @@ from homeassistant.const import EVENT_HOMEASSISTANT_STOP, CONF_PORT
 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 @@ async def async_setup(hass, config):
         modules = controller.get_modules()
         discovery_info = {
             'switch': [],
+            'climate': [],
             'binary_sensor': [],
             'sensor': []
         }
@@ -60,6 +61,8 @@ async def async_setup(hass, config):
                         ))
         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,
diff --git a/requirements_all.txt b/requirements_all.txt
index 2b080a5751d..466eb6ead54 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -1210,7 +1210,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