More fault tolerant discovery

2016-05-08 21:23:03 -07:00 · 2016-05-08 21:23:03 -07:00 · 09483e3be4
parent ab2e85840f
commit 09483e3be4
1 changed files with 1 additions and 1 deletions
--- a/homeassistant/components/discovery.py
+++ b/homeassistant/components/discovery.py
@ -52,7 +52,7 @@ def listen(hass, service, callback):

    def discovery_event_listener(event):
        """Listen for discovery events."""
-        if event.data[ATTR_SERVICE] in service:
+        if ATTR_SERVICE in event.data and event.data[ATTR_SERVICE] in service:
            callback(event.data[ATTR_SERVICE], event.data.get(ATTR_DISCOVERED))

    hass.bus.listen(EVENT_PLATFORM_DISCOVERED, discovery_event_listener)