evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add data model to Tessie (#106285)

Browse source 
Add data model
This commit is contained in:
Brett Adams 2023-12-25 22:01:13 +10:00 • committed by GitHub
parent 123b2669f3
commit 04a56eaabe
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
16 changed files with 92 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/tessie/update.py
View file

@ -7,7 +7,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN, TessieUpdateStatus
from .coordinator import TessieDataUpdateCoordinator
from .coordinator import TessieStateUpdateCoordinator
from .entity import TessieEntity
@ -15,9 +15,11 @@ async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None:
"""Set up the Tessie Update platform from a config entry."""
coordinators = hass.data[DOMAIN][entry.entry_id]
data = hass.data[DOMAIN][entry.entry_id]
async_add_entities(TessieUpdateEntity(coordinator) for coordinator in coordinators)
async_add_entities(
TessieUpdateEntity(vehicle.state_coordinator) for vehicle in data
)
class TessieUpdateEntity(TessieEntity, UpdateEntity):
@ -28,7 +30,7 @@ class TessieUpdateEntity(TessieEntity, UpdateEntity):
def __init__(
self,
coordinator: TessieDataUpdateCoordinator,
coordinator: TessieStateUpdateCoordinator,
) -> None:
"""Initialize the Update."""
super().__init__(coordinator, "update")