coordinator.py

Go to the documentation of this file.

"""Define an object to manage fetching NYT Games data."""
 
from __future__ import annotations
 
from dataclasses import dataclass
from datetime import timedelta
from typing import TYPE_CHECKING
 
from nyt_games import Connections, NYTGamesClient, NYTGamesError, SpellingBee, Wordle
 
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
from .const import LOGGER
 
if TYPE_CHECKING:
    from . import NYTGamesConfigEntry
 
 
@dataclass
class NYTGamesData:
    """Class for NYT Games data."""
 
    wordle: Wordle
    spelling_bee: SpellingBee | None
    connections: Connections | None
 
 
class NYTGamesCoordinator(DataUpdateCoordinator[NYTGamesData]):
    """Class to manage fetching NYT Games data."""
 
    config_entry: NYTGamesConfigEntry
 
    def __init__(self, hass: HomeAssistant, client: NYTGamesClient) -> None:
        """Initialize coordinator."""
        super().__init__(
            hass,
            logger=LOGGER,
            name="NYT Games",
            update_interval=timedelta(minutes=15),
        )
        self.clientclient = client
 
    async def _async_update_data(self) -> NYTGamesData:
        try:
            stats_data = await self.clientclient.get_latest_stats()
            connections_data = await self.clientclient.get_connections()
        except NYTGamesError as error:
            raise UpdateFailed(error) from error
        return NYTGamesData(
            wordle=stats_data.wordle,
            spelling_bee=stats_data.spelling_bee,
            connections=connections_data,
        )