Migrate baf to use config entry runtime_data (#117081)

This commit is contained in:
J. Nick Koston 2024-05-08 15:37:14 -05:00 committed by GitHub
parent 20b29242f1
commit 159f0fcce7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 44 additions and 68 deletions

View file

@ -7,7 +7,6 @@ from typing import Any
from aiobafi6 import OffOnAuto
from homeassistant import config_entries
from homeassistant.components.fan import (
DIRECTION_FORWARD,
DIRECTION_REVERSE,
@ -21,20 +20,20 @@ from homeassistant.util.percentage import (
ranged_value_to_percentage,
)
from .const import DOMAIN, PRESET_MODE_AUTO, SPEED_COUNT, SPEED_RANGE
from . import BAFConfigEntry
from .const import PRESET_MODE_AUTO, SPEED_COUNT, SPEED_RANGE
from .entity import BAFEntity
from .models import BAFData
async def async_setup_entry(
hass: HomeAssistant,
entry: config_entries.ConfigEntry,
entry: BAFConfigEntry,
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up SenseME fans."""
data: BAFData = hass.data[DOMAIN][entry.entry_id]
if data.device.has_fan:
async_add_entities([BAFFan(data.device)])
device = entry.runtime_data
if device.has_fan:
async_add_entities([BAFFan(device)])
class BAFFan(BAFEntity, FanEntity):