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."""