Migrate baf to use config entry runtime_data (#117081)
This commit is contained in:
parent
20b29242f1
commit
159f0fcce7
9 changed files with 44 additions and 68 deletions
|
@ -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):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue