"""The openweathermap component.""" from __future__ import annotations from dataclasses import dataclass import logging from pyopenweathermap import create_owm_client from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_API_KEY, CONF_LANGUAGE, CONF_MODE, CONF_NAME from homeassistant.core import HomeAssistant from .const import CONFIG_FLOW_VERSION, DEFAULT_OWM_MODE, OWM_MODES, PLATFORMS from .coordinator import WeatherUpdateCoordinator from .repairs import async_create_issue, async_delete_issue from .utils import build_data_and_options _LOGGER = logging.getLogger(__name__) type OpenweathermapConfigEntry = ConfigEntry[OpenweathermapData] @dataclass class OpenweathermapData: """Runtime data definition.""" name: str mode: str coordinator: WeatherUpdateCoordinator async def async_setup_entry( hass: HomeAssistant, entry: OpenweathermapConfigEntry ) -> bool: """Set up OpenWeatherMap as config entry.""" name = entry.data[CONF_NAME] api_key = entry.data[CONF_API_KEY] language = entry.options[CONF_LANGUAGE] mode = entry.options[CONF_MODE] if mode not in OWM_MODES: async_create_issue(hass, entry.entry_id) else: async_delete_issue(hass, entry.entry_id) owm_client = create_owm_client(api_key, mode, lang=language) weather_coordinator = WeatherUpdateCoordinator(hass, entry, owm_client) await weather_coordinator.async_config_entry_first_refresh() entry.async_on_unload(entry.add_update_listener(async_update_options)) entry.runtime_data = OpenweathermapData(name, mode, weather_coordinator) await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS) return True async def async_migrate_entry( hass: HomeAssistant, entry: OpenweathermapConfigEntry ) -> bool: """Migrate old entry.""" config_entries = hass.config_entries data = entry.data options = entry.options version = entry.version _LOGGER.debug("Migrating OpenWeatherMap entry from version %s", version) if version < 5: combined_data = {**data, **options, CONF_MODE: DEFAULT_OWM_MODE} new_data, new_options = build_data_and_options(combined_data) config_entries.async_update_entry( entry, data=new_data, options=new_options, version=CONFIG_FLOW_VERSION, ) _LOGGER.debug("Migration to version %s successful", CONFIG_FLOW_VERSION) return True async def async_update_options( hass: HomeAssistant, entry: OpenweathermapConfigEntry ) -> None: """Update options.""" await hass.config_entries.async_reload(entry.entry_id) async def async_unload_entry( hass: HomeAssistant, entry: OpenweathermapConfigEntry ) -> bool: """Unload a config entry.""" return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)