Skip to content

Commit 434cd95

Browse files
Use ConfigEntry.runtime_data to store runtime data in NINA (home-assistant#146754)
Co-authored-by: Joost Lekkerkerker <[email protected]>
1 parent 1a5bc2c commit 434cd95

File tree

3 files changed

+10
-10
lines changed

3 files changed

+10
-10
lines changed

homeassistant/components/nina/__init__.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
from __future__ import annotations
44

5-
from homeassistant.config_entries import ConfigEntry
65
from homeassistant.const import Platform
76
from homeassistant.core import HomeAssistant
87

@@ -11,15 +10,14 @@
1110
CONF_AREA_FILTER,
1211
CONF_FILTER_CORONA,
1312
CONF_HEADLINE_FILTER,
14-
DOMAIN,
1513
NO_MATCH_REGEX,
1614
)
17-
from .coordinator import NINADataUpdateCoordinator
15+
from .coordinator import NinaConfigEntry, NINADataUpdateCoordinator
1816

1917
PLATFORMS: list[str] = [Platform.BINARY_SENSOR]
2018

2119

22-
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
20+
async def async_setup_entry(hass: HomeAssistant, entry: NinaConfigEntry) -> bool:
2321
"""Set up platform from a ConfigEntry."""
2422
if CONF_HEADLINE_FILTER not in entry.data:
2523
filter_regex = NO_MATCH_REGEX
@@ -41,18 +39,18 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
4139

4240
entry.async_on_unload(entry.add_update_listener(_async_update_listener))
4341

44-
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
42+
entry.runtime_data = coordinator
4543

4644
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
4745

4846
return True
4947

5048

51-
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
49+
async def async_unload_entry(hass: HomeAssistant, entry: NinaConfigEntry) -> bool:
5250
"""Unload a config entry."""
5351
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
5452

5553

56-
async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
54+
async def _async_update_listener(hass: HomeAssistant, entry: NinaConfigEntry) -> None:
5755
"""Handle options update."""
5856
await hass.config_entries.async_reload(entry.entry_id)

homeassistant/components/nina/binary_sensor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -30,17 +30,17 @@
3030
CONF_REGIONS,
3131
DOMAIN,
3232
)
33-
from .coordinator import NINADataUpdateCoordinator
33+
from .coordinator import NinaConfigEntry, NINADataUpdateCoordinator
3434

3535

3636
async def async_setup_entry(
3737
hass: HomeAssistant,
38-
config_entry: ConfigEntry,
38+
config_entry: NinaConfigEntry,
3939
async_add_entities: AddConfigEntryEntitiesCallback,
4040
) -> None:
4141
"""Set up entries."""
4242

43-
coordinator: NINADataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
43+
coordinator = config_entry.runtime_data
4444

4545
regions: dict[str, str] = config_entry.data[CONF_REGIONS]
4646
message_slots: int = config_entry.data[CONF_MESSAGE_SLOTS]

homeassistant/components/nina/coordinator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323
SCAN_INTERVAL,
2424
)
2525

26+
type NinaConfigEntry = ConfigEntry[NINADataUpdateCoordinator]
27+
2628

2729
@dataclass
2830
class NinaWarningData:

0 commit comments

Comments
 (0)