mirror of
https://github.com/home-assistant/core.git
synced 2025-05-02 13:17:53 +00:00
Explicitly pass in the config_entry in nanoleaf coordinator (#138075)
explicitly pass in the config_entry in coordinator
This commit is contained in:
parent
41532ed509
commit
711d423877
@ -8,7 +8,6 @@ import logging
|
|||||||
|
|
||||||
from aionanoleaf import EffectsEvent, Nanoleaf, StateEvent, TouchEvent
|
from aionanoleaf import EffectsEvent, Nanoleaf, StateEvent, TouchEvent
|
||||||
|
|
||||||
from homeassistant.config_entries import ConfigEntry
|
|
||||||
from homeassistant.const import (
|
from homeassistant.const import (
|
||||||
CONF_DEVICE_ID,
|
CONF_DEVICE_ID,
|
||||||
CONF_HOST,
|
CONF_HOST,
|
||||||
@ -22,23 +21,20 @@ from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|||||||
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
||||||
|
|
||||||
from .const import DOMAIN, NANOLEAF_EVENT, TOUCH_GESTURE_TRIGGER_MAP, TOUCH_MODELS
|
from .const import DOMAIN, NANOLEAF_EVENT, TOUCH_GESTURE_TRIGGER_MAP, TOUCH_MODELS
|
||||||
from .coordinator import NanoleafCoordinator
|
from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
|
||||||
|
|
||||||
_LOGGER = logging.getLogger(__name__)
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
PLATFORMS = [Platform.BUTTON, Platform.EVENT, Platform.LIGHT]
|
PLATFORMS = [Platform.BUTTON, Platform.EVENT, Platform.LIGHT]
|
||||||
|
|
||||||
|
|
||||||
type NanoleafConfigEntry = ConfigEntry[NanoleafCoordinator]
|
|
||||||
|
|
||||||
|
|
||||||
async def async_setup_entry(hass: HomeAssistant, entry: NanoleafConfigEntry) -> bool:
|
async def async_setup_entry(hass: HomeAssistant, entry: NanoleafConfigEntry) -> bool:
|
||||||
"""Set up Nanoleaf from a config entry."""
|
"""Set up Nanoleaf from a config entry."""
|
||||||
nanoleaf = Nanoleaf(
|
nanoleaf = Nanoleaf(
|
||||||
async_get_clientsession(hass), entry.data[CONF_HOST], entry.data[CONF_TOKEN]
|
async_get_clientsession(hass), entry.data[CONF_HOST], entry.data[CONF_TOKEN]
|
||||||
)
|
)
|
||||||
|
|
||||||
coordinator = NanoleafCoordinator(hass, nanoleaf)
|
coordinator = NanoleafCoordinator(hass, entry, nanoleaf)
|
||||||
|
|
||||||
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 NanoleafConfigEntry
|
from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
|
||||||
from .coordinator import NanoleafCoordinator
|
|
||||||
from .entity import NanoleafEntity
|
from .entity import NanoleafEntity
|
||||||
|
|
||||||
|
|
||||||
|
@ -5,20 +5,31 @@ import logging
|
|||||||
|
|
||||||
from aionanoleaf import InvalidToken, Nanoleaf, Unavailable
|
from aionanoleaf import InvalidToken, Nanoleaf, Unavailable
|
||||||
|
|
||||||
|
from homeassistant.config_entries import ConfigEntry
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.exceptions import ConfigEntryAuthFailed
|
from homeassistant.exceptions import ConfigEntryAuthFailed
|
||||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
||||||
|
|
||||||
_LOGGER = logging.getLogger(__name__)
|
_LOGGER = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
type NanoleafConfigEntry = ConfigEntry[NanoleafCoordinator]
|
||||||
|
|
||||||
|
|
||||||
class NanoleafCoordinator(DataUpdateCoordinator[None]):
|
class NanoleafCoordinator(DataUpdateCoordinator[None]):
|
||||||
"""Class to manage fetching Nanoleaf data."""
|
"""Class to manage fetching Nanoleaf data."""
|
||||||
|
|
||||||
def __init__(self, hass: HomeAssistant, nanoleaf: Nanoleaf) -> None:
|
config_entry: NanoleafConfigEntry
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self, hass: HomeAssistant, config_entry: NanoleafConfigEntry, nanoleaf: Nanoleaf
|
||||||
|
) -> None:
|
||||||
"""Initialize the Nanoleaf data coordinator."""
|
"""Initialize the Nanoleaf data coordinator."""
|
||||||
super().__init__(
|
super().__init__(
|
||||||
hass, _LOGGER, name="Nanoleaf", update_interval=timedelta(minutes=1)
|
hass,
|
||||||
|
_LOGGER,
|
||||||
|
config_entry=config_entry,
|
||||||
|
name="Nanoleaf",
|
||||||
|
update_interval=timedelta(minutes=1),
|
||||||
)
|
)
|
||||||
self.nanoleaf = nanoleaf
|
self.nanoleaf = nanoleaf
|
||||||
|
|
||||||
|
@ -8,7 +8,7 @@ from homeassistant.components.diagnostics import async_redact_data
|
|||||||
from homeassistant.const import CONF_TOKEN
|
from homeassistant.const import CONF_TOKEN
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
from . import NanoleafConfigEntry
|
from .coordinator import NanoleafConfigEntry
|
||||||
|
|
||||||
|
|
||||||
async def async_get_config_entry_diagnostics(
|
async def async_get_config_entry_diagnostics(
|
||||||
|
@ -3,8 +3,8 @@
|
|||||||
from homeassistant.helpers.device_registry import DeviceInfo
|
from homeassistant.helpers.device_registry import DeviceInfo
|
||||||
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||||||
|
|
||||||
from . import NanoleafCoordinator
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
from .coordinator import NanoleafCoordinator
|
||||||
|
|
||||||
|
|
||||||
class NanoleafEntity(CoordinatorEntity[NanoleafCoordinator]):
|
class NanoleafEntity(CoordinatorEntity[NanoleafCoordinator]):
|
||||||
|
@ -5,8 +5,8 @@ from homeassistant.core import HomeAssistant, callback
|
|||||||
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
from . import NanoleafConfigEntry, NanoleafCoordinator
|
|
||||||
from .const import TOUCH_MODELS
|
from .const import TOUCH_MODELS
|
||||||
|
from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
|
||||||
from .entity import NanoleafEntity
|
from .entity import NanoleafEntity
|
||||||
|
|
||||||
|
|
||||||
|
@ -17,8 +17,7 @@ from homeassistant.components.light import (
|
|||||||
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 NanoleafConfigEntry
|
from .coordinator import NanoleafConfigEntry, NanoleafCoordinator
|
||||||
from .coordinator import NanoleafCoordinator
|
|
||||||
from .entity import NanoleafEntity
|
from .entity import NanoleafEntity
|
||||||
|
|
||||||
RESERVED_EFFECTS = ("*Solid*", "*Static*", "*Dynamic*")
|
RESERVED_EFFECTS = ("*Solid*", "*Static*", "*Dynamic*")
|
||||||
|
Loading…
x
Reference in New Issue
Block a user