Migrate doorbird to use entry.runtime_data (#121413)

This commit is contained in:
J. Nick Koston 2024-07-07 07:57:10 -07:00 committed by GitHub
parent b6609fa77c
commit 792c6a9cd9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 40 additions and 42 deletions

View file

@ -6,13 +6,11 @@ from dataclasses import dataclass
from doorbirdpy import DoorBird
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import DOMAIN
from .entity import DoorBirdEntity
from .models import DoorBirdData
from .models import DoorBirdConfigEntry, DoorBirdData
IR_RELAY = "__ir_light__"
@ -38,12 +36,11 @@ IR_ENTITY_DESCRIPTION = DoorbirdButtonEntityDescription(
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
config_entry: DoorBirdConfigEntry,
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up the DoorBird button platform."""
config_entry_id = config_entry.entry_id
door_bird_data: DoorBirdData = hass.data[DOMAIN][config_entry_id]
door_bird_data = config_entry.runtime_data
relays = door_bird_data.door_station_info["RELAYS"]
entities = [