Add new tomorrow.io integration to replace Climacell - Part 1/3

CODEOWNERS

@@ -533,6 +533,7 @@ homeassistant/components/tile/* @bachya
 homeassistant/components/time_date/* @fabaff
 homeassistant/components/tmb/* @alemuro
 homeassistant/components/todoist/* @boralyl
+homeassistant/components/tomorrowio/* @raman325
 homeassistant/components/toon/* @frenck
 homeassistant/components/totalconnect/* @austinmroczek
 homeassistant/components/tplink/* @rytilahti @thegardenmonkey

homeassistant/components/tomorrowio/__init__.py

@@ -0,0 +1,268 @@
+"""The Tomorrow.io integration."""
+from __future__ import annotations
+
+from datetime import timedelta
+import logging
+from math import ceil
+from typing import Any
+
+from pytomorrowio import TomorrowioV4
+from pytomorrowio.const import CURRENT, FORECASTS
+from pytomorrowio.exceptions import (
+    CantConnectException,
+    InvalidAPIKeyException,
+    RateLimitedException,
+    UnknownException,
+)
+
+from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
+from homeassistant.components.weather import DOMAIN as WEATHER_DOMAIN
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE, CONF_NAME
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+    DataUpdateCoordinator,
+    UpdateFailed,
+)
+
+from .const import (
+    ATTRIBUTION,
+    CONF_TIMESTEP,
+    DOMAIN,
+    MAX_REQUESTS_PER_DAY,
+    TMRW_ATTR_CARBON_MONOXIDE,
+    TMRW_ATTR_CHINA_AQI,
+    TMRW_ATTR_CHINA_HEALTH_CONCERN,
+    TMRW_ATTR_CHINA_PRIMARY_POLLUTANT,
+    TMRW_ATTR_CLOUD_BASE,
+    TMRW_ATTR_CLOUD_CEILING,
+    TMRW_ATTR_CLOUD_COVER,
+    TMRW_ATTR_CONDITION,
+    TMRW_ATTR_DEW_POINT,
+    TMRW_ATTR_EPA_AQI,
+    TMRW_ATTR_EPA_HEALTH_CONCERN,
+    TMRW_ATTR_EPA_PRIMARY_POLLUTANT,
+    TMRW_ATTR_FEELS_LIKE,
+    TMRW_ATTR_FIRE_INDEX,
+    TMRW_ATTR_HUMIDITY,
+    TMRW_ATTR_NITROGEN_DIOXIDE,
+    TMRW_ATTR_OZONE,
+    TMRW_ATTR_PARTICULATE_MATTER_10,
+    TMRW_ATTR_PARTICULATE_MATTER_25,
+    TMRW_ATTR_POLLEN_GRASS,
+    TMRW_ATTR_POLLEN_TREE,
+    TMRW_ATTR_POLLEN_WEED,
+    TMRW_ATTR_PRECIPITATION,
+    TMRW_ATTR_PRECIPITATION_PROBABILITY,
+    TMRW_ATTR_PRECIPITATION_TYPE,
+    TMRW_ATTR_PRESSURE,
+    TMRW_ATTR_PRESSURE_SURFACE_LEVEL,
+    TMRW_ATTR_SOLAR_GHI,
+    TMRW_ATTR_SULPHUR_DIOXIDE,
+    TMRW_ATTR_TEMPERATURE,
+    TMRW_ATTR_TEMPERATURE_HIGH,
+    TMRW_ATTR_TEMPERATURE_LOW,
+    TMRW_ATTR_VISIBILITY,
+    TMRW_ATTR_WIND_DIRECTION,
+    TMRW_ATTR_WIND_GUST,
+    TMRW_ATTR_WIND_SPEED,
+)
+
+_LOGGER = logging.getLogger(__name__)
+
+PLATFORMS = [SENSOR_DOMAIN, WEATHER_DOMAIN]
+
+
+def _set_update_interval(hass: HomeAssistant, current_entry: ConfigEntry) -> timedelta:
+    """Recalculate update_interval based on existing Tomorrow.io instances and update them."""
+    api_calls = 2
+    # We check how many Tomorrow.io configured instances are using the same API key and
+    # calculate interval to not exceed allowed numbers of requests. Divide 90% of
+    # MAX_REQUESTS_PER_DAY by the number of API calls because we want a buffer in the
+    # number of API calls left at the end of the day.
+    other_instance_entry_ids = [
+        entry.entry_id
+        for entry in hass.config_entries.async_entries(DOMAIN)
+        if entry.entry_id != current_entry.entry_id
+        and entry.data[CONF_API_KEY] == current_entry.data[CONF_API_KEY]
+    ]
+
+    interval = timedelta(
+        minutes=(
+            ceil(
+                (24 * 60 * (len(other_instance_entry_ids) + 1) * api_calls)
+                / (MAX_REQUESTS_PER_DAY * 0.9)
+            )
+        )
+    )
+
+    for entry_id in other_instance_entry_ids:
+        if entry_id in hass.data[DOMAIN]:
+            hass.data[DOMAIN][entry_id].update_interval = interval
+
+    return interval
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
+    """Set up Tomorrow.io API from a config entry."""
+    hass.data.setdefault(DOMAIN, {})
+
+    api = TomorrowioV4(
+        entry.data[CONF_API_KEY],
+        entry.data[CONF_LATITUDE],
+        entry.data[CONF_LONGITUDE],
+        session=async_get_clientsession(hass),
+    )
+
+    coordinator = TomorrowioDataUpdateCoordinator(
+        hass,
+        entry,
+        api,
+        _set_update_interval(hass, entry),
+    )
+
+    await coordinator.async_config_entry_first_refresh()
+
+    hass.data[DOMAIN][entry.entry_id] = coordinator
+
+    hass.config_entries.async_setup_platforms(entry, PLATFORMS)
+
+    return True
+
+
+async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
+    """Unload a config entry."""
+    unload_ok = await hass.config_entries.async_unload_platforms(
+        config_entry, PLATFORMS
+    )
+
+    hass.data[DOMAIN].pop(config_entry.entry_id)
+    if not hass.data[DOMAIN]:
+        hass.data.pop(DOMAIN)
+
+    return unload_ok
+
+
+class TomorrowioDataUpdateCoordinator(DataUpdateCoordinator):
+    """Define an object to hold Tomorrow.io data."""
+
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        config_entry: ConfigEntry,
+        api: TomorrowioV4,
+        update_interval: timedelta,
+    ) -> None:
+        """Initialize."""
+
+        self._config_entry = config_entry
+        self._api = api
+        self.name = config_entry.data[CONF_NAME]
+        self.data = {CURRENT: {}, FORECASTS: {}}
+
+        super().__init__(
+            hass,
+            _LOGGER,
+            name=config_entry.data[CONF_NAME],
+            update_interval=update_interval,
+        )
+
+    async def _async_update_data(self) -> dict[str, Any]:
+        """Update data via library."""
+        try:
+            return await self._api.realtime_and_all_forecasts(
+                [
+                    TMRW_ATTR_TEMPERATURE,
+                    TMRW_ATTR_HUMIDITY,
+                    TMRW_ATTR_PRESSURE,
+                    TMRW_ATTR_WIND_SPEED,
+                    TMRW_ATTR_WIND_DIRECTION,
+                    TMRW_ATTR_CONDITION,
+                    TMRW_ATTR_VISIBILITY,
+                    TMRW_ATTR_OZONE,
+                    TMRW_ATTR_WIND_GUST,
+                    TMRW_ATTR_CLOUD_COVER,
+                    TMRW_ATTR_PRECIPITATION_TYPE,
+                    *(
+                        TMRW_ATTR_CARBON_MONOXIDE,
+                        TMRW_ATTR_CHINA_AQI,
+                        TMRW_ATTR_CHINA_HEALTH_CONCERN,
+                        TMRW_ATTR_CHINA_PRIMARY_POLLUTANT,
+                        TMRW_ATTR_CLOUD_BASE,
+                        TMRW_ATTR_CLOUD_CEILING,
+                        TMRW_ATTR_CLOUD_COVER,
+                        TMRW_ATTR_DEW_POINT,
+                        TMRW_ATTR_EPA_AQI,
+                        TMRW_ATTR_EPA_HEALTH_CONCERN,
+                        TMRW_ATTR_EPA_PRIMARY_POLLUTANT,
+                        TMRW_ATTR_FEELS_LIKE,
+                        TMRW_ATTR_FIRE_INDEX,
+                        TMRW_ATTR_NITROGEN_DIOXIDE,
+                        TMRW_ATTR_OZONE,
+                        TMRW_ATTR_PARTICULATE_MATTER_10,
+                        TMRW_ATTR_PARTICULATE_MATTER_25,
+                        TMRW_ATTR_POLLEN_GRASS,
+                        TMRW_ATTR_POLLEN_TREE,
+                        TMRW_ATTR_POLLEN_WEED,
+                        TMRW_ATTR_PRECIPITATION_TYPE,
+                        TMRW_ATTR_PRESSURE_SURFACE_LEVEL,
+                        TMRW_ATTR_SOLAR_GHI,
+                        TMRW_ATTR_SULPHUR_DIOXIDE,
+                        TMRW_ATTR_WIND_GUST,
+                    ),
+                ],
+                [
+                    TMRW_ATTR_TEMPERATURE_LOW,
+                    TMRW_ATTR_TEMPERATURE_HIGH,
+                    TMRW_ATTR_WIND_SPEED,
+                    TMRW_ATTR_WIND_DIRECTION,
+                    TMRW_ATTR_CONDITION,
+                    TMRW_ATTR_PRECIPITATION,
+                    TMRW_ATTR_PRECIPITATION_PROBABILITY,
+                ],
+                nowcast_timestep=self._config_entry.options[CONF_TIMESTEP],
+            )
+        except (
+            CantConnectException,
+            InvalidAPIKeyException,
+            RateLimitedException,
+            UnknownException,
+        ) as error:
+            raise UpdateFailed from error
+
+
+class TomorrowioEntity(CoordinatorEntity):
+    """Base Tomorrow.io Entity."""
+
+    def __init__(
+        self,
+        config_entry: ConfigEntry,
+        coordinator: TomorrowioDataUpdateCoordinator,
+        api_version: int,
+    ) -> None:
+        """Initialize Tomorrow.io Entity."""
+        super().__init__(coordinator)
+        self.api_version = api_version
+        self._config_entry = config_entry
+        self._attr_device_info = {
+            "identifiers": {(DOMAIN, self._config_entry.data[CONF_API_KEY])},
+            "name": "Tomorrow.io",
+            "manufacturer": "Tomorrow.io",
+            "sw_version": f"v{self.api_version}",
+            "entry_type": "service",
+        }
+
+    def _get_current_property(self, property_name: str) -> int | str | float | None:
+        """
+        Get property from current conditions.
+
+        Used for V4 API.
+        """
+        return self.coordinator.data.get(CURRENT, {}).get(property_name)
+
+    @property
+    def attribution(self):
+        """Return the attribution."""
+        return ATTRIBUTION