From 4bdb21a87100d275e0f50a124c7265e1bf1ccc2d Mon Sep 17 00:00:00 2001
From: "David F. Mulcahey" <david.mulcahey@icloud.com>
Date: Sat, 22 Dec 2018 12:18:48 -0500
Subject: [PATCH] Update ZHA entity state on ZigBee zdo device announce
 (#19208)

* call async_update if defined on device_announce

* lint

* change update method

* remove unneeded listener
---
 homeassistant/components/zha/entities/entity.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/homeassistant/components/zha/entities/entity.py b/homeassistant/components/zha/entities/entity.py
index 563433c5ff2..c45bd551377 100644
--- a/homeassistant/components/zha/entities/entity.py
+++ b/homeassistant/components/zha/entities/entity.py
@@ -77,6 +77,8 @@ class ZhaEntity(entity.Entity):
         for cluster_id, cluster in self._out_clusters.items():
             cluster.add_listener(self._out_listeners.get(cluster_id, self))
 
+        self._endpoint.device.zdo.add_listener(self)
+
         self._initialized = True
 
     async def async_configure(self):
@@ -172,6 +174,16 @@ class ZhaEntity(entity.Entity):
         """Handle a ZDO command received on this cluster."""
         pass
 
+    @callback
+    def device_announce(self, device):
+        """Handle device_announce zdo event."""
+        self.async_schedule_update_ha_state(force_refresh=True)
+
+    @callback
+    def permit_duration(self, permit_duration):
+        """Handle permit_duration zdo event."""
+        pass
+
     @property
     def device_info(self):
         """Return a device description for device registry."""