33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
"""DataUpdateCoordinator for integration_blueprint."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
from homeassistant.exceptions import ConfigEntryAuthFailed
|
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
|
|
|
from .api import (
|
|
IntegrationBlueprintApiClientAuthenticationError,
|
|
IntegrationBlueprintApiClientError,
|
|
)
|
|
|
|
if TYPE_CHECKING:
|
|
from .data import IntegrationBlueprintConfigEntry
|
|
|
|
|
|
# https://developers.home-assistant.io/docs/integration_fetching_data#coordinated-single-api-poll-for-data-for-all-entities
|
|
class BlueprintDataUpdateCoordinator(DataUpdateCoordinator):
|
|
"""Class to manage fetching data from the API."""
|
|
|
|
config_entry: IntegrationBlueprintConfigEntry
|
|
|
|
async def _async_update_data(self) -> Any:
|
|
"""Update data via library."""
|
|
try:
|
|
return await self.config_entry.runtime_data.client.async_get_data()
|
|
except IntegrationBlueprintApiClientAuthenticationError as exception:
|
|
raise ConfigEntryAuthFailed(exception) from exception
|
|
except IntegrationBlueprintApiClientError as exception:
|
|
raise UpdateFailed(exception) from exception
|