Explicitly pass in the config_entry in hunterdouglas_powerview coordinator (#138151)

explicitly pass in the config_entry in coordinator
pull/138159/head
Michael 2025-02-09 21:19:57 +01:00 committed by GitHub
parent 39bcef63bd
commit 6b24bae084
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 2 deletions

View File

@ -91,7 +91,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: PowerviewConfigEntry) ->
entry, unique_id=device_info.serial_number
)
coordinator = PowerviewShadeUpdateCoordinator(hass, shades, hub)
coordinator = PowerviewShadeUpdateCoordinator(hass, entry, shades, hub)
coordinator.async_set_updated_data(PowerviewShadeData())
# populate raw shade data into the coordinator for diagnostics
coordinator.data.store_group_data(shade_data)

View File

@ -10,6 +10,7 @@ from aiopvapi.helpers.aiorequest import PvApiMaintenance
from aiopvapi.hub import Hub
from aiopvapi.shades import Shades
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@ -22,7 +23,11 @@ _LOGGER = logging.getLogger(__name__)
class PowerviewShadeUpdateCoordinator(DataUpdateCoordinator[PowerviewShadeData]):
"""DataUpdateCoordinator to gather data from a powerview hub."""
def __init__(self, hass: HomeAssistant, shades: Shades, hub: Hub) -> None:
config_entry: ConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: ConfigEntry, shades: Shades, hub: Hub
) -> None:
"""Initialize DataUpdateCoordinator to gather data for specific Powerview Hub."""
self.shades = shades
self.hub = hub
@ -33,6 +38,7 @@ class PowerviewShadeUpdateCoordinator(DataUpdateCoordinator[PowerviewShadeData])
super().__init__(
hass,
_LOGGER,
config_entry=config_entry,
name=f"powerview hub {hub.hub_address}",
update_interval=timedelta(seconds=60),
)