147 lines
4.6 KiB
Python
147 lines
4.6 KiB
Python
"""UniFi Protect Platform."""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
from datetime import timedelta
|
|
import functools
|
|
import logging
|
|
|
|
from aiohttp import CookieJar
|
|
from aiohttp.client_exceptions import ServerDisconnectedError
|
|
from pyunifiprotect import NotAuthorized, NvrError, ProtectApiClient
|
|
|
|
from homeassistant.config_entries import ConfigEntry, ConfigEntryState
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_PASSWORD,
|
|
CONF_PORT,
|
|
CONF_USERNAME,
|
|
CONF_VERIFY_SSL,
|
|
EVENT_HOMEASSISTANT_STOP,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
|
|
from homeassistant.helpers.aiohttp_client import async_create_clientsession
|
|
|
|
from .const import (
|
|
ALL_GLOBAL_SERIVCES,
|
|
CONF_ALL_UPDATES,
|
|
CONF_OVERRIDE_CHOST,
|
|
DEFAULT_SCAN_INTERVAL,
|
|
DEVICES_FOR_SUBSCRIBE,
|
|
DOMAIN,
|
|
DOORBELL_TEXT_SCHEMA,
|
|
MIN_REQUIRED_PROTECT_V,
|
|
OUTDATED_LOG_MESSAGE,
|
|
PLATFORMS,
|
|
SERVICE_ADD_DOORBELL_TEXT,
|
|
SERVICE_REMOVE_DOORBELL_TEXT,
|
|
SERVICE_SET_DEFAULT_DOORBELL_TEXT,
|
|
)
|
|
from .data import ProtectData
|
|
from .services import add_doorbell_text, remove_doorbell_text, set_default_doorbell_text
|
|
from .views import ThumbnailProxyView
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SCAN_INTERVAL = timedelta(seconds=DEFAULT_SCAN_INTERVAL)
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up the UniFi Protect config entries."""
|
|
|
|
session = async_create_clientsession(hass, cookie_jar=CookieJar(unsafe=True))
|
|
protect = ProtectApiClient(
|
|
host=entry.data[CONF_HOST],
|
|
port=entry.data[CONF_PORT],
|
|
username=entry.data[CONF_USERNAME],
|
|
password=entry.data[CONF_PASSWORD],
|
|
verify_ssl=entry.data[CONF_VERIFY_SSL],
|
|
session=session,
|
|
subscribed_models=DEVICES_FOR_SUBSCRIBE,
|
|
override_connection_host=entry.options.get(CONF_OVERRIDE_CHOST, False),
|
|
ignore_stats=not entry.options.get(CONF_ALL_UPDATES, False),
|
|
)
|
|
_LOGGER.debug("Connect to UniFi Protect")
|
|
data_service = ProtectData(hass, protect, SCAN_INTERVAL, entry)
|
|
|
|
try:
|
|
nvr_info = await protect.get_nvr()
|
|
except NotAuthorized as err:
|
|
raise ConfigEntryAuthFailed(err) from err
|
|
except (asyncio.TimeoutError, NvrError, ServerDisconnectedError) as err:
|
|
raise ConfigEntryNotReady from err
|
|
|
|
if nvr_info.version < MIN_REQUIRED_PROTECT_V:
|
|
_LOGGER.error(
|
|
OUTDATED_LOG_MESSAGE,
|
|
nvr_info.version,
|
|
MIN_REQUIRED_PROTECT_V,
|
|
)
|
|
return False
|
|
|
|
if entry.unique_id is None:
|
|
hass.config_entries.async_update_entry(entry, unique_id=nvr_info.mac)
|
|
|
|
await data_service.async_setup()
|
|
if not data_service.last_update_success:
|
|
raise ConfigEntryNotReady
|
|
|
|
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = data_service
|
|
hass.config_entries.async_setup_platforms(entry, PLATFORMS)
|
|
|
|
services = [
|
|
(
|
|
SERVICE_ADD_DOORBELL_TEXT,
|
|
functools.partial(add_doorbell_text, hass),
|
|
DOORBELL_TEXT_SCHEMA,
|
|
),
|
|
(
|
|
SERVICE_REMOVE_DOORBELL_TEXT,
|
|
functools.partial(remove_doorbell_text, hass),
|
|
DOORBELL_TEXT_SCHEMA,
|
|
),
|
|
(
|
|
SERVICE_SET_DEFAULT_DOORBELL_TEXT,
|
|
functools.partial(set_default_doorbell_text, hass),
|
|
DOORBELL_TEXT_SCHEMA,
|
|
),
|
|
]
|
|
for name, method, schema in services:
|
|
if hass.services.has_service(DOMAIN, name):
|
|
continue
|
|
hass.services.async_register(DOMAIN, name, method, schema=schema)
|
|
|
|
hass.http.register_view(ThumbnailProxyView(hass))
|
|
|
|
entry.async_on_unload(entry.add_update_listener(_async_options_updated))
|
|
entry.async_on_unload(
|
|
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, data_service.async_stop)
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
async def _async_options_updated(hass: HomeAssistant, entry: ConfigEntry) -> None:
|
|
"""Update options."""
|
|
await hass.config_entries.async_reload(entry.entry_id)
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload UniFi Protect config entry."""
|
|
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
|
|
data: ProtectData = hass.data[DOMAIN][entry.entry_id]
|
|
await data.async_stop()
|
|
hass.data[DOMAIN].pop(entry.entry_id)
|
|
|
|
loaded_entries = [
|
|
entry
|
|
for entry in hass.config_entries.async_entries(DOMAIN)
|
|
if entry.state == ConfigEntryState.LOADED
|
|
]
|
|
if len(loaded_entries) == 1:
|
|
for name in ALL_GLOBAL_SERIVCES:
|
|
hass.services.async_remove(DOMAIN, name)
|
|
|
|
return bool(unload_ok)
|