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

Explicitly pass in the config_entry in tessie coordinator (#137906)

Browse Source 
explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 14:42:04 +01:00 committed by GitHub
parent 4646d35054
commit 0baa6b3668
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 29 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
homeassistant/components/tessie/__init__.py
View File

@ -69,6 +69,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TessieConfigEntry) -> bo
vin=vehicle["vin"],
data_coordinator=TessieStateUpdateCoordinator(
hass,
entry,
api_key=api_key,
vin=vehicle["vin"],
data=vehicle["last_state"],
@ -127,8 +128,12 @@ async def async_setup_entry(hass: HomeAssistant, entry: TessieConfigEntry) -> bo
TessieEnergyData(
api=api,
id=site_id,
live_coordinator=TessieEnergySiteLiveCoordinator(hass, api),
info_coordinator=TessieEnergySiteInfoCoordinator(hass, api),
live_coordinator=TessieEnergySiteLiveCoordinator(
hass, entry, api
),
info_coordinator=TessieEnergySiteInfoCoordinator(
hass, entry, api
),
device=DeviceInfo(
identifiers={(DOMAIN, str(site_id))},
manufacturer="Tesla",

25
homeassistant/components/tessie/coordinator.py
View File

@ -1,9 +1,11 @@
"""Tessie Data Coordinator."""
from __future__ import annotations
from datetime import timedelta
from http import HTTPStatus
import logging
from typing import Any
from typing import TYPE_CHECKING, Any
from aiohttp import ClientResponseError
from tesla_fleet_api import EnergySpecific
@ -15,6 +17,9 @@ from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
if TYPE_CHECKING:
from . import TessieConfigEntry
from .const import TessieStatus
# This matches the update interval Tessie performs server side
@ -40,9 +45,12 @@ def flatten(data: dict[str, Any], parent: str | None = None) -> dict[str, Any]:
class TessieStateUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching data from the Tessie API."""
config_entry: TessieConfigEntry
def __init__(
self,
hass: HomeAssistant,
config_entry: TessieConfigEntry,
api_key: str,
vin: str,
data: dict[str, Any],
@ -51,6 +59,7 @@ class TessieStateUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
super().__init__(
hass,
_LOGGER,
config_entry=config_entry,
name="Tessie",
update_interval=timedelta(seconds=TESSIE_SYNC_INTERVAL),
)
@ -90,11 +99,16 @@ class TessieStateUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class TessieEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site live status from the Tessie API."""
def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
config_entry: TessieConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: TessieConfigEntry, api: EnergySpecific
) -> None:
"""Initialize Tessie Energy Site Live coordinator."""
super().__init__(
hass,
_LOGGER,
config_entry=config_entry,
name="Tessie Energy Site Live",
update_interval=TESSIE_FLEET_API_SYNC_INTERVAL,
)
@ -121,11 +135,16 @@ class TessieEnergySiteLiveCoordinator(DataUpdateCoordinator[dict[str, Any]]):
class TessieEnergySiteInfoCoordinator(DataUpdateCoordinator[dict[str, Any]]):
"""Class to manage fetching energy site info from the Tessie API."""
def __init__(self, hass: HomeAssistant, api: EnergySpecific) -> None:
config_entry: TessieConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: TessieConfigEntry, api: EnergySpecific
) -> None:
"""Initialize Tessie Energy Info coordinator."""
super().__init__(
hass,
_LOGGER,
config_entry=config_entry,
name="Tessie Energy Site Info",
update_interval=TESSIE_FLEET_API_SYNC_INTERVAL,
)