Better teardown and setup of Roborock connections (#106092)

Co-authored-by: Robert Resch <robert@resch.dev>
This commit is contained in:
Luke Lashley 2024-02-12 03:37:37 -05:00 committed by GitHub
parent 2516eafba6
commit 33cdcce191
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 47 additions and 27 deletions

View file

@ -232,10 +232,8 @@ class RoborockSensorEntity(RoborockCoordinatedEntity, SensorEntity):
description: RoborockSensorDescription,
) -> None:
"""Initialize the entity."""
super().__init__(unique_id, coordinator)
self.entity_description = description
if (protocol := self.entity_description.protocol_listener) is not None:
self.api.add_listener(protocol, self._update_from_listener, self.api.cache)
super().__init__(unique_id, coordinator, description.protocol_listener)
@property
def native_value(self) -> StateType | datetime.datetime: