82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
"""The met component."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry as dr
|
|
|
|
from .const import (
|
|
CONF_TRACK_HOME,
|
|
DEFAULT_HOME_LATITUDE,
|
|
DEFAULT_HOME_LONGITUDE,
|
|
DOMAIN,
|
|
)
|
|
from .coordinator import MetDataUpdateCoordinator
|
|
|
|
PLATFORMS = [Platform.WEATHER]
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
type MetWeatherConfigEntry = ConfigEntry[MetDataUpdateCoordinator]
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant, config_entry: MetWeatherConfigEntry
|
|
) -> bool:
|
|
"""Set up Met as config entry."""
|
|
# Don't setup if tracking home location and latitude or longitude isn't set.
|
|
# Also, filters out our onboarding default location.
|
|
if config_entry.data.get(CONF_TRACK_HOME, False) and (
|
|
(not hass.config.latitude and not hass.config.longitude)
|
|
or (
|
|
hass.config.latitude == DEFAULT_HOME_LATITUDE
|
|
and hass.config.longitude == DEFAULT_HOME_LONGITUDE
|
|
)
|
|
):
|
|
_LOGGER.warning(
|
|
"Skip setting up met.no integration; No Home location has been set"
|
|
)
|
|
return False
|
|
|
|
coordinator = MetDataUpdateCoordinator(hass, config_entry)
|
|
await coordinator.async_config_entry_first_refresh()
|
|
|
|
if config_entry.data.get(CONF_TRACK_HOME, False):
|
|
coordinator.track_home()
|
|
|
|
config_entry.runtime_data = coordinator
|
|
|
|
config_entry.async_on_unload(config_entry.add_update_listener(async_update_entry))
|
|
config_entry.async_on_unload(coordinator.untrack_home)
|
|
|
|
await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
|
|
|
|
await cleanup_old_device(hass)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(
|
|
hass: HomeAssistant, config_entry: MetWeatherConfigEntry
|
|
) -> bool:
|
|
"""Unload a config entry."""
|
|
return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
|
|
|
|
|
|
async def async_update_entry(hass: HomeAssistant, config_entry: MetWeatherConfigEntry):
|
|
"""Reload Met component when options changed."""
|
|
await hass.config_entries.async_reload(config_entry.entry_id)
|
|
|
|
|
|
async def cleanup_old_device(hass: HomeAssistant) -> None:
|
|
"""Cleanup device without proper device identifier."""
|
|
device_reg = dr.async_get(hass)
|
|
device = device_reg.async_get_device(identifiers={(DOMAIN,)}) # type: ignore[arg-type]
|
|
if device:
|
|
_LOGGER.debug("Removing improper device %s", device.name)
|
|
device_reg.async_remove_device(device.id)
|