From f946115cbb573dfd358c62f1f232479af37596f0 Mon Sep 17 00:00:00 2001 From: Luis Miranda <161006+luuuis@users.noreply.github.com> Date: Tue, 2 Apr 2024 17:09:18 +0100 Subject: [PATCH] fix: introduce OMIEDailyCoordinator.__job --- custom_components/omie/coordinator.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/custom_components/omie/coordinator.py b/custom_components/omie/coordinator.py index 18ab85e..f304003 100644 --- a/custom_components/omie/coordinator.py +++ b/custom_components/omie/coordinator.py @@ -72,6 +72,10 @@ def __init__(self, delay_μs = random.randint(0, _SCHEDULE_MAX_DELAY.seconds * 10 ** 6) self._schedule_second = delay_μs // 10 ** 6 self._schedule_microsecond = delay_μs % 10 ** 6 + self.__job = HassJob( + self._handle_refresh_interval, + f'OMIEDailyCoordinator {name}', + job_type=HassJobType.Coroutinefunction) async def _async_update_data(self) -> OMIEModel | None: if self._wait_for_none_before(): @@ -114,11 +118,7 @@ def _omie_schedule_refresh(self): _LOGGER.debug("%s: _schedule_refresh scheduling an update at %s (none_before=%s, next_hour=%s)", self.name, next_refresh, none_before, next_hour) - self._unsub_refresh = event.async_track_point_in_utc_time(self.hass, self.__omie_handle_refresh_interval, next_refresh) - - @callback - def __omie_handle_refresh_interval(self, _now: datetime | None = None) -> None: - self.hass.async_create_task(self._handle_refresh_interval(), eager_start=True) + self._unsub_refresh = event.async_track_point_in_utc_time(self.hass, self.__job, next_refresh) def _wait_for_none_before(self) -> bool: """Whether the coordinator should wait for `none_before`."""