Fix discovery on add-on uninstall (#731)

* Fix discovery on add-on uninstall

* Update discovery.py

* Update discovery.py
This commit is contained in:
Pascal Vizeli 2018-09-30 18:24:10 +02:00 committed by GitHub
parent 2872be6385
commit 2f6df3a946
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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