mirror of
https://github.com/home-assistant/core.git
synced 2025-07-12 15:57:06 +00:00
Explicitly pass in the config_entry in autarco coordinator init (#137718)
explicitly pass in the config_entry in autarco coordinator init
This commit is contained in:
parent
2eb6a03640
commit
1f0f1ac388
@ -6,18 +6,15 @@ import asyncio
|
|||||||
|
|
||||||
from autarco import Autarco, AutarcoConnectionError
|
from autarco import Autarco, AutarcoConnectionError
|
||||||
|
|
||||||
from homeassistant.config_entries import ConfigEntry
|
|
||||||
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
|
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, Platform
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.exceptions import ConfigEntryNotReady
|
from homeassistant.exceptions import ConfigEntryNotReady
|
||||||
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
||||||
|
|
||||||
from .coordinator import AutarcoDataUpdateCoordinator
|
from .coordinator import AutarcoConfigEntry, AutarcoDataUpdateCoordinator
|
||||||
|
|
||||||
PLATFORMS: list[Platform] = [Platform.SENSOR]
|
PLATFORMS: list[Platform] = [Platform.SENSOR]
|
||||||
|
|
||||||
type AutarcoConfigEntry = ConfigEntry[list[AutarcoDataUpdateCoordinator]]
|
|
||||||
|
|
||||||
|
|
||||||
async def async_setup_entry(hass: HomeAssistant, entry: AutarcoConfigEntry) -> bool:
|
async def async_setup_entry(hass: HomeAssistant, entry: AutarcoConfigEntry) -> bool:
|
||||||
"""Set up Autarco from a config entry."""
|
"""Set up Autarco from a config entry."""
|
||||||
@ -34,7 +31,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: AutarcoConfigEntry) -> b
|
|||||||
raise ConfigEntryNotReady from err
|
raise ConfigEntryNotReady from err
|
||||||
|
|
||||||
coordinators: list[AutarcoDataUpdateCoordinator] = [
|
coordinators: list[AutarcoDataUpdateCoordinator] = [
|
||||||
AutarcoDataUpdateCoordinator(hass, client, site) for site in account_sites
|
AutarcoDataUpdateCoordinator(hass, entry, client, site)
|
||||||
|
for site in account_sites
|
||||||
]
|
]
|
||||||
|
|
||||||
await asyncio.gather(
|
await asyncio.gather(
|
||||||
|
@ -22,6 +22,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
|
|||||||
|
|
||||||
from .const import DOMAIN, LOGGER, SCAN_INTERVAL
|
from .const import DOMAIN, LOGGER, SCAN_INTERVAL
|
||||||
|
|
||||||
|
type AutarcoConfigEntry = ConfigEntry[list[AutarcoDataUpdateCoordinator]]
|
||||||
|
|
||||||
|
|
||||||
class AutarcoData(NamedTuple):
|
class AutarcoData(NamedTuple):
|
||||||
"""Class for defining data in dict."""
|
"""Class for defining data in dict."""
|
||||||
@ -35,11 +37,12 @@ class AutarcoData(NamedTuple):
|
|||||||
class AutarcoDataUpdateCoordinator(DataUpdateCoordinator[AutarcoData]):
|
class AutarcoDataUpdateCoordinator(DataUpdateCoordinator[AutarcoData]):
|
||||||
"""Class to manage fetching Autarco data from the API."""
|
"""Class to manage fetching Autarco data from the API."""
|
||||||
|
|
||||||
config_entry: ConfigEntry
|
config_entry: AutarcoConfigEntry
|
||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant,
|
||||||
|
config_entry: AutarcoConfigEntry,
|
||||||
client: Autarco,
|
client: Autarco,
|
||||||
account_site: AccountSite,
|
account_site: AccountSite,
|
||||||
) -> None:
|
) -> None:
|
||||||
@ -47,6 +50,7 @@ class AutarcoDataUpdateCoordinator(DataUpdateCoordinator[AutarcoData]):
|
|||||||
super().__init__(
|
super().__init__(
|
||||||
hass,
|
hass,
|
||||||
LOGGER,
|
LOGGER,
|
||||||
|
config_entry=config_entry,
|
||||||
name=DOMAIN,
|
name=DOMAIN,
|
||||||
update_interval=SCAN_INTERVAL,
|
update_interval=SCAN_INTERVAL,
|
||||||
)
|
)
|
||||||
|
@ -6,7 +6,7 @@ from typing import Any
|
|||||||
|
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
from . import AutarcoConfigEntry, AutarcoDataUpdateCoordinator
|
from .coordinator import AutarcoConfigEntry, AutarcoDataUpdateCoordinator
|
||||||
|
|
||||||
|
|
||||||
async def async_get_config_entry_diagnostics(
|
async def async_get_config_entry_diagnostics(
|
||||||
|
@ -20,9 +20,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|||||||
from homeassistant.helpers.typing import StateType
|
from homeassistant.helpers.typing import StateType
|
||||||
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||||||
|
|
||||||
from . import AutarcoConfigEntry
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
from .coordinator import AutarcoDataUpdateCoordinator
|
from .coordinator import AutarcoConfigEntry, AutarcoDataUpdateCoordinator
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, kw_only=True)
|
@dataclass(frozen=True, kw_only=True)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user