Add Switch entity to SleepIQ (#66966)

Co-authored-by: J. Nick Koston <nick@koston.org>
This commit is contained in:
Keilin Bickar 2022-02-22 02:53:04 -05:00 committed by GitHub
parent e0ff7dc2e7
commit 7f5304b6c2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 214 additions and 28 deletions

View file

@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN
from .coordinator import SleepIQDataUpdateCoordinator
from .coordinator import SleepIQData
from .entity import SleepIQEntity
@ -53,11 +53,11 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up the sleep number buttons."""
coordinator: SleepIQDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
data: SleepIQData = hass.data[DOMAIN][entry.entry_id]
async_add_entities(
SleepNumberButton(bed, ed)
for bed in coordinator.client.beds.values()
for bed in data.client.beds.values()
for ed in ENTITY_DESCRIPTIONS
)