44For more details about this integration, please refer to
55https://github.com/golles/ha-knmi/
66"""
7- import asyncio
8- from datetime import timedelta
9- import logging
7+ from typing import Any , Callable
108
119from homeassistant .config_entries import ConfigEntry
1210from homeassistant .const import CONF_LATITUDE , CONF_LONGITUDE , CONF_API_KEY
1311from homeassistant .core import Config , HomeAssistant
1412from homeassistant .exceptions import ConfigEntryNotReady
1513from homeassistant .helpers .aiohttp_client import async_get_clientsession
14+ from homeassistant .helpers .device_registry import DeviceEntryType
15+ from homeassistant .helpers .entity import DeviceInfo
1616from homeassistant .helpers .update_coordinator import DataUpdateCoordinator , UpdateFailed
1717
1818from .api import KnmiApiClient
19- from .const import DOMAIN , PLATFORMS
19+ from .const import _LOGGER , DOMAIN , NAME , VERSION , PLATFORMS , SCAN_INTERVAL
2020
21- SCAN_INTERVAL = timedelta (seconds = 300 )
2221
23- _LOGGER : logging .Logger = logging .getLogger (__package__ )
24-
25-
26- async def async_setup (_hass : HomeAssistant , _config : Config ):
22+ async def async_setup (_hass : HomeAssistant , _config : Config ) -> bool :
2723 """Set up this integration using YAML is not supported."""
2824 return True
2925
3026
31- async def async_setup_entry (hass : HomeAssistant , entry : ConfigEntry ):
32- """Set up this integration using UI ."""
27+ async def async_setup_entry (hass : HomeAssistant , entry : ConfigEntry ) -> bool :
28+ """Set up KNMI as config entry ."""
3329 if hass .data .get (DOMAIN ) is None :
3430 hass .data .setdefault (DOMAIN , {})
3531
@@ -40,62 +36,76 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
4036 session = async_get_clientsession (hass )
4137 client = KnmiApiClient (api_key , latitude , longitude , session )
4238
43- coordinator = KnmiDataUpdateCoordinator (hass , client = client )
44- await coordinator .async_refresh ()
39+ device_info = DeviceInfo (
40+ entry_type = DeviceEntryType .SERVICE ,
41+ identifiers = {(DOMAIN , entry .entry_id )},
42+ manufacturer = NAME ,
43+ name = NAME ,
44+ model = VERSION ,
45+ configuration_url = "http://weerlive.nl/api/toegang/account.php" ,
46+ )
47+
48+ coordinator = KnmiDataUpdateCoordinator (
49+ hass = hass , client = client , device_info = device_info
50+ )
51+ await coordinator .async_config_entry_first_refresh ()
4552
4653 if not coordinator .last_update_success :
4754 raise ConfigEntryNotReady
4855
49- hass .data [DOMAIN ][entry .entry_id ] = coordinator
56+ entry .async_on_unload (entry .add_update_listener (async_reload_entry ))
57+
58+ hass .data .setdefault (DOMAIN , {})[entry .entry_id ] = coordinator
5059
5160 for platform in PLATFORMS :
5261 if entry .options .get (platform , True ):
5362 coordinator .platforms .append (platform )
5463 hass .async_add_job (
5564 hass .config_entries .async_forward_entry_setup (entry , platform )
5665 )
57-
58- entry .async_on_unload (entry .add_update_listener (async_reload_entry ))
5966 return True
6067
6168
69+ async def async_unload_entry (hass : HomeAssistant , entry : ConfigEntry ) -> bool :
70+ """Unload KNMI config entry."""
71+ unload_ok = await hass .config_entries .async_unload_platforms (entry , PLATFORMS )
72+ if unload_ok :
73+ del hass .data [DOMAIN ][entry .entry_id ]
74+ return unload_ok
75+
76+
77+ async def async_reload_entry (hass : HomeAssistant , entry : ConfigEntry ) -> None :
78+ """Reload config entry."""
79+ await async_unload_entry (hass , entry )
80+ await async_setup_entry (hass , entry )
81+
82+
6283class KnmiDataUpdateCoordinator (DataUpdateCoordinator ):
6384 """Class to manage fetching data from the API."""
6485
65- def __init__ (self , hass : HomeAssistant , client : KnmiApiClient ) -> None :
86+ def __init__ (
87+ self , hass : HomeAssistant , client : KnmiApiClient , device_info : DeviceInfo
88+ ) -> None :
6689 """Initialize."""
6790 self .api = client
91+ self .device_info = device_info
6892 self .platforms = []
6993
70- super ().__init__ (hass , _LOGGER , name = DOMAIN , update_interval = SCAN_INTERVAL )
94+ super ().__init__ (
95+ hass = hass , logger = _LOGGER , name = DOMAIN , update_interval = SCAN_INTERVAL
96+ )
7197
72- async def _async_update_data (self ):
98+ async def _async_update_data (self ) -> dict [ str , Any ] :
7399 """Update data via library."""
74100 try :
75101 return await self .api .async_get_data ()
76102 except Exception as exception :
77103 raise UpdateFailed () from exception
78104
79-
80- async def async_unload_entry (hass : HomeAssistant , entry : ConfigEntry ) -> bool :
81- """Handle removal of an entry."""
82- coordinator = hass .data [DOMAIN ][entry .entry_id ]
83- unloaded = all (
84- await asyncio .gather (
85- * [
86- hass .config_entries .async_forward_entry_unload (entry , platform )
87- for platform in PLATFORMS
88- if platform in coordinator .platforms
89- ]
90- )
91- )
92- if unloaded :
93- hass .data [DOMAIN ].pop (entry .entry_id )
94-
95- return unloaded
96-
97-
98- async def async_reload_entry (hass : HomeAssistant , entry : ConfigEntry ) -> None :
99- """Reload config entry."""
100- await async_unload_entry (hass , entry )
101- await async_setup_entry (hass , entry )
105+ def get_value (
106+ self , key : str , convert_to : Callable = str
107+ ) -> float | int | str | None :
108+ """Get a value from the retrieved data and convert to given type"""
109+ if self .data and self .data [key ]:
110+ return convert_to (self .data [key ])
111+ return None
0 commit comments