mirror of
https://github.com/home-assistant/core.git
synced 2025-07-17 02:07:09 +00:00
Explicitly pass in the config_entry in tado coordinator (#137916)
explicitly pass in the config_entry in coordinator
This commit is contained in:
parent
c47a97a4f0
commit
b338de9a30
@ -87,7 +87,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: TadoConfigEntry) -> bool
|
|||||||
|
|
||||||
|
|
||||||
@callback
|
@callback
|
||||||
def _async_import_options_from_data_if_missing(hass: HomeAssistant, entry: ConfigEntry):
|
def _async_import_options_from_data_if_missing(
|
||||||
|
hass: HomeAssistant, entry: TadoConfigEntry
|
||||||
|
):
|
||||||
options = dict(entry.options)
|
options = dict(entry.options)
|
||||||
if CONF_FALLBACK not in options:
|
if CONF_FALLBACK not in options:
|
||||||
options[CONF_FALLBACK] = entry.data.get(
|
options[CONF_FALLBACK] = entry.data.get(
|
||||||
|
@ -4,18 +4,20 @@ from __future__ import annotations
|
|||||||
|
|
||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
import logging
|
import logging
|
||||||
from typing import Any
|
from typing import TYPE_CHECKING, Any
|
||||||
|
|
||||||
from PyTado.interface import Tado
|
from PyTado.interface import Tado
|
||||||
from requests import RequestException
|
from requests import RequestException
|
||||||
|
|
||||||
from homeassistant.components.climate import PRESET_AWAY, PRESET_HOME
|
from homeassistant.components.climate import PRESET_AWAY, PRESET_HOME
|
||||||
from homeassistant.config_entries import ConfigEntry
|
|
||||||
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.exceptions import HomeAssistantError
|
from homeassistant.exceptions import HomeAssistantError
|
||||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
||||||
|
|
||||||
|
if TYPE_CHECKING:
|
||||||
|
from . import TadoConfigEntry
|
||||||
|
|
||||||
from .const import (
|
from .const import (
|
||||||
CONF_FALLBACK,
|
CONF_FALLBACK,
|
||||||
CONST_OVERLAY_TADO_DEFAULT,
|
CONST_OVERLAY_TADO_DEFAULT,
|
||||||
@ -31,8 +33,6 @@ MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=4)
|
|||||||
SCAN_INTERVAL = timedelta(minutes=5)
|
SCAN_INTERVAL = timedelta(minutes=5)
|
||||||
SCAN_MOBILE_DEVICE_INTERVAL = timedelta(seconds=30)
|
SCAN_MOBILE_DEVICE_INTERVAL = timedelta(seconds=30)
|
||||||
|
|
||||||
type TadoConfigEntry = ConfigEntry[TadoDataUpdateCoordinator]
|
|
||||||
|
|
||||||
|
|
||||||
class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
|
class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
|
||||||
"""Class to manage API calls from and to Tado via PyTado."""
|
"""Class to manage API calls from and to Tado via PyTado."""
|
||||||
@ -45,7 +45,7 @@ class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
|
|||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
entry: ConfigEntry,
|
config_entry: TadoConfigEntry,
|
||||||
tado: Tado,
|
tado: Tado,
|
||||||
debug: bool = False,
|
debug: bool = False,
|
||||||
) -> None:
|
) -> None:
|
||||||
@ -53,13 +53,16 @@ class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
|
|||||||
super().__init__(
|
super().__init__(
|
||||||
hass,
|
hass,
|
||||||
_LOGGER,
|
_LOGGER,
|
||||||
|
config_entry=config_entry,
|
||||||
name=DOMAIN,
|
name=DOMAIN,
|
||||||
update_interval=SCAN_INTERVAL,
|
update_interval=SCAN_INTERVAL,
|
||||||
)
|
)
|
||||||
self._tado = tado
|
self._tado = tado
|
||||||
self._username = entry.data[CONF_USERNAME]
|
self._username = config_entry.data[CONF_USERNAME]
|
||||||
self._password = entry.data[CONF_PASSWORD]
|
self._password = config_entry.data[CONF_PASSWORD]
|
||||||
self._fallback = entry.options.get(CONF_FALLBACK, CONST_OVERLAY_TADO_DEFAULT)
|
self._fallback = config_entry.options.get(
|
||||||
|
CONF_FALLBACK, CONST_OVERLAY_TADO_DEFAULT
|
||||||
|
)
|
||||||
self._debug = debug
|
self._debug = debug
|
||||||
|
|
||||||
self.home_id: int
|
self.home_id: int
|
||||||
@ -343,16 +346,19 @@ class TadoDataUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
|
|||||||
class TadoMobileDeviceUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
|
class TadoMobileDeviceUpdateCoordinator(DataUpdateCoordinator[dict[str, dict]]):
|
||||||
"""Class to manage the mobile devices from Tado via PyTado."""
|
"""Class to manage the mobile devices from Tado via PyTado."""
|
||||||
|
|
||||||
|
config_entry: TadoConfigEntry
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
entry: ConfigEntry,
|
config_entry: TadoConfigEntry,
|
||||||
tado: Tado,
|
tado: Tado,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Initialize the Tado data update coordinator."""
|
"""Initialize the Tado data update coordinator."""
|
||||||
super().__init__(
|
super().__init__(
|
||||||
hass,
|
hass,
|
||||||
_LOGGER,
|
_LOGGER,
|
||||||
|
config_entry=config_entry,
|
||||||
name=DOMAIN,
|
name=DOMAIN,
|
||||||
update_interval=SCAN_MOBILE_DEVICE_INTERVAL,
|
update_interval=SCAN_MOBILE_DEVICE_INTERVAL,
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user