From 2f6df3a946de81fe474776d823b8121ff296162a Mon Sep 17 00:00:00 2001 From: Pascal Vizeli Date: Sun, 30 Sep 2018 18:24:10 +0200 Subject: [PATCH] Fix discovery on add-on uninstall (#731) * Fix discovery on add-on uninstall * Update discovery.py * Update discovery.py --- hassio/discovery.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hassio/discovery.py b/hassio/discovery.py index 98fb0e7e6..e5db61fa9 100644 --- a/hassio/discovery.py +++ b/hassio/discovery.py @@ -41,7 +41,7 @@ class Discovery(CoreSysAttributes, JsonConfig): def save(self): """Write discovery message into data file.""" messages = [] - for message in self.message_obj.values(): + for message in self.list_messages: messages.append(attr.asdict(message)) self._data[ATTR_DISCOVERY].clear() @@ -55,7 +55,7 @@ class Discovery(CoreSysAttributes, JsonConfig): @property def list_messages(self): """Return list of available discovery messages.""" - return self.message_obj.values() + return list(self.message_obj.values()) def send(self, addon, service, component, platform, config): """Send a discovery message to Home Assistant."""