Project skeleton

custom_components/pv_microinverter/coordinator.py

@@ -0,0 +1,54 @@
+"""Data update coordinator for PV Microinverter integration."""
+
+import logging
+from datetime import timedelta
+
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.update_coordinator import (
+    DataUpdateCoordinator,
+    UpdateFailed,
+)
+
+from .api import PVMicroinverterApiClient, PVMicroinverterApiClientError
+from .const import DOMAIN, PVMicroinverterData
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class PVMicroinverterDataUpdateCoordinator(DataUpdateCoordinator[PVMicroinverterData]):
+    """Class to manage fetching PV Microinverter data."""
+
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        api_client: PVMicroinverterApiClient,
+        update_interval: int,
+    ) -> None:
+        """Initialize the coordinator.
+
+        Args:
+            hass: The Home Assistant instance
+            api_client: The API client
+            update_interval: The update interval in seconds
+        """
+        super().__init__(
+            hass,
+            _LOGGER,
+            name=DOMAIN,
+            update_interval=timedelta(seconds=update_interval),
+        )
+        self.api_client = api_client
+
+    async def _async_update_data(self) -> PVMicroinverterData:
+        """Fetch data from the API.
+
+        Returns:
+            PVMicroinverterData: The fetched data
+
+        Raises:
+            UpdateFailed: If the update fails
+        """
+        try:
+            return await self.api_client.async_get_data()
+        except PVMicroinverterApiClientError as error:
+            raise UpdateFailed(f"Error communicating with API: {error}") from error