coordinator.py

Go to the documentation of this file.

"""The Minecraft Server integration."""
 
from __future__ import annotations
 
from datetime import timedelta
import logging
 
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
from .api import (
    MinecraftServer,
    MinecraftServerConnectionError,
    MinecraftServerData,
    MinecraftServerNotInitializedError,
)
 
SCAN_INTERVAL = timedelta(seconds=60)
 
_LOGGER = logging.getLogger(__name__)
 
 
class MinecraftServerCoordinator(DataUpdateCoordinator[MinecraftServerData]):
    """Minecraft Server data update coordinator."""
 
    def __init__(self, hass: HomeAssistant, name: str, api: MinecraftServer) -> None:
        """Initialize coordinator instance."""
        self._api_api = api
 
        super().__init__(
            hass=hass,
            name=name,
            logger=_LOGGER,
            update_interval=SCAN_INTERVAL,
        )
 
    async def _async_update_data(self) -> MinecraftServerData:
        """Get updated data from the server."""
        try:
            return await self._api_api.async_get_data()
        except (
            MinecraftServerConnectionError,
            MinecraftServerNotInitializedError,
        ) as error:
            raise UpdateFailed(error) from error