Store SB data Class. (#59266)

This commit is contained in:
RenierM26 2021-11-09 08:00:51 +02:00 committed by GitHub
parent 4e59e6c9a7
commit 012e005e0d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -30,6 +30,7 @@ class SwitchbotDataUpdateCoordinator(DataUpdateCoordinator):
) -> None:
"""Initialize global switchbot data updater."""
self.switchbot_api = api
self.switchbot_data = self.switchbot_api.GetSwitchbotDevices()
self.retry_count = retry_count
self.scan_timeout = scan_timeout
self.update_interval = timedelta(seconds=update_interval)
@ -43,7 +44,7 @@ class SwitchbotDataUpdateCoordinator(DataUpdateCoordinator):
def _update_data(self) -> dict | None:
"""Fetch device states from switchbot api."""
return self.switchbot_api.GetSwitchbotDevices().discover(
return self.switchbot_data.discover(
retry=self.retry_count, scan_timeout=self.scan_timeout
)