jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 22:57:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/homewizard/__init__.py
View File

@ -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()

3
homeassistant/components/homewizard/button.py
View File

@ -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

19
homeassistant/components/homewizard/coordinator.py
View File

@ -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:

2
homeassistant/components/homewizard/diagnostics.py
View File

@ -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,

3
homeassistant/components/homewizard/number.py
View File

@ -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

3
homeassistant/components/homewizard/sensor.py
View File

@ -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

3
homeassistant/components/homewizard/switch.py
View File

@ -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