From aac9f972cfe1bff6b8e067f3b34d0abb143a2381 Mon Sep 17 00:00:00 2001
From: Andrey <andrey-git@users.noreply.github.com>
Date: Sat, 4 Mar 2017 19:13:24 +0200
Subject: [PATCH] Add Zwave refresh services (#6377)

* Add Zwave refresh services

* services file

* Use dispatcher

* Add zwave prefix to signal
---
 homeassistant/components/zwave/__init__.py   | 49 ++++++++++++++++++++
 homeassistant/components/zwave/const.py      |  2 +
 homeassistant/components/zwave/services.yaml | 14 ++++++
 3 files changed, 65 insertions(+)

diff --git a/homeassistant/components/zwave/__init__.py b/homeassistant/components/zwave/__init__.py
index 30cf16fc9e5..5e651f69213 100755
--- a/homeassistant/components/zwave/__init__.py
+++ b/homeassistant/components/zwave/__init__.py
@@ -23,6 +23,8 @@ from homeassistant.helpers.event import track_time_change
 from homeassistant.util import convert, slugify
 import homeassistant.config as conf_util
 import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.dispatcher import (
+    async_dispatcher_connect, async_dispatcher_send)
 
 from . import const
 from . import workaround
@@ -162,6 +164,10 @@ NODE_SERVICE_SCHEMA = vol.Schema({
     vol.Required(const.ATTR_NODE_ID): vol.Coerce(int),
 })
 
+REFRESH_ENTITY_SCHEMA = vol.Schema({
+    vol.Required(ATTR_ENTITY_ID): cv.entity_id,
+})
+
 CHANGE_ASSOCIATION_SCHEMA = vol.Schema({
     vol.Required(const.ATTR_ASSOCIATION): cv.string,
     vol.Required(const.ATTR_NODE_ID): vol.Coerce(int),
@@ -185,6 +191,8 @@ DEVICE_CONFIG_SCHEMA_ENTRY = vol.Schema({
         cv.positive_int
 })
 
+SIGNAL_REFRESH_ENTITY_FORMAT = 'zwave_refresh_entity_{}'
+
 CONFIG_SCHEMA = vol.Schema({
     DOMAIN: vol.Schema({
         vol.Optional(CONF_AUTOHEAL, default=DEFAULT_CONF_AUTOHEAL): cv.boolean,
@@ -615,6 +623,19 @@ def setup(hass, config):
                          "target node:%s, instance=%s", node_id, group,
                          target_node_id, instance)
 
+    @asyncio.coroutine
+    def async_refresh_entity(service):
+        """Refresh values that specific entity depends on."""
+        entity_id = service.data.get(ATTR_ENTITY_ID)
+        async_dispatcher_send(
+            hass, SIGNAL_REFRESH_ENTITY_FORMAT.format(entity_id))
+
+    def refresh_node(service):
+        """Refresh all node info."""
+        node_id = service.data.get(const.ATTR_NODE_ID)
+        node = NETWORK.nodes[node_id]
+        node.refresh_info()
+
     def start_zwave(_service_or_event):
         """Startup Z-Wave network."""
         _LOGGER.info("Starting ZWave network.")
@@ -709,6 +730,16 @@ def setup(hass, config):
                                descriptions[
                                    const.SERVICE_PRINT_NODE],
                                schema=NODE_SERVICE_SCHEMA)
+        hass.services.register(DOMAIN, const.SERVICE_REFRESH_ENTITY,
+                               async_refresh_entity,
+                               descriptions[
+                                   const.SERVICE_REFRESH_ENTITY],
+                               schema=REFRESH_ENTITY_SCHEMA)
+        hass.services.register(DOMAIN, const.SERVICE_REFRESH_NODE,
+                               refresh_node,
+                               descriptions[
+                                   const.SERVICE_REFRESH_NODE],
+                               schema=NODE_SERVICE_SCHEMA)
 
     # Setup autoheal
     if autoheal:
@@ -788,6 +819,14 @@ class ZWaveDeviceEntity(Entity):
         """
         return []
 
+    @asyncio.coroutine
+    def async_added_to_hass(self):
+        """Add device to dict."""
+        async_dispatcher_connect(
+            self.hass,
+            SIGNAL_REFRESH_ENTITY_FORMAT.format(self.entity_id),
+            self.refresh_from_network)
+
     def _get_dependent_value_ids(self):
         """Return a list of value_ids this device depend on.
 
@@ -867,3 +906,13 @@ class ZWaveDeviceEntity(Entity):
             attrs[ATTR_POWER] = self.power_consumption
 
         return attrs
+
+    def refresh_from_network(self):
+        """Refresh all dependent values from zwave network."""
+        dependent_ids = self._get_dependent_value_ids()
+        if dependent_ids is None:
+            # Entity depends on the whole node
+            self._value.node.refresh_info()
+            return
+        for value_id in dependent_ids + [self._value.value_id]:
+            self._value.node.refresh_value(value_id)
diff --git a/homeassistant/components/zwave/const.py b/homeassistant/components/zwave/const.py
index 52ccdc0a752..ab4c7604dc4 100644
--- a/homeassistant/components/zwave/const.py
+++ b/homeassistant/components/zwave/const.py
@@ -34,6 +34,8 @@ SERVICE_SET_WAKEUP = "set_wakeup"
 SERVICE_STOP_NETWORK = "stop_network"
 SERVICE_START_NETWORK = "start_network"
 SERVICE_RENAME_NODE = "rename_node"
+SERVICE_REFRESH_ENTITY = "refresh_entity"
+SERVICE_REFRESH_NODE = "refresh_node"
 
 EVENT_SCENE_ACTIVATED = "zwave.scene_activated"
 EVENT_NODE_EVENT = "zwave.node_event"
diff --git a/homeassistant/components/zwave/services.yaml b/homeassistant/components/zwave/services.yaml
index 08cd8069d83..00881602dc0 100644
--- a/homeassistant/components/zwave/services.yaml
+++ b/homeassistant/components/zwave/services.yaml
@@ -65,6 +65,20 @@ print_node:
     node_id:
       description: Node id of the device to print.
 
+refresh_entity:
+  description: Refresh zwave entity.
+  fields:
+    entity_id:
+      description: Name of the entity to refresh.
+      example: 'light.leviton_vrmx11lz_multilevel_scene_switch_level_40'
+
+refresh_node:
+  description: Refresh zwave node.
+  fields:
+    entity_id:
+      description: ID of the node to refresh.
+      example: 10
+
 set_wakeup:
   description: Sets wake-up interval of a node.
   fields: