b4bac0f7a0
* Remove unnecessary exception re-wraps * Preserve exception chains on re-raise We slap "from cause" to almost all possible cases here. In some cases it could conceivably be better to do "from None" if we really want to hide the cause. However those should be in the minority, and "from cause" should be an improvement over the corresponding raise without a "from" in all cases anyway. The only case where we raise from None here is in plex, where the exception for an original invalid SSL cert is not the root cause for failure to validate a newly fetched one. Follow local convention on exception variable names if there is a consistent one, otherwise `err` to match with majority of codebase. * Fix mistaken re-wrap in homematicip_cloud/hap.py Missed the difference between HmipConnectionError and HmipcConnectionError. * Do not hide original error on plex new cert validation error Original is not the cause for the new one, but showing old in the traceback is useful nevertheless.
155 lines
4.9 KiB
Python
155 lines
4.9 KiB
Python
"""The met component."""
|
|
from datetime import timedelta
|
|
import logging
|
|
from random import randrange
|
|
|
|
import metno
|
|
|
|
from homeassistant.const import (
|
|
CONF_ELEVATION,
|
|
CONF_LATITUDE,
|
|
CONF_LONGITUDE,
|
|
EVENT_CORE_CONFIG_UPDATE,
|
|
LENGTH_FEET,
|
|
LENGTH_METERS,
|
|
)
|
|
from homeassistant.core import Config, HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
|
from homeassistant.util.distance import convert as convert_distance
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
from .const import CONF_TRACK_HOME, DOMAIN
|
|
|
|
URL = "https://api.met.no/weatherapi/locationforecast/2.0/classic"
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config: Config) -> bool:
|
|
"""Set up configured Met."""
|
|
hass.data.setdefault(DOMAIN, {})
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry):
|
|
"""Set up Met as config entry."""
|
|
coordinator = MetDataUpdateCoordinator(hass, config_entry)
|
|
await coordinator.async_refresh()
|
|
|
|
if not coordinator.last_update_success:
|
|
raise ConfigEntryNotReady
|
|
|
|
if config_entry.data.get(CONF_TRACK_HOME, False):
|
|
coordinator.track_home()
|
|
|
|
hass.data[DOMAIN][config_entry.entry_id] = coordinator
|
|
|
|
hass.async_create_task(
|
|
hass.config_entries.async_forward_entry_setup(config_entry, "weather")
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass, config_entry):
|
|
"""Unload a config entry."""
|
|
await hass.config_entries.async_forward_entry_unload(config_entry, "weather")
|
|
hass.data[DOMAIN][config_entry.entry_id].untrack_home()
|
|
hass.data[DOMAIN].pop(config_entry.entry_id)
|
|
|
|
return True
|
|
|
|
|
|
class MetDataUpdateCoordinator(DataUpdateCoordinator):
|
|
"""Class to manage fetching Met data."""
|
|
|
|
def __init__(self, hass, config_entry):
|
|
"""Initialize global Met data updater."""
|
|
self._unsub_track_home = None
|
|
self.weather = MetWeatherData(
|
|
hass, config_entry.data, hass.config.units.is_metric
|
|
)
|
|
self.weather.init_data()
|
|
|
|
update_interval = timedelta(minutes=randrange(55, 65))
|
|
|
|
super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=update_interval)
|
|
|
|
async def _async_update_data(self):
|
|
"""Fetch data from Met."""
|
|
try:
|
|
return await self.weather.fetch_data()
|
|
except Exception as err:
|
|
raise UpdateFailed(f"Update failed: {err}") from err
|
|
|
|
def track_home(self):
|
|
"""Start tracking changes to HA home setting."""
|
|
if self._unsub_track_home:
|
|
return
|
|
|
|
async def _async_update_weather_data(_event=None):
|
|
"""Update weather data."""
|
|
self.weather.init_data()
|
|
await self.async_refresh()
|
|
|
|
self._unsub_track_home = self.hass.bus.async_listen(
|
|
EVENT_CORE_CONFIG_UPDATE, _async_update_weather_data
|
|
)
|
|
|
|
def untrack_home(self):
|
|
"""Stop tracking changes to HA home setting."""
|
|
if self._unsub_track_home:
|
|
self._unsub_track_home()
|
|
self._unsub_track_home = None
|
|
|
|
|
|
class MetWeatherData:
|
|
"""Keep data for Met.no weather entities."""
|
|
|
|
def __init__(self, hass, config, is_metric):
|
|
"""Initialise the weather entity data."""
|
|
self.hass = hass
|
|
self._config = config
|
|
self._is_metric = is_metric
|
|
self._weather_data = None
|
|
self.current_weather_data = {}
|
|
self.daily_forecast = None
|
|
self.hourly_forecast = None
|
|
|
|
def init_data(self):
|
|
"""Weather data inialization - get the coordinates."""
|
|
if self._config.get(CONF_TRACK_HOME, False):
|
|
latitude = self.hass.config.latitude
|
|
longitude = self.hass.config.longitude
|
|
elevation = self.hass.config.elevation
|
|
else:
|
|
latitude = self._config[CONF_LATITUDE]
|
|
longitude = self._config[CONF_LONGITUDE]
|
|
elevation = self._config[CONF_ELEVATION]
|
|
|
|
if not self._is_metric:
|
|
elevation = int(
|
|
round(convert_distance(elevation, LENGTH_FEET, LENGTH_METERS))
|
|
)
|
|
|
|
coordinates = {
|
|
"lat": str(latitude),
|
|
"lon": str(longitude),
|
|
"msl": str(elevation),
|
|
}
|
|
|
|
self._weather_data = metno.MetWeatherData(
|
|
coordinates, async_get_clientsession(self.hass), URL
|
|
)
|
|
|
|
async def fetch_data(self):
|
|
"""Fetch data from API - (current weather and forecast)."""
|
|
await self._weather_data.fetching_data()
|
|
self.current_weather_data = self._weather_data.get_current_weather()
|
|
time_zone = dt_util.DEFAULT_TIME_ZONE
|
|
self.daily_forecast = self._weather_data.get_forecast(time_zone, False)
|
|
self.hourly_forecast = self._weather_data.get_forecast(time_zone, True)
|
|
return self
|