Explicitly pass in the config_entry in homewizard coordinator (#138152)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 21:10:21 +01:00 committed by GitHub
parent 8c27a75d6b
commit 8c3dab199e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 24 additions and 17 deletions

View File

@ -7,7 +7,7 @@ from homewizard_energy import (
has_v2_api,
)
from homeassistant.config_entries import SOURCE_REAUTH, ConfigEntry
from homeassistant.config_entries import SOURCE_REAUTH
from homeassistant.const import CONF_IP_ADDRESS, CONF_TOKEN
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
@ -15,9 +15,7 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from .const import DOMAIN, PLATFORMS
from .coordinator import HWEnergyDeviceUpdateCoordinator
type HomeWizardConfigEntry = ConfigEntry[HWEnergyDeviceUpdateCoordinator]
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
async def async_setup_entry(hass: HomeAssistant, entry: HomeWizardConfigEntry) -> bool:
@ -42,7 +40,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: HomeWizardConfigEntry) -
if is_battery:
await async_check_v2_support_and_create_issue(hass, entry)
coordinator = HWEnergyDeviceUpdateCoordinator(hass, api)
coordinator = HWEnergyDeviceUpdateCoordinator(hass, entry, api)
try:
await coordinator.async_config_entry_first_refresh()

View File

@ -5,8 +5,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import HomeWizardConfigEntry
from .coordinator import HWEnergyDeviceUpdateCoordinator
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
from .helpers import homewizard_exception_handler

View File

@ -13,6 +13,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
from .const import DOMAIN, LOGGER, UPDATE_INTERVAL
type HomeWizardConfigEntry = ConfigEntry[HWEnergyDeviceUpdateCoordinator]
class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator[DeviceResponseEntry]):
"""Gather data for the energy device."""
@ -20,11 +22,22 @@ class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator[DeviceResponseEntry]
api: HomeWizardEnergy
api_disabled: bool = False
config_entry: ConfigEntry
config_entry: HomeWizardConfigEntry
def __init__(self, hass: HomeAssistant, api: HomeWizardEnergy) -> None:
def __init__(
self,
hass: HomeAssistant,
config_entry: HomeWizardConfigEntry,
api: HomeWizardEnergy,
) -> None:
"""Initialize update coordinator."""
super().__init__(hass, LOGGER, name=DOMAIN, update_interval=UPDATE_INTERVAL)
super().__init__(
hass,
LOGGER,
config_entry=config_entry,
name=DOMAIN,
update_interval=UPDATE_INTERVAL,
)
self.api = api
async def _async_update_data(self) -> DeviceResponseEntry:

View File

@ -9,7 +9,7 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_IP_ADDRESS
from homeassistant.core import HomeAssistant
from . import HomeWizardConfigEntry
from .coordinator import HomeWizardConfigEntry
TO_REDACT = {
CONF_IP_ADDRESS,

View File

@ -7,8 +7,7 @@ from homeassistant.const import PERCENTAGE, EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import HomeWizardConfigEntry
from .coordinator import HWEnergyDeviceUpdateCoordinator
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
from .helpers import homewizard_exception_handler

View File

@ -37,9 +37,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType
from homeassistant.util.dt import utcnow
from . import HomeWizardConfigEntry
from .const import DOMAIN
from .coordinator import HWEnergyDeviceUpdateCoordinator
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
PARALLEL_UPDATES = 1

View File

@ -18,8 +18,7 @@ from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import HomeWizardConfigEntry
from .coordinator import HWEnergyDeviceUpdateCoordinator
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
from .entity import HomeWizardEntity
from .helpers import homewizard_exception_handler