Initial commit

custom_components/oasis_mini/coordinator.py

@@ -0,0 +1,47 @@
+"""Oasis Mini coordinator."""
+
+from __future__ import annotations
+
+from datetime import datetime, timedelta
+import logging
+
+import async_timeout
+
+from homeassistant.core import HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+from .const import DOMAIN
+from .pyoasismini import OasisMini
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class OasisMiniCoordinator(DataUpdateCoordinator[str]):
+    """Oasis Mini data update coordinator."""
+
+    last_updated: datetime | None = None
+
+    def __init__(self, hass: HomeAssistant, device: OasisMini) -> None:
+        """Initialize."""
+        super().__init__(
+            hass, _LOGGER, name=DOMAIN, update_interval=timedelta(seconds=10)
+        )
+        self.device = device
+
+    async def _async_update_data(self):
+        try:
+            async with async_timeout.timeout(10):
+                if not self.device.serial_number:
+                    await self.device.async_get_serial_number()
+                if not self.device.software_version:
+                    await self.device.async_get_software_version()
+                data = await self.device.async_get_status()
+                await self.device.async_get_current_track_details()
+        except Exception as ex:
+            raise UpdateFailed("Couldn't read oasis_mini") from ex
+        if data is None:
+            raise ConfigEntryAuthFailed
+        if data != self.data:
+            self.last_updated = datetime.now()
+        return data