coordinator.py

Go to the documentation of this file.

"""Coordinator for the Environment Canada (EC) component."""
 
import logging
import xml.etree.ElementTree as ET
 
from env_canada import ec_exc
 
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
from .const import DOMAIN
 
_LOGGER = logging.getLogger(__name__)
 
 
class ECDataUpdateCoordinator(DataUpdateCoordinator):
    """Class to manage fetching EC data."""
 
    def __init__(self, hass, ec_data, name, update_interval):
        """Initialize global EC data updater."""
        super().__init__(
            hass, _LOGGER, name=f"{DOMAIN} {name}", update_interval=update_interval
        )
        self.ec_dataec_data = ec_data
        self.last_update_successlast_update_successlast_update_success = False
 
    async def _async_update_data(self):
        """Fetch data from EC."""
        try:
            await self.ec_dataec_data.update()
        except (ET.ParseError, ec_exc.UnknownStationId) as ex:
            raise UpdateFailed(f"Error fetching {self.name} data: {ex}") from ex
        return self.ec_dataec_data