Add Broadlink-wifi climate support

.coveragerc

@@ -50,6 +50,7 @@ omit =
     homeassistant/components/blink/*
     homeassistant/components/bloomsky/*
     homeassistant/components/bmw_connected_drive/*
+    homeassistant/components/broadlink/*
     homeassistant/components/browser/*
     homeassistant/components/calendar/caldav.py
     homeassistant/components/calendar/todoist.py

homeassistant/components/broadlink/__init__.py

@@ -0,0 +1 @@
+"""Component to embed broadlink."""

homeassistant/components/broadlink/climate.py

@@ -0,0 +1,250 @@
+"""
+Support for Chinese wifi thermostats (Floureon, Beok, Beca Energy).
+
+For more details about this platform, please refer to the documentation at
+https://home-assistant.io/components/climate.broadlink/
+"""
+import logging
+
+import voluptuous as vol
+
+from homeassistant.components.climate import PLATFORM_SCHEMA, ClimateDevice
+from homeassistant.components.climate.const import (
+    DOMAIN, STATE_AUTO, STATE_HEAT, SUPPORT_AWAY_MODE, SUPPORT_OPERATION_MODE,
+    SUPPORT_TARGET_TEMPERATURE)
+from homeassistant.const import (
+    ATTR_ENTITY_ID, ATTR_TEMPERATURE, CONF_FRIENDLY_NAME, CONF_HOST, CONF_MAC,
+    PRECISION_HALVES, STATE_OFF, TEMP_CELSIUS)
+import homeassistant.helpers.config_validation as cv
+
+
+REQUIREMENTS = ['BroadlinkWifiThermostat==2.4.1']
+
+CONF_EXTERNAL_TEMP = 'external_temp'
+CONF_AWAY_TEMP = 'away_temp'
+
+ATTR_LOOP_MODE = 'loop_mode'
+ATTR_SEN = 'sen'
+ATTR_OSV = 'osv'
+ATTR_DIF = 'dif'
+ATTR_SVH = 'svh'
+ATTR_SVL = 'svl'
+ATTR_ADJ = 'adj'
+ATTR_FRE = 'freeze'
+ATTR_PON = 'pon'
+
+ATTR_WEEK_START_1 = 'week_start_1'
+ATTR_WEEK_STOP_1 = 'week_stop_1'
+ATTR_WEEK_START_2 = 'week_start_2'
+ATTR_WEEK_STOP_2 = 'week_stop_2'
+ATTR_WEEK_START_3 = 'week_start_3'
+ATTR_WEEK_STOP_3 = 'week_stop_3'
+
+ATTR_WEEKEND_START = 'weekend_start'
+ATTR_WEEKEND_STOP = 'weekend_stop'
+ATTR_AWAY_TEMP = 'away_temp'
+ATTR_HOME_TEMP = 'home_temp'
+
+SERVICE_SET_SCHEDULE = 'broadlink_set_schedule'
+SERVICE_SET_ADVANCED_CONF = 'broadlink_set_advanced_conf'
+
+# Validation of the user's configuration
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
+    vol.Required(CONF_HOST): cv.string,
+    vol.Required(CONF_MAC): cv.string,
+    vol.Required(CONF_FRIENDLY_NAME): cv.string,
+    vol.Optional(CONF_EXTERNAL_TEMP, default=False): cv.boolean,
+    vol.Optional(CONF_AWAY_TEMP, default=12): vol.Coerce(float),
+    vol.Optional(ATTR_LOOP_MODE, default=0): vol.Coerce(int)
+})
+
+SET_SCHEDULE_SCHEMA = vol.Schema({
+    vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
+    vol.Required(ATTR_WEEK_START_1): cv.time,
+    vol.Required(ATTR_WEEK_STOP_1): cv.time,
+    vol.Required(ATTR_WEEK_START_2): cv.time,
+    vol.Required(ATTR_WEEK_STOP_2): cv.time,
+    vol.Required(ATTR_WEEK_START_3): cv.time,
+    vol.Required(ATTR_WEEK_STOP_3): cv.time,
+    vol.Required(ATTR_WEEKEND_START): cv.time,
+    vol.Required(ATTR_WEEKEND_STOP): cv.time,
+    vol.Required(ATTR_HOME_TEMP): vol.Coerce(float),
+    vol.Required(ATTR_AWAY_TEMP): vol.Coerce(float)
+})
+
+SET_ADVANCED_CONF_SCHEMA = vol.Schema({
+    vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
+    vol.Required(ATTR_LOOP_MODE, default=0): vol.Coerce(int),
+    vol.Required(ATTR_SEN, default=0): vol.Coerce(int),
+    vol.Required(ATTR_OSV, default=42): vol.Coerce(int),
+    vol.Required(ATTR_DIF, default=2): vol.Coerce(int),
+    vol.Required(ATTR_SVH, default=35): vol.Coerce(int),
+    vol.Required(ATTR_SVL, default=5): vol.Coerce(int),
+    vol.Required(ATTR_ADJ, default=0.1): vol.Coerce(float),
+    vol.Required(ATTR_FRE, default=0): vol.Coerce(int),
+    vol.Required(ATTR_PON, default=0): vol.Coerce(int)
+})
+
+_LOGGER = logging.getLogger(__name__)
+
+
+def setup_platform(hass, config, async_add_entities,
+                   discovery_info=None):
+    """Set up the broadlink thermostat platform."""
+    import BroadlinkWifiThermostat
+    wifi_thermostat = BroadlinkWifiThermostat.\
+        Thermostat(config[CONF_MAC],
+                   config[CONF_HOST],
+                   config[CONF_FRIENDLY_NAME],
+                   config[CONF_EXTERNAL_TEMP],
+                   config[CONF_AWAY_TEMP])
+
+    thermostats = [BroadlinkThermostat(wifi_thermostat)]
+
+    async_add_entities(thermostats, True)
+
+    def set_schedule(service):
+        """Handle data for the set_schedule service call."""
+        entity_id = service.data.get(ATTR_ENTITY_ID)
+        if entity_id:
+            target_thermostats = [device for device in thermostats
+                                  if device.entity_id in entity_id]
+        else:
+            target_thermostats = thermostats
+
+        schedules = dict()
+        schedules['week_start_1'] = service.data.get(ATTR_WEEK_START_1)
+        schedules['week_stop_1'] = service.data.get(ATTR_WEEK_STOP_1)
+        schedules['week_start_2'] = service.data.get(ATTR_WEEK_START_2)
+        schedules['week_stop_2'] = service.data.get(ATTR_WEEK_STOP_2)
+        schedules['week_start_3'] = service.data.get(ATTR_WEEK_START_3)
+        schedules['week_stop_3'] = service.data.get(ATTR_WEEK_STOP_3)
+
+        schedules['weekend_start'] = service.data.get(ATTR_WEEKEND_START)
+        schedules['weekend_stop'] = service.data.get(ATTR_WEEKEND_STOP)
+        schedules['away_temp'] = service.data.get(ATTR_AWAY_TEMP)
+        schedules['home_temp'] = service.data.get(ATTR_HOME_TEMP)
+
+        for thermostat in target_thermostats:
+            thermostat.set_schedule(schedules)
+
+    hass.services.register(DOMAIN, SERVICE_SET_SCHEDULE,
+                           set_schedule, schema=SET_SCHEDULE_SCHEMA)
+
+    def set_advanced_conf(service):
+        """Handle data for the set_advanced_conf service call."""
+        entity_id = service.data.get(ATTR_ENTITY_ID)
+        if entity_id:
+            target_thermostats = [device for device in thermostats
+                                  if device.entity_id in entity_id]
+        else:
+            target_thermostats = thermostats
+
+        advance_conf = dict()
+        advance_conf["loop_mode"] = service.data.get(ATTR_LOOP_MODE)
+        advance_conf['sen'] = service.data.get(ATTR_SEN)
+        advance_conf['osv'] = service.data.get(ATTR_OSV)
+        advance_conf['dif'] = service.data.get(ATTR_DIF)
+        advance_conf['svh'] = service.data.get(ATTR_SVH)
+        advance_conf['svl'] = service.data.get(ATTR_SVL)
+        advance_conf['adj'] = service.data.get(ATTR_ADJ)
+        advance_conf['fre'] = service.data.get(ATTR_FRE)
+        advance_conf['pon'] = service.data.get(ATTR_PON)
+        target_thermostats.set_advanced_config(advance_conf)
+
+    hass.services.register(DOMAIN, SERVICE_SET_ADVANCED_CONF,
+                           set_advanced_conf, schema=SET_SCHEDULE_SCHEMA)
+
+
+class BroadlinkThermostat(ClimateDevice):
+    """Representation of a Broadlink Thermostat device."""
+
+    def __init__(self, device):
+        """Initialize the climate device."""
+        self._device = device
+        device.set_time()
+
+    @property
+    def state(self):
+        """Return climate state."""
+        return self._device.state
+
+    @property
+    def supported_features(self):
+        """Return the list of supported features."""
+        return SUPPORT_TARGET_TEMPERATURE | SUPPORT_OPERATION_MODE \
+                                          | SUPPORT_AWAY_MODE
+
+    @property
+    def name(self):
+        """Return the name of the thermostat."""
+        return self._device.name
+
+    @property
+    def temperature_unit(self):
+        """Return the unit of measurement."""
+        return TEMP_CELSIUS
+
+    @property
+    def current_temperature(self):
+        """Return the sensor temperature."""
+        return self._device.current_temp
+
+    @property
+    def target_temperature(self):
+        """Return the temperature we try to reach."""
+        return self._device.target_temperature
+
+    @property
+    def current_operation(self):
+        """Return current operation."""
+        return self._device.current_operation
+
+    @property
+    def operation_list(self):
+        """List of available operation modes."""
+        return [STATE_AUTO, STATE_HEAT, STATE_OFF]
+
+    @property
+    def is_away_mode_on(self):
+        """Return if away mode is on."""
+        return self._device.away
+
+    @property
+    def is_on(self):
+        """Return true if the device is on."""
+        return not self._device.current_operation == STATE_OFF
+
+    @property
+    def precision(self):
+        """Return the precision of the system."""
+        return PRECISION_HALVES
+
+    def set_temperature(self, **kwargs):
+        """Set new target temperature."""
+        if kwargs.get(ATTR_TEMPERATURE) is not None:
+            self._device.set_temperature(kwargs.get(ATTR_TEMPERATURE))
+
+    def set_operation_mode(self, operation_mode):
+        """Set operation mode."""
+        self._device.set_operation_mode(operation_mode)
+
+    def set_schedule(self, schedules):
+        """Set automatic schedule."""
+        self._device.set_schedule(schedules)
+
+    def set_advanced_config(self, advanced_conf):
+        """Set advanced configuration."""
+        self._device.set_advanced_config(self, advanced_conf)
+
+    def turn_away_mode_on(self):
+        """Turn away mode on."""
+        self._device.set_away(True)
+
+    def turn_away_mode_off(self):
+        """Turn away mode off."""
+        self._device.set_away(False)
+
+    def update(self):
+        """Update component data."""
+        self._device.read_status()

requirements_all.txt

@@ -32,6 +32,9 @@ Adafruit-SHT31==1.0.2
 # homeassistant.components.bbb_gpio
 # Adafruit_BBIO==1.0.0
 
+# homeassistant.components.broadlink.broadlink
+BroadlinkWifiThermostat==2.4.1
+
 # homeassistant.components.homekit
 HAP-python==2.4.2