Migrate switchbot to use entry.runtime_data (#122530)

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

View file

@ -9,7 +9,6 @@ from homeassistant.components.sensor import (
SensorEntityDescription,
SensorStateClass,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
PERCENTAGE,
SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
@ -20,8 +19,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN
from .coordinator import SwitchbotDataUpdateCoordinator
from .coordinator import SwitchbotConfigEntry, SwitchbotDataUpdateCoordinator
from .entity import SwitchbotEntity
PARALLEL_UPDATES = 0
@ -81,10 +79,12 @@ SENSOR_TYPES: dict[str, SensorEntityDescription] = {
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 sensor based on a config entry."""
coordinator: SwitchbotDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
coordinator = entry.runtime_data
entities = [
SwitchBotSensor(coordinator, sensor)
for sensor in coordinator.device.parsed_data