jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-15 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use runtime data for FritzBox Call Monitor (#116553)

Browse Source
This commit is contained in:
Marc Mueller 2024-05-01 21:27:34 +02:00 committed by GitHub
parent 7fd60ddba4
commit a25e202ef0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 18 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
homeassistant/components/fritzbox_callmonitor/__init__.py
View File

@ -11,19 +11,16 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
from .base import FritzBoxPhonebook from .base import FritzBoxPhonebook
from .const import ( from .const import CONF_PHONEBOOK, CONF_PREFIXES, PLATFORMS
CONF_PHONEBOOK,
CONF_PREFIXES,
DOMAIN,
FRITZBOX_PHONEBOOK,
PLATFORMS,
UNDO_UPDATE_LISTENER,
)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
FritzBoxCallMonitorConfigEntry = ConfigEntry[FritzBoxPhonebook]
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
async def async_setup_entry(
hass: HomeAssistant, config_entry: FritzBoxCallMonitorConfigEntry
) -> bool:
"""Set up the fritzbox_callmonitor platforms.""" """Set up the fritzbox_callmonitor platforms."""
fritzbox_phonebook = FritzBoxPhonebook( fritzbox_phonebook = FritzBoxPhonebook(
host=config_entry.data[CONF_HOST], host=config_entry.data[CONF_HOST],
@ -51,34 +48,22 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
_LOGGER.error("Unable to connect to AVM FRITZ!Box call monitor: %s", ex) _LOGGER.error("Unable to connect to AVM FRITZ!Box call monitor: %s", ex)
raise ConfigEntryNotReady from ex raise ConfigEntryNotReady from ex
undo_listener = config_entry.add_update_listener(update_listener) config_entry.runtime_data = fritzbox_phonebook
config_entry.async_on_unload(config_entry.add_update_listener(update_listener))
hass.data.setdefault(DOMAIN, {})
hass.data[DOMAIN][config_entry.entry_id] = {
FRITZBOX_PHONEBOOK: fritzbox_phonebook,
UNDO_UPDATE_LISTENER: undo_listener,
}
await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS) await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
return True return True
async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: async def async_unload_entry(
hass: HomeAssistant, config_entry: FritzBoxCallMonitorConfigEntry
) -> bool:
"""Unloading the fritzbox_callmonitor platforms.""" """Unloading the fritzbox_callmonitor platforms."""
return await hass.config_entries.async_unload_platforms(config_entry, PLATFORMS)
unload_ok = await hass.config_entries.async_unload_platforms(
config_entry, PLATFORMS
)
hass.data[DOMAIN][config_entry.entry_id][UNDO_UPDATE_LISTENER]()
if unload_ok:
hass.data[DOMAIN].pop(config_entry.entry_id)
return unload_ok
async def update_listener(hass: HomeAssistant, config_entry: ConfigEntry) -> None: async def update_listener(
hass: HomeAssistant, config_entry: FritzBoxCallMonitorConfigEntry
) -> None:
"""Update listener to reload after option has changed.""" """Update listener to reload after option has changed."""
await hass.config_entries.async_reload(config_entry.entry_id) await hass.config_entries.async_reload(config_entry.entry_id)

2
homeassistant/components/fritzbox_callmonitor/const.py
View File

@ -38,5 +38,3 @@ DOMAIN: Final = "fritzbox_callmonitor"
MANUFACTURER: Final = "AVM" MANUFACTURER: Final = "AVM"
PLATFORMS = [Platform.SENSOR] PLATFORMS = [Platform.SENSOR]
UNDO_UPDATE_LISTENER: Final = "undo_update_listener"
FRITZBOX_PHONEBOOK: Final = "fritzbox_phonebook"

9
homeassistant/components/fritzbox_callmonitor/sensor.py
View File

@ -14,19 +14,18 @@ from typing import Any, cast
from fritzconnection.core.fritzmonitor import FritzMonitor from fritzconnection.core.fritzmonitor import FritzMonitor
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT, EVENT_HOMEASSISTANT_STOP from homeassistant.const import CONF_HOST, CONF_PORT, EVENT_HOMEASSISTANT_STOP
from homeassistant.core import Event, HomeAssistant from homeassistant.core import Event, HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import FritzBoxCallMonitorConfigEntry
from .base import FritzBoxPhonebook from .base import FritzBoxPhonebook
from .const import ( from .const import (
ATTR_PREFIXES, ATTR_PREFIXES,
CONF_PHONEBOOK, CONF_PHONEBOOK,
CONF_PREFIXES, CONF_PREFIXES,
DOMAIN, DOMAIN,
FRITZBOX_PHONEBOOK,
MANUFACTURER, MANUFACTURER,
SERIAL_NUMBER, SERIAL_NUMBER,
FritzState, FritzState,
@ -48,13 +47,11 @@ class CallState(StrEnum):
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
config_entry: ConfigEntry, config_entry: FritzBoxCallMonitorConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Set up the fritzbox_callmonitor sensor from config_entry.""" """Set up the fritzbox_callmonitor sensor from config_entry."""
fritzbox_phonebook: FritzBoxPhonebook = hass.data[DOMAIN][config_entry.entry_id][ fritzbox_phonebook = config_entry.runtime_data
FRITZBOX_PHONEBOOK
]
phonebook_id: int = config_entry.data[CONF_PHONEBOOK] phonebook_id: int = config_entry.data[CONF_PHONEBOOK]
prefixes: list[str] | None = config_entry.options.get(CONF_PREFIXES) prefixes: list[str] | None = config_entry.options.get(CONF_PREFIXES)