Slow down requests for device data

Author: cdpuk

custom_components/bestway/coordinator.py

@@ -1,7 +1,7 @@
 """Data update coordinator for the Bestway API."""
 
 import asyncio
-from datetime import timedelta
+from datetime import datetime, timedelta
 from logging import getLogger
 
 from homeassistant.core import HomeAssistant
@@ -10,6 +10,7 @@
 from .bestway.api import BestwayApi, BestwayApiResults
 
 _LOGGER = getLogger(__name__)
+_BINDINGS_REFRESH_INTERVAL = timedelta(minutes=10)
 
 
 class BestwayUpdateCoordinator(DataUpdateCoordinator[BestwayApiResults]):
@@ -24,6 +25,7 @@ def __init__(self, hass: HomeAssistant, api: BestwayApi) -> None:
             update_interval=timedelta(seconds=30),
         )
         self.api = api
+        self.last_bindings_refresh = datetime.min
 
     async def _async_update_data(self) -> BestwayApiResults:
         """Fetch data from API endpoint.
@@ -32,5 +34,9 @@ async def _async_update_data(self) -> BestwayApiResults:
         so entities can quickly look up their data.
         """
         async with asyncio.timeout(10):
-            await self.api.refresh_bindings()
+            # Refresh the device list at a slower rate
+            # This may help with rate limiting
+            if self.last_bindings_refresh + _BINDINGS_REFRESH_INTERVAL < datetime.now():
+                await self.api.refresh_bindings()
+                self.last_bindings_refresh = datetime.now()
             return await self.api.fetch_data()

custom_components/bestway/entity.py

@@ -1,6 +1,7 @@
 """Home Assistant entity descriptions."""
 
 from __future__ import annotations
+from datetime import datetime, timedelta
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.helpers.entity import DeviceInfo
@@ -10,6 +11,8 @@
 from .bestway.model import BestwayDevice, BestwayDeviceStatus
 from .const import DOMAIN
 
+_OFFLINE_DEVICE_TIMEOUT = timedelta(minutes=5)
+
 
 class BestwayEntity(CoordinatorEntity[BestwayUpdateCoordinator]):
     """Bestway base entity type."""
@@ -55,8 +58,9 @@ def status(self) -> BestwayDeviceStatus | None:
     @property
     def available(self) -> bool:
         """Return True if entity is available."""
+        offline_cutoff = int((datetime.now() - _OFFLINE_DEVICE_TIMEOUT).timestamp())
         return (
             self.coordinator.last_update_success
-            and self.bestway_device is not None
-            and self.bestway_device.is_online
+            and self.status is not None
+            and self.status.timestamp > offline_cutoff
         )