Move esphome coordinator to separate module (#117427)

This commit is contained in:
epenet 2024-05-14 15:16:47 +02:00 committed by GitHub
parent 92bb76ed24
commit 2e155f4de5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 84 additions and 71 deletions

View file

@ -20,7 +20,8 @@ from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .dashboard import ESPHomeDashboard, async_get_dashboard
from .coordinator import ESPHomeDashboardCoordinator
from .dashboard import async_get_dashboard
from .domain_data import DomainData
from .entry_data import RuntimeEntryData
@ -65,7 +66,7 @@ async def async_setup_entry(
]
class ESPHomeUpdateEntity(CoordinatorEntity[ESPHomeDashboard], UpdateEntity):
class ESPHomeUpdateEntity(CoordinatorEntity[ESPHomeDashboardCoordinator], UpdateEntity):
"""Defines an ESPHome update entity."""
_attr_has_entity_name = True
@ -75,7 +76,7 @@ class ESPHomeUpdateEntity(CoordinatorEntity[ESPHomeDashboard], UpdateEntity):
_attr_release_url = "https://esphome.io/changelog/"
def __init__(
self, entry_data: RuntimeEntryData, coordinator: ESPHomeDashboard
self, entry_data: RuntimeEntryData, coordinator: ESPHomeDashboardCoordinator
) -> None:
"""Initialize the update entity."""
super().__init__(coordinator=coordinator)