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

Explicitly pass in the config_entry in nice_go coordinator (#138070)

Browse Source 
explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 16:31:48 +01:00 committed by GitHub
parent b7949dba44
commit ac9444a9ba
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 13 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/nice_go/__init__.py
View File

@ -4,11 +4,10 @@ from __future__ import annotations
import logging import logging
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EVENT_HOMEASSISTANT_STOP, Platform from homeassistant.const import EVENT_HOMEASSISTANT_STOP, Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from .coordinator import NiceGOUpdateCoordinator from .coordinator import NiceGOConfigEntry, NiceGOUpdateCoordinator
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
PLATFORMS: list[Platform] = [ PLATFORMS: list[Platform] = [
@ -18,13 +17,11 @@ PLATFORMS: list[Platform] = [
Platform.SWITCH, Platform.SWITCH,
] ]
type NiceGOConfigEntry = ConfigEntry[NiceGOUpdateCoordinator]
async def async_setup_entry(hass: HomeAssistant, entry: NiceGOConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: NiceGOConfigEntry) -> bool:
"""Set up Nice G.O. from a config entry.""" """Set up Nice G.O. from a config entry."""
coordinator = NiceGOUpdateCoordinator(hass) coordinator = NiceGOUpdateCoordinator(hass, entry)
entry.async_on_unload( entry.async_on_unload(
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, coordinator.async_ha_stop) hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, coordinator.async_ha_stop)
) )

13
homeassistant/components/nice_go/coordinator.py
View File

@ -54,19 +54,18 @@ class NiceGODevice:
vacation_mode: bool | None vacation_mode: bool | None
type NiceGOConfigEntry = ConfigEntry[NiceGOUpdateCoordinator]
class NiceGOUpdateCoordinator(DataUpdateCoordinator[dict[str, NiceGODevice]]): class NiceGOUpdateCoordinator(DataUpdateCoordinator[dict[str, NiceGODevice]]):
"""DataUpdateCoordinator for Nice G.O.""" """DataUpdateCoordinator for Nice G.O."""
config_entry: ConfigEntry config_entry: NiceGOConfigEntry
organization_id: str organization_id: str
def __init__(self, hass: HomeAssistant) -> None: def __init__(self, hass: HomeAssistant, config_entry: NiceGOConfigEntry) -> None:
"""Initialize DataUpdateCoordinator for Nice G.O.""" """Initialize DataUpdateCoordinator for Nice G.O."""
super().__init__( super().__init__(hass, _LOGGER, config_entry=config_entry, name="Nice G.O.")
hass,
_LOGGER,
name="Nice G.O.",
)
self.refresh_token = self.config_entry.data[CONF_REFRESH_TOKEN] self.refresh_token = self.config_entry.data[CONF_REFRESH_TOKEN]
self.refresh_token_creation_time = self.config_entry.data[ self.refresh_token_creation_time = self.config_entry.data[

2
homeassistant/components/nice_go/cover.py
View File

@ -14,8 +14,8 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import NiceGOConfigEntry
from .const import DOMAIN from .const import DOMAIN
from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity from .entity import NiceGOEntity
DEVICE_CLASSES = { DEVICE_CLASSES = {

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

@ -9,8 +9,8 @@ from homeassistant.components.diagnostics import async_redact_data
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from . import NiceGOConfigEntry
from .const import CONF_REFRESH_TOKEN from .const import CONF_REFRESH_TOKEN
from .coordinator import NiceGOConfigEntry
TO_REDACT = {CONF_PASSWORD, CONF_EMAIL, CONF_REFRESH_TOKEN, "title", "unique_id"} TO_REDACT = {CONF_PASSWORD, CONF_EMAIL, CONF_REFRESH_TOKEN, "title", "unique_id"}

2
homeassistant/components/nice_go/event.py
View File

@ -7,7 +7,7 @@ from homeassistant.components.event import EventEntity
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 NiceGOConfigEntry from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity from .entity import NiceGOEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)

2
homeassistant/components/nice_go/light.py
View File

@ -12,13 +12,13 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import NiceGOConfigEntry
from .const import ( from .const import (
DOMAIN, DOMAIN,
KNOWN_UNSUPPORTED_DEVICE_TYPES, KNOWN_UNSUPPORTED_DEVICE_TYPES,
SUPPORTED_DEVICE_TYPES, SUPPORTED_DEVICE_TYPES,
UNSUPPORTED_DEVICE_WARNING, UNSUPPORTED_DEVICE_WARNING,
) )
from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity from .entity import NiceGOEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)

2
homeassistant/components/nice_go/switch.py
View File

@ -14,13 +14,13 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import NiceGOConfigEntry
from .const import ( from .const import (
DOMAIN, DOMAIN,
KNOWN_UNSUPPORTED_DEVICE_TYPES, KNOWN_UNSUPPORTED_DEVICE_TYPES,
SUPPORTED_DEVICE_TYPES, SUPPORTED_DEVICE_TYPES,
UNSUPPORTED_DEVICE_WARNING, UNSUPPORTED_DEVICE_WARNING,
) )
from .coordinator import NiceGOConfigEntry
from .entity import NiceGOEntity from .entity import NiceGOEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)