mirror of
https://github.com/home-assistant/core.git
synced 2025-07-19 11:17:21 +00:00
Explicitly pass in the config_entry in loqed coordinator (#138106)
explicitly pass in the config_entry in coordinator
This commit is contained in:
parent
3175cb9c4d
commit
7beb1c0921
@ -44,7 +44,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|||||||
aiohttp.ClientError,
|
aiohttp.ClientError,
|
||||||
) as ex:
|
) as ex:
|
||||||
raise ConfigEntryNotReady(f"Unable to connect to bridge at {host}") from ex
|
raise ConfigEntryNotReady(f"Unable to connect to bridge at {host}") from ex
|
||||||
coordinator = LoqedDataCoordinator(hass, api, lock, entry)
|
coordinator = LoqedDataCoordinator(hass, entry, api, lock)
|
||||||
await coordinator.ensure_webhooks()
|
await coordinator.ensure_webhooks()
|
||||||
|
|
||||||
await coordinator.async_config_entry_first_refresh()
|
await coordinator.async_config_entry_first_refresh()
|
||||||
|
@ -71,19 +71,20 @@ class StatusMessage(TypedDict):
|
|||||||
class LoqedDataCoordinator(DataUpdateCoordinator[StatusMessage]):
|
class LoqedDataCoordinator(DataUpdateCoordinator[StatusMessage]):
|
||||||
"""Data update coordinator for the loqed platform."""
|
"""Data update coordinator for the loqed platform."""
|
||||||
|
|
||||||
|
config_entry: ConfigEntry
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
|
config_entry: ConfigEntry,
|
||||||
api: loqed.LoqedAPI,
|
api: loqed.LoqedAPI,
|
||||||
lock: loqed.Lock,
|
lock: loqed.Lock,
|
||||||
entry: ConfigEntry,
|
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Initialize the Loqed Data Update coordinator."""
|
"""Initialize the Loqed Data Update coordinator."""
|
||||||
super().__init__(hass, _LOGGER, name="Loqed sensors")
|
super().__init__(hass, _LOGGER, config_entry=config_entry, name="Loqed sensors")
|
||||||
self._api = api
|
self._api = api
|
||||||
self._entry = entry
|
|
||||||
self.lock = lock
|
self.lock = lock
|
||||||
self.device_name = self._entry.data[CONF_NAME]
|
self.device_name = config_entry.data[CONF_NAME]
|
||||||
|
|
||||||
async def _async_update_data(self) -> StatusMessage:
|
async def _async_update_data(self) -> StatusMessage:
|
||||||
"""Fetch data from API endpoint."""
|
"""Fetch data from API endpoint."""
|
||||||
@ -110,17 +111,19 @@ class LoqedDataCoordinator(DataUpdateCoordinator[StatusMessage]):
|
|||||||
|
|
||||||
async def ensure_webhooks(self) -> None:
|
async def ensure_webhooks(self) -> None:
|
||||||
"""Register webhook on LOQED bridge."""
|
"""Register webhook on LOQED bridge."""
|
||||||
webhook_id = self._entry.data[CONF_WEBHOOK_ID]
|
webhook_id = self.config_entry.data[CONF_WEBHOOK_ID]
|
||||||
|
|
||||||
webhook.async_register(
|
webhook.async_register(
|
||||||
self.hass, DOMAIN, "Loqed", webhook_id, self._handle_webhook
|
self.hass, DOMAIN, "Loqed", webhook_id, self._handle_webhook
|
||||||
)
|
)
|
||||||
|
|
||||||
if cloud.async_active_subscription(self.hass):
|
if cloud.async_active_subscription(self.hass):
|
||||||
webhook_url = await async_cloudhook_generate_url(self.hass, self._entry)
|
webhook_url = await async_cloudhook_generate_url(
|
||||||
|
self.hass, self.config_entry
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
webhook_url = webhook.async_generate_url(
|
webhook_url = webhook.async_generate_url(
|
||||||
self.hass, self._entry.data[CONF_WEBHOOK_ID]
|
self.hass, self.config_entry.data[CONF_WEBHOOK_ID]
|
||||||
)
|
)
|
||||||
|
|
||||||
_LOGGER.debug("Webhook URL: %s", webhook_url)
|
_LOGGER.debug("Webhook URL: %s", webhook_url)
|
||||||
@ -140,10 +143,10 @@ class LoqedDataCoordinator(DataUpdateCoordinator[StatusMessage]):
|
|||||||
|
|
||||||
async def remove_webhooks(self) -> None:
|
async def remove_webhooks(self) -> None:
|
||||||
"""Remove webhook from LOQED bridge."""
|
"""Remove webhook from LOQED bridge."""
|
||||||
webhook_id = self._entry.data[CONF_WEBHOOK_ID]
|
webhook_id = self.config_entry.data[CONF_WEBHOOK_ID]
|
||||||
|
|
||||||
if CONF_CLOUDHOOK_URL in self._entry.data:
|
if CONF_CLOUDHOOK_URL in self.config_entry.data:
|
||||||
webhook_url = self._entry.data[CONF_CLOUDHOOK_URL]
|
webhook_url = self.config_entry.data[CONF_CLOUDHOOK_URL]
|
||||||
else:
|
else:
|
||||||
webhook_url = webhook.async_generate_url(self.hass, webhook_id)
|
webhook_url = webhook.async_generate_url(self.hass, webhook_id)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user