evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Migrate switchbot to use entry.runtime_data (#122530)

Browse source
This commit is contained in:
J. Nick Koston 2024-07-25 05:18:24 -05:00 • committed by GitHub
parent 3b01a57de3
commit a89853da9d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 40 additions and 58 deletions
Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/switchbot/cover.py
View file

@ -16,13 +16,11 @@ from homeassistant.components.cover import (
CoverEntity,
CoverEntityFeature,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.restore_state import RestoreEntity
from .const import DOMAIN
from .coordinator import SwitchbotDataUpdateCoordinator
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
from .entity import SwitchbotEntity
# Initialize the logger
@ -31,10 +29,12 @@ PARALLEL_UPDATES = 0
async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
hass: HomeAssistant,
entry: SwitchbotConfigEntry,
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up Switchbot curtain based on a config entry."""
coordinator: SwitchbotDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
coordinator = entry.runtime_data
if isinstance(coordinator.device, switchbot.SwitchbotBlindTilt):
async_add_entities([SwitchBotBlindTiltEntity(coordinator)])
else: