mirror of
https://github.com/home-assistant/core.git
synced 2025-12-21 03:20:01 +00:00
fe34da19e2
Co-authored-by: RaHehl <rahehl@users.noreply.github.com> Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
85 lines
2.5 KiB
Python
85 lines
2.5 KiB
Python
"""The unifiprotect integration discovery."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import asdict, dataclass, field
|
|
from datetime import timedelta
|
|
import logging
|
|
from typing import Any
|
|
|
|
from unifi_discovery import AIOUnifiScanner, UnifiDevice, UnifiService
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.helpers import discovery_flow
|
|
from homeassistant.helpers.event import async_track_time_interval
|
|
from homeassistant.util.hass_dict import HassKey
|
|
|
|
from .const import DOMAIN
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class UniFiProtectRuntimeData:
|
|
"""Runtime data stored in hass.data[DOMAIN]."""
|
|
|
|
auth_retries: dict[str, int] = field(default_factory=dict)
|
|
discovery_started: bool = False
|
|
|
|
|
|
# Typed key for hass.data access at DOMAIN level
|
|
DATA_UNIFIPROTECT: HassKey[UniFiProtectRuntimeData] = HassKey(DOMAIN)
|
|
|
|
DISCOVERY_INTERVAL = timedelta(minutes=60)
|
|
|
|
|
|
@callback
|
|
def async_start_discovery(hass: HomeAssistant) -> None:
|
|
"""Start discovery."""
|
|
domain_data = hass.data.setdefault(DATA_UNIFIPROTECT, UniFiProtectRuntimeData())
|
|
if domain_data.discovery_started:
|
|
return
|
|
domain_data.discovery_started = True
|
|
|
|
async def _async_discovery() -> None:
|
|
async_trigger_discovery(hass, await async_discover_devices())
|
|
|
|
@callback
|
|
def _async_start_background_discovery(*_: Any) -> None:
|
|
"""Run discovery in the background."""
|
|
hass.async_create_background_task(_async_discovery(), "unifiprotect-discovery")
|
|
|
|
# Do not block startup since discovery takes 31s or more
|
|
_async_start_background_discovery()
|
|
async_track_time_interval(
|
|
hass,
|
|
_async_start_background_discovery,
|
|
DISCOVERY_INTERVAL,
|
|
cancel_on_shutdown=True,
|
|
)
|
|
|
|
|
|
async def async_discover_devices() -> list[UnifiDevice]:
|
|
"""Discover devices."""
|
|
scanner = AIOUnifiScanner()
|
|
devices = await scanner.async_scan()
|
|
_LOGGER.debug("Found devices: %s", devices)
|
|
return devices
|
|
|
|
|
|
@callback
|
|
def async_trigger_discovery(
|
|
hass: HomeAssistant,
|
|
discovered_devices: list[UnifiDevice],
|
|
) -> None:
|
|
"""Trigger config flows for discovered devices."""
|
|
for device in discovered_devices:
|
|
if device.services[UnifiService.Protect] and device.hw_addr:
|
|
discovery_flow.async_create_flow(
|
|
hass,
|
|
DOMAIN,
|
|
context={"source": config_entries.SOURCE_INTEGRATION_DISCOVERY},
|
|
data=asdict(device),
|
|
)
|