mirror of
https://github.com/home-assistant/core.git
synced 2025-10-26 12:09:32 +00:00
4f24d63de1
* Update metoffice to use DataHub API * Reauth test * Updated to datapoint 0.11.0 * Less hacky check for day/night in twice-daily forecasts * Updated to datapoint 0.12.1, added daily forecast * addressed review comments * one more nit * validate credewntials in reauth flow * Addressed review comments * Attempt to improve coverage * Addressed comments * Reverted unnecessary reordering * Update homeassistant/components/metoffice/sensor.py * Update tests/components/metoffice/test_sensor.py * Update homeassistant/components/metoffice/sensor.py --------- Co-authored-by: Franck Nijhof <git@frenck.dev> Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""Helpers used for Met Office integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any, Literal
|
|
|
|
import datapoint
|
|
from datapoint.Forecast import Forecast
|
|
from requests import HTTPError
|
|
|
|
from homeassistant.exceptions import ConfigEntryAuthFailed
|
|
from homeassistant.helpers.update_coordinator import UpdateFailed
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def fetch_data(
|
|
connection: datapoint.Manager,
|
|
latitude: float,
|
|
longitude: float,
|
|
frequency: Literal["daily", "twice-daily", "hourly"],
|
|
) -> Forecast:
|
|
"""Fetch weather and forecast from Datapoint API."""
|
|
try:
|
|
return connection.get_forecast(
|
|
latitude, longitude, frequency, convert_weather_code=False
|
|
)
|
|
except (ValueError, datapoint.exceptions.APIException) as err:
|
|
_LOGGER.error("Check Met Office connection: %s", err.args)
|
|
raise UpdateFailed from err
|
|
except HTTPError as err:
|
|
if err.response.status_code == 401:
|
|
raise ConfigEntryAuthFailed from err
|
|
raise
|
|
|
|
|
|
def get_attribute(data: dict[str, Any] | None, attr_name: str) -> Any | None:
|
|
"""Get an attribute from weather data."""
|
|
if data:
|
|
return data.get(attr_name, {}).get("value")
|
|
return None
|