62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
"""The Big Ass Fans integration."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from asyncio import timeout
|
|
|
|
from aiobafi6 import Device, Service
|
|
from aiobafi6.discovery import PORT
|
|
from aiobafi6.exceptions import DeviceUUIDMismatchError
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_IP_ADDRESS, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
|
|
from .const import DOMAIN, QUERY_INTERVAL, RUN_TIMEOUT
|
|
from .models import BAFData
|
|
|
|
PLATFORMS: list[Platform] = [
|
|
Platform.BINARY_SENSOR,
|
|
Platform.CLIMATE,
|
|
Platform.FAN,
|
|
Platform.LIGHT,
|
|
Platform.NUMBER,
|
|
Platform.SENSOR,
|
|
Platform.SWITCH,
|
|
]
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up Big Ass Fans from a config entry."""
|
|
ip_address = entry.data[CONF_IP_ADDRESS]
|
|
|
|
service = Service(ip_addresses=[ip_address], uuid=entry.unique_id, port=PORT)
|
|
device = Device(service, query_interval_seconds=QUERY_INTERVAL)
|
|
run_future = device.async_run()
|
|
|
|
try:
|
|
async with timeout(RUN_TIMEOUT):
|
|
await device.async_wait_available()
|
|
except DeviceUUIDMismatchError as ex:
|
|
raise ConfigEntryNotReady(
|
|
f"Unexpected device found at {ip_address}; expected {entry.unique_id}, found {device.dns_sd_uuid}"
|
|
) from ex
|
|
except asyncio.TimeoutError as ex:
|
|
run_future.cancel()
|
|
raise ConfigEntryNotReady(f"Timed out connecting to {ip_address}") from ex
|
|
|
|
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = BAFData(device, run_future)
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
|
|
data: BAFData = hass.data[DOMAIN].pop(entry.entry_id)
|
|
data.run_future.cancel()
|
|
|
|
return unload_ok
|