mirror of
https://github.com/home-assistant/core.git
synced 2025-04-26 02:07:54 +00:00
Explicitly pass in the config_entry in Proximity coordinator init (#136775)
This commit is contained in:
parent
01b278c547
commit
e07e8b8706
@ -4,7 +4,6 @@ from __future__ import annotations
|
|||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from homeassistant.config_entries import ConfigEntry
|
|
||||||
from homeassistant.const import Platform
|
from homeassistant.const import Platform
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.event import (
|
from homeassistant.helpers.event import (
|
||||||
@ -22,7 +21,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ProximityConfigEntry) ->
|
|||||||
"""Set up Proximity from a config entry."""
|
"""Set up Proximity from a config entry."""
|
||||||
_LOGGER.debug("setup %s with config:%s", entry.title, entry.data)
|
_LOGGER.debug("setup %s with config:%s", entry.title, entry.data)
|
||||||
|
|
||||||
coordinator = ProximityDataUpdateCoordinator(hass, entry.title, dict(entry.data))
|
coordinator = ProximityDataUpdateCoordinator(hass, entry)
|
||||||
|
|
||||||
entry.async_on_unload(
|
entry.async_on_unload(
|
||||||
async_track_state_change_event(
|
async_track_state_change_event(
|
||||||
@ -48,11 +47,13 @@ async def async_setup_entry(hass: HomeAssistant, entry: ProximityConfigEntry) ->
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
async def async_unload_entry(hass: HomeAssistant, entry: ProximityConfigEntry) -> bool:
|
||||||
"""Unload a config entry."""
|
"""Unload a config entry."""
|
||||||
return await hass.config_entries.async_unload_platforms(entry, [Platform.SENSOR])
|
return await hass.config_entries.async_unload_platforms(entry, [Platform.SENSOR])
|
||||||
|
|
||||||
|
|
||||||
async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
|
async def _async_update_listener(
|
||||||
|
hass: HomeAssistant, entry: ProximityConfigEntry
|
||||||
|
) -> None:
|
||||||
"""Handle options update."""
|
"""Handle options update."""
|
||||||
await hass.config_entries.async_reload(entry.entry_id)
|
await hass.config_entries.async_reload(entry.entry_id)
|
||||||
|
@ -23,7 +23,6 @@ from homeassistant.core import (
|
|||||||
)
|
)
|
||||||
from homeassistant.helpers import entity_registry as er
|
from homeassistant.helpers import entity_registry as er
|
||||||
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
|
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
|
||||||
from homeassistant.helpers.typing import ConfigType
|
|
||||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
||||||
from homeassistant.util.location import distance
|
from homeassistant.util.location import distance
|
||||||
|
|
||||||
@ -75,16 +74,14 @@ class ProximityDataUpdateCoordinator(DataUpdateCoordinator[ProximityData]):
|
|||||||
|
|
||||||
config_entry: ProximityConfigEntry
|
config_entry: ProximityConfigEntry
|
||||||
|
|
||||||
def __init__(
|
def __init__(self, hass: HomeAssistant, config_entry: ProximityConfigEntry) -> None:
|
||||||
self, hass: HomeAssistant, friendly_name: str, config: ConfigType
|
|
||||||
) -> None:
|
|
||||||
"""Initialize the Proximity coordinator."""
|
"""Initialize the Proximity coordinator."""
|
||||||
self.ignored_zone_ids: list[str] = config[CONF_IGNORED_ZONES]
|
self.ignored_zone_ids: list[str] = config_entry.data[CONF_IGNORED_ZONES]
|
||||||
self.tracked_entities: list[str] = config[CONF_TRACKED_ENTITIES]
|
self.tracked_entities: list[str] = config_entry.data[CONF_TRACKED_ENTITIES]
|
||||||
self.tolerance: int = config[CONF_TOLERANCE]
|
self.tolerance: int = config_entry.data[CONF_TOLERANCE]
|
||||||
self.proximity_zone_id: str = config[CONF_ZONE]
|
self.proximity_zone_id: str = config_entry.data[CONF_ZONE]
|
||||||
self.proximity_zone_name: str = self.proximity_zone_id.split(".")[-1]
|
self.proximity_zone_name: str = self.proximity_zone_id.split(".")[-1]
|
||||||
self.unit_of_measurement: str = config.get(
|
self.unit_of_measurement: str = config_entry.data.get(
|
||||||
CONF_UNIT_OF_MEASUREMENT, hass.config.units.length_unit
|
CONF_UNIT_OF_MEASUREMENT, hass.config.units.length_unit
|
||||||
)
|
)
|
||||||
self.entity_mapping: dict[str, list[str]] = defaultdict(list)
|
self.entity_mapping: dict[str, list[str]] = defaultdict(list)
|
||||||
@ -92,7 +89,8 @@ class ProximityDataUpdateCoordinator(DataUpdateCoordinator[ProximityData]):
|
|||||||
super().__init__(
|
super().__init__(
|
||||||
hass,
|
hass,
|
||||||
_LOGGER,
|
_LOGGER,
|
||||||
name=friendly_name,
|
config_entry=config_entry,
|
||||||
|
name=config_entry.title,
|
||||||
update_interval=None,
|
update_interval=None,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user