mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 21:27:38 +00:00
Explicitly pass in the config_entry in homewizard coordinator (#138152)
explicitly pass in the config_entry in coordinator
This commit is contained in:
parent
8c27a75d6b
commit
8c3dab199e
@ -7,7 +7,7 @@ from homewizard_energy import (
|
|||||||
has_v2_api,
|
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.const import CONF_IP_ADDRESS, CONF_TOKEN
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.exceptions import ConfigEntryNotReady
|
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 homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
|
||||||
|
|
||||||
from .const import DOMAIN, PLATFORMS
|
from .const import DOMAIN, PLATFORMS
|
||||||
from .coordinator import HWEnergyDeviceUpdateCoordinator
|
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
|
||||||
|
|
||||||
type HomeWizardConfigEntry = ConfigEntry[HWEnergyDeviceUpdateCoordinator]
|
|
||||||
|
|
||||||
|
|
||||||
async def async_setup_entry(hass: HomeAssistant, entry: HomeWizardConfigEntry) -> bool:
|
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:
|
if is_battery:
|
||||||
await async_check_v2_support_and_create_issue(hass, entry)
|
await async_check_v2_support_and_create_issue(hass, entry)
|
||||||
|
|
||||||
coordinator = HWEnergyDeviceUpdateCoordinator(hass, api)
|
coordinator = HWEnergyDeviceUpdateCoordinator(hass, entry, api)
|
||||||
try:
|
try:
|
||||||
await coordinator.async_config_entry_first_refresh()
|
await coordinator.async_config_entry_first_refresh()
|
||||||
|
|
||||||
|
@ -5,8 +5,7 @@ from homeassistant.const import EntityCategory
|
|||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
from . import HomeWizardConfigEntry
|
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
|
||||||
from .coordinator import HWEnergyDeviceUpdateCoordinator
|
|
||||||
from .entity import HomeWizardEntity
|
from .entity import HomeWizardEntity
|
||||||
from .helpers import homewizard_exception_handler
|
from .helpers import homewizard_exception_handler
|
||||||
|
|
||||||
|
@ -13,6 +13,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
|
|||||||
|
|
||||||
from .const import DOMAIN, LOGGER, UPDATE_INTERVAL
|
from .const import DOMAIN, LOGGER, UPDATE_INTERVAL
|
||||||
|
|
||||||
|
type HomeWizardConfigEntry = ConfigEntry[HWEnergyDeviceUpdateCoordinator]
|
||||||
|
|
||||||
|
|
||||||
class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator[DeviceResponseEntry]):
|
class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator[DeviceResponseEntry]):
|
||||||
"""Gather data for the energy device."""
|
"""Gather data for the energy device."""
|
||||||
@ -20,11 +22,22 @@ class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator[DeviceResponseEntry]
|
|||||||
api: HomeWizardEnergy
|
api: HomeWizardEnergy
|
||||||
api_disabled: bool = False
|
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."""
|
"""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
|
self.api = api
|
||||||
|
|
||||||
async def _async_update_data(self) -> DeviceResponseEntry:
|
async def _async_update_data(self) -> DeviceResponseEntry:
|
||||||
|
@ -9,7 +9,7 @@ from homeassistant.components.diagnostics import async_redact_data
|
|||||||
from homeassistant.const import CONF_IP_ADDRESS
|
from homeassistant.const import CONF_IP_ADDRESS
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
from . import HomeWizardConfigEntry
|
from .coordinator import HomeWizardConfigEntry
|
||||||
|
|
||||||
TO_REDACT = {
|
TO_REDACT = {
|
||||||
CONF_IP_ADDRESS,
|
CONF_IP_ADDRESS,
|
||||||
|
@ -7,8 +7,7 @@ from homeassistant.const import PERCENTAGE, EntityCategory
|
|||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
from . import HomeWizardConfigEntry
|
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
|
||||||
from .coordinator import HWEnergyDeviceUpdateCoordinator
|
|
||||||
from .entity import HomeWizardEntity
|
from .entity import HomeWizardEntity
|
||||||
from .helpers import homewizard_exception_handler
|
from .helpers import homewizard_exception_handler
|
||||||
|
|
||||||
|
@ -37,9 +37,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|||||||
from homeassistant.helpers.typing import StateType
|
from homeassistant.helpers.typing import StateType
|
||||||
from homeassistant.util.dt import utcnow
|
from homeassistant.util.dt import utcnow
|
||||||
|
|
||||||
from . import HomeWizardConfigEntry
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
from .coordinator import HWEnergyDeviceUpdateCoordinator
|
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
|
||||||
from .entity import HomeWizardEntity
|
from .entity import HomeWizardEntity
|
||||||
|
|
||||||
PARALLEL_UPDATES = 1
|
PARALLEL_UPDATES = 1
|
||||||
|
@ -18,8 +18,7 @@ from homeassistant.const import EntityCategory
|
|||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
from . import HomeWizardConfigEntry
|
from .coordinator import HomeWizardConfigEntry, HWEnergyDeviceUpdateCoordinator
|
||||||
from .coordinator import HWEnergyDeviceUpdateCoordinator
|
|
||||||
from .entity import HomeWizardEntity
|
from .entity import HomeWizardEntity
|
||||||
from .helpers import homewizard_exception_handler
|
from .helpers import homewizard_exception_handler
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user