From e75f9b36f907b747bc364fdee86bdd14340561d0 Mon Sep 17 00:00:00 2001
From: Antoine Meillet <antoine.meillet@gmail.com>
Date: Sat, 10 Nov 2018 21:08:03 +0100
Subject: [PATCH] add heartbeat support to mysensors (#18359)

---
 homeassistant/components/mysensors/device.py  | 2 ++
 homeassistant/components/mysensors/handler.py | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/homeassistant/components/mysensors/device.py b/homeassistant/components/mysensors/device.py
index 7f4f6100204..07261b1c2a6 100644
--- a/homeassistant/components/mysensors/device.py
+++ b/homeassistant/components/mysensors/device.py
@@ -15,6 +15,7 @@ ATTR_CHILD_ID = 'child_id'
 ATTR_DESCRIPTION = 'description'
 ATTR_DEVICE = 'device'
 ATTR_NODE_ID = 'node_id'
+ATTR_HEARTBEAT = 'heartbeat'
 MYSENSORS_PLATFORM_DEVICES = 'mysensors_devices_{}'
 
 
@@ -51,6 +52,7 @@ class MySensorsDevice:
         child = node.children[self.child_id]
         attr = {
             ATTR_BATTERY_LEVEL: node.battery_level,
+            ATTR_HEARTBEAT: node.heartbeat,
             ATTR_CHILD_ID: self.child_id,
             ATTR_DESCRIPTION: child.description,
             ATTR_DEVICE: self.gateway.device,
diff --git a/homeassistant/components/mysensors/handler.py b/homeassistant/components/mysensors/handler.py
index 3403c589639..39af1173706 100644
--- a/homeassistant/components/mysensors/handler.py
+++ b/homeassistant/components/mysensors/handler.py
@@ -45,6 +45,12 @@ async def handle_battery_level(hass, hass_config, msg):
     _handle_node_update(hass, msg)
 
 
+@HANDLERS.register('I_HEARTBEAT_RESPONSE')
+async def handle_heartbeat(hass, hass_config, msg):
+    """Handle an heartbeat."""
+    _handle_node_update(hass, msg)
+
+
 @HANDLERS.register('I_SKETCH_NAME')
 async def handle_sketch_name(hass, hass_config, msg):
     """Handle an internal sketch name message."""