jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 22:57:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make unifiprotect fallback update an eager task (#111580)

Browse Source
This commit is contained in:
J. Nick Koston 2024-02-26 16:30:53 -10:00 committed by GitHub
parent 86ced5a85f
commit bf866732c9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
homeassistant/components/unifiprotect/data.py
View File

@ -2,7 +2,7 @@
from __future__ import annotations from __future__ import annotations
from collections.abc import Callable, Generator, Iterable from collections.abc import Callable, Generator, Iterable
from datetime import timedelta from datetime import datetime, timedelta
import logging import logging
from typing import Any, cast from typing import Any, cast
@ -282,6 +282,16 @@ class ProtectData:
for device in self.get_by_types(DEVICES_THAT_ADOPT): for device in self.get_by_types(DEVICES_THAT_ADOPT):
self._async_signal_device_update(device) self._async_signal_device_update(device)
@callback
def _async_poll(self, now: datetime) -> None:
"""Poll the Protect API.
If the websocket is connected, most of the time
this will be a no-op. If the websocket is disconnected,
this will trigger a reconnect and refresh.
"""
self._hass.async_create_task(self.async_refresh(), eager_start=True)
@callback @callback
def async_subscribe_device_id( def async_subscribe_device_id(
self, mac: str, update_callback: Callable[[ProtectDeviceType], None] self, mac: str, update_callback: Callable[[ProtectDeviceType], None]
@ -289,7 +299,7 @@ class ProtectData:
"""Add an callback subscriber.""" """Add an callback subscriber."""
if not self._subscriptions: if not self._subscriptions:
self._unsub_interval = async_track_time_interval( self._unsub_interval = async_track_time_interval(
self._hass, self.async_refresh, self._update_interval self._hass, self._async_poll, self._update_interval
) )
self._subscriptions.setdefault(mac, []).append(update_callback) self._subscriptions.setdefault(mac, []).append(update_callback)