coordinator.py

Go to the documentation of this file.

"""Coordinator for the scrape component."""
 
from __future__ import annotations
 
from datetime import timedelta
import logging
 
from bs4 import BeautifulSoup
 
from homeassistant.components.rest import RestData
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
_LOGGER = logging.getLogger(__name__)
 
 
class ScrapeCoordinator(DataUpdateCoordinator[BeautifulSoup]):
    """Scrape Coordinator."""
 
    def __init__(
        self,
        hass: HomeAssistant,
        config_entry: ConfigEntry | None,
        rest: RestData,
        update_interval: timedelta,
    ) -> None:
        """Initialize Scrape coordinator."""
        super().__init__(
            hass,
            _LOGGER,
            config_entry=config_entry,
            name="Scrape Coordinator",
            update_interval=update_interval,
        )
        self._rest_rest = rest
 
    async def _async_update_data(self) -> BeautifulSoup:
        """Fetch data from Rest."""
        await self._rest_rest.async_update()
        if (data := self._rest_rest.data) is None:
            raise UpdateFailed("REST data is not available")
        soup = await self.hasshass.async_add_executor_job(BeautifulSoup, data, "lxml")
        _LOGGER.debug("Raw beautiful soup: %s", soup)
        return soup