From 0b5191a24700549d1003a2a1e455022c04c2f747 Mon Sep 17 00:00:00 2001
From: Andrey <andrey-git@users.noreply.github.com>
Date: Fri, 10 Feb 2017 18:54:48 +0200
Subject: [PATCH] Add support for zwave wakeup setting. (#5849)

* Add support for zwave wakeup setting.

* rename wakeup
---
 homeassistant/components/zwave/__init__.py   | 41 +++++++++++++++++++-
 homeassistant/components/zwave/const.py      |  1 +
 homeassistant/components/zwave/services.yaml |  8 ++++
 homeassistant/const.py                       |  1 +
 4 files changed, 49 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/zwave/__init__.py b/homeassistant/components/zwave/__init__.py
index 867a0f96062..2abc79754ae 100755
--- a/homeassistant/components/zwave/__init__.py
+++ b/homeassistant/components/zwave/__init__.py
@@ -13,8 +13,9 @@ import voluptuous as vol
 
 from homeassistant.helpers import discovery, customize
 from homeassistant.const import (
-    ATTR_BATTERY_LEVEL, ATTR_LOCATION, ATTR_ENTITY_ID, CONF_CUSTOMIZE,
-    EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP, CONF_ENTITY_ID)
+    ATTR_BATTERY_LEVEL, ATTR_LOCATION, ATTR_ENTITY_ID, ATTR_WAKEUP,
+    CONF_CUSTOMIZE, EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP,
+    CONF_ENTITY_ID)
 from homeassistant.helpers.entity import Entity
 from homeassistant.helpers.event import track_time_change
 from homeassistant.util import convert, slugify
@@ -152,6 +153,12 @@ CHANGE_ASSOCIATION_SCHEMA = vol.Schema({
     vol.Optional(const.ATTR_INSTANCE, default=0x00): vol.Coerce(int)
 })
 
+SET_WAKEUP_SCHEMA = vol.Schema({
+    vol.Required(const.ATTR_NODE_ID): vol.Coerce(int),
+    vol.Required(const.ATTR_CONFIG_VALUE):
+        vol.All(vol.Coerce(int), cv.positive_int),
+})
+
 _ZWAVE_CUSTOMIZE_SCHEMA_ENTRY = vol.Schema({
     vol.Required(CONF_ENTITY_ID): cv.match_all,
     vol.Optional(CONF_POLLING_INTENSITY): cv.positive_int,
@@ -242,6 +249,14 @@ def get_config_value(node, value_index):
         return get_config_value(node, value_index)
 
 
+def _get_wakeup(node):
+    """Return wakeup interval of the node or None if node is not wakable."""
+    if node.can_wake_up():
+        for value_id in node.get_values(class_id=const.COMMAND_CLASS_WAKE_UP):
+            return node.values[value_id].data
+    return None
+
+
 # pylint: disable=R0914
 def setup(hass, config):
     """Setup Z-Wave.
@@ -503,6 +518,19 @@ def setup(hass, config):
         _LOGGER.info("Config parameter %s on Node %s : %s",
                      param, node_id, get_config_value(node, param))
 
+    def set_wakeup(service):
+        """Set wake-up interval of a node."""
+        node_id = service.data.get(const.ATTR_NODE_ID)
+        node = NETWORK.nodes[node_id]
+        value = service.data.get(const.ATTR_CONFIG_VALUE)
+        if node.can_wake_up():
+            for value_id in node.get_values(
+                    class_id=const.COMMAND_CLASS_WAKE_UP):
+                node.values[value_id].data = value
+                _LOGGER.info("Node %s wake-up set to %d", node_id, value)
+        else:
+            _LOGGER.info("Node %s is not wakeable", node_id)
+
     def change_association(service):
         """Change an association in the zwave network."""
         association_type = service.data.get(const.ATTR_ASSOCIATION)
@@ -598,6 +626,11 @@ def setup(hass, config):
                                descriptions[
                                    const.SERVICE_CHANGE_ASSOCIATION],
                                schema=CHANGE_ASSOCIATION_SCHEMA)
+        hass.services.register(DOMAIN, const.SERVICE_SET_WAKEUP,
+                               set_wakeup,
+                               descriptions[
+                                   const.SERVICE_SET_WAKEUP],
+                               schema=SET_WAKEUP_SCHEMA)
 
     # Setup autoheal
     if autoheal:
@@ -727,4 +760,8 @@ class ZWaveDeviceEntity(Entity):
         if location:
             attrs[ATTR_LOCATION] = location
 
+        wakeup = _get_wakeup(self._value.node)
+        if wakeup:
+            attrs[ATTR_WAKEUP] = wakeup
+
         return attrs
diff --git a/homeassistant/components/zwave/const.py b/homeassistant/components/zwave/const.py
index 2350b4bee75..8e28d924cfa 100644
--- a/homeassistant/components/zwave/const.py
+++ b/homeassistant/components/zwave/const.py
@@ -25,6 +25,7 @@ SERVICE_SOFT_RESET = "soft_reset"
 SERVICE_TEST_NETWORK = "test_network"
 SERVICE_SET_CONFIG_PARAMETER = "set_config_parameter"
 SERVICE_PRINT_CONFIG_PARAMETER = "print_config_parameter"
+SERVICE_SET_WAKEUP = "set_wakeup"
 SERVICE_STOP_NETWORK = "stop_network"
 SERVICE_START_NETWORK = "start_network"
 SERVICE_RENAME_NODE = "rename_node"
diff --git a/homeassistant/components/zwave/services.yaml b/homeassistant/components/zwave/services.yaml
index 2a44096ce0e..c1105eaec18 100644
--- a/homeassistant/components/zwave/services.yaml
+++ b/homeassistant/components/zwave/services.yaml
@@ -47,6 +47,14 @@ print_config_parameter:
     parameter:
       description: Parameter number to print (integer).
 
+set_wakeup:
+  description: Sets wake-up interval of a node.
+  fields:
+    node_id:
+      description: Node id of the device to set the wake-up interval for. (integer)
+    value:
+      description: Value of the interval to set. (integer)
+
 start_network:
   description: Start the Z-Wave network. This might take a while, depending on how big your Z-Wave network is.
 
diff --git a/homeassistant/const.py b/homeassistant/const.py
index eda303a552e..5df9bfe15f1 100644
--- a/homeassistant/const.py
+++ b/homeassistant/const.py
@@ -253,6 +253,7 @@ ATTR_DISCOVERED = 'discovered'
 ATTR_LOCATION = 'location'
 
 ATTR_BATTERY_LEVEL = 'battery_level'
+ATTR_WAKEUP = 'wake_up_interval'
 
 # For devices which support a code attribute
 ATTR_CODE = 'code'