mirror of
https://github.com/home-assistant/core
synced 2024-08-02 23:40:32 +02:00
287 lines
9.0 KiB
Python
287 lines
9.0 KiB
Python
"""The Netatmo integration."""
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from http import HTTPStatus
|
|
import logging
|
|
import secrets
|
|
|
|
import aiohttp
|
|
import pyatmo
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components import cloud
|
|
from homeassistant.components.webhook import (
|
|
async_generate_url as webhook_generate_url,
|
|
async_register as webhook_register,
|
|
async_unregister as webhook_unregister,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
CONF_CLIENT_ID,
|
|
CONF_CLIENT_SECRET,
|
|
CONF_WEBHOOK_ID,
|
|
EVENT_HOMEASSISTANT_STARTED,
|
|
EVENT_HOMEASSISTANT_STOP,
|
|
)
|
|
from homeassistant.core import CoreState, Event, HomeAssistant, ServiceCall
|
|
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
|
|
from homeassistant.helpers import (
|
|
aiohttp_client,
|
|
config_entry_oauth2_flow,
|
|
config_validation as cv,
|
|
)
|
|
from homeassistant.helpers.dispatcher import (
|
|
async_dispatcher_connect,
|
|
async_dispatcher_send,
|
|
)
|
|
from homeassistant.helpers.event import async_call_later
|
|
from homeassistant.helpers.typing import ConfigType
|
|
|
|
from . import api, config_flow
|
|
from .const import (
|
|
AUTH,
|
|
CONF_CLOUDHOOK_URL,
|
|
DATA_CAMERAS,
|
|
DATA_DEVICE_IDS,
|
|
DATA_EVENTS,
|
|
DATA_HANDLER,
|
|
DATA_HOMES,
|
|
DATA_PERSONS,
|
|
DATA_SCHEDULES,
|
|
DOMAIN,
|
|
NETATMO_SCOPES,
|
|
OAUTH2_AUTHORIZE,
|
|
OAUTH2_TOKEN,
|
|
PLATFORMS,
|
|
WEBHOOK_ACTIVATION,
|
|
WEBHOOK_DEACTIVATION,
|
|
WEBHOOK_PUSH_TYPE,
|
|
)
|
|
from .data_handler import NetatmoDataHandler
|
|
from .webhook import async_handle_webhook
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
{
|
|
vol.Required(CONF_CLIENT_ID): cv.string,
|
|
vol.Required(CONF_CLIENT_SECRET): cv.string,
|
|
}
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
MAX_WEBHOOK_RETRIES = 3
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
|
|
"""Set up the Netatmo component."""
|
|
hass.data[DOMAIN] = {
|
|
DATA_PERSONS: {},
|
|
DATA_DEVICE_IDS: {},
|
|
DATA_SCHEDULES: {},
|
|
DATA_HOMES: {},
|
|
DATA_EVENTS: {},
|
|
DATA_CAMERAS: {},
|
|
}
|
|
|
|
if DOMAIN not in config:
|
|
return True
|
|
|
|
config_flow.NetatmoFlowHandler.async_register_implementation(
|
|
hass,
|
|
config_entry_oauth2_flow.LocalOAuth2Implementation(
|
|
hass,
|
|
DOMAIN,
|
|
config[DOMAIN][CONF_CLIENT_ID],
|
|
config[DOMAIN][CONF_CLIENT_SECRET],
|
|
OAUTH2_AUTHORIZE,
|
|
OAUTH2_TOKEN,
|
|
),
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up Netatmo from a config entry."""
|
|
implementation = (
|
|
await config_entry_oauth2_flow.async_get_config_entry_implementation(
|
|
hass, entry
|
|
)
|
|
)
|
|
|
|
# Set unique id if non was set (migration)
|
|
if not entry.unique_id:
|
|
hass.config_entries.async_update_entry(entry, unique_id=DOMAIN)
|
|
|
|
session = config_entry_oauth2_flow.OAuth2Session(hass, entry, implementation)
|
|
try:
|
|
await session.async_ensure_token_valid()
|
|
except aiohttp.ClientResponseError as ex:
|
|
_LOGGER.debug("API error: %s (%s)", ex.code, ex.message)
|
|
if ex.code in (
|
|
HTTPStatus.BAD_REQUEST,
|
|
HTTPStatus.UNAUTHORIZED,
|
|
HTTPStatus.FORBIDDEN,
|
|
):
|
|
raise ConfigEntryAuthFailed("Token not valid, trigger renewal") from ex
|
|
raise ConfigEntryNotReady from ex
|
|
|
|
if sorted(session.token["scope"]) != sorted(NETATMO_SCOPES):
|
|
_LOGGER.debug(
|
|
"Scope is invalid: %s != %s", session.token["scope"], NETATMO_SCOPES
|
|
)
|
|
raise ConfigEntryAuthFailed("Token scope not valid, trigger renewal")
|
|
|
|
hass.data[DOMAIN][entry.entry_id] = {
|
|
AUTH: api.AsyncConfigEntryNetatmoAuth(
|
|
aiohttp_client.async_get_clientsession(hass), session
|
|
)
|
|
}
|
|
|
|
data_handler = NetatmoDataHandler(hass, entry)
|
|
await data_handler.async_setup()
|
|
hass.data[DOMAIN][entry.entry_id][DATA_HANDLER] = data_handler
|
|
|
|
hass.config_entries.async_setup_platforms(entry, PLATFORMS)
|
|
|
|
_webhook_retries = 0
|
|
|
|
async def unregister_webhook(
|
|
call_or_event_or_dt: ServiceCall | Event | datetime | None,
|
|
) -> None:
|
|
if CONF_WEBHOOK_ID not in entry.data:
|
|
return
|
|
_LOGGER.debug("Unregister Netatmo webhook (%s)", entry.data[CONF_WEBHOOK_ID])
|
|
async_dispatcher_send(
|
|
hass,
|
|
f"signal-{DOMAIN}-webhook-None",
|
|
{"type": "None", "data": {WEBHOOK_PUSH_TYPE: WEBHOOK_DEACTIVATION}},
|
|
)
|
|
webhook_unregister(hass, entry.data[CONF_WEBHOOK_ID])
|
|
try:
|
|
await hass.data[DOMAIN][entry.entry_id][AUTH].async_dropwebhook()
|
|
except pyatmo.ApiError:
|
|
_LOGGER.debug(
|
|
"No webhook to be dropped for %s", entry.data[CONF_WEBHOOK_ID]
|
|
)
|
|
|
|
nonlocal _webhook_retries
|
|
if _webhook_retries < MAX_WEBHOOK_RETRIES:
|
|
_webhook_retries += 1
|
|
async_call_later(hass, 30, register_webhook)
|
|
|
|
async def register_webhook(
|
|
call_or_event_or_dt: ServiceCall | Event | datetime | None,
|
|
) -> None:
|
|
if CONF_WEBHOOK_ID not in entry.data:
|
|
data = {**entry.data, CONF_WEBHOOK_ID: secrets.token_hex()}
|
|
hass.config_entries.async_update_entry(entry, data=data)
|
|
|
|
if cloud.async_active_subscription(hass):
|
|
if CONF_CLOUDHOOK_URL not in entry.data:
|
|
webhook_url = await cloud.async_create_cloudhook(
|
|
hass, entry.data[CONF_WEBHOOK_ID]
|
|
)
|
|
data = {**entry.data, CONF_CLOUDHOOK_URL: webhook_url}
|
|
hass.config_entries.async_update_entry(entry, data=data)
|
|
else:
|
|
webhook_url = entry.data[CONF_CLOUDHOOK_URL]
|
|
else:
|
|
webhook_url = webhook_generate_url(hass, entry.data[CONF_WEBHOOK_ID])
|
|
|
|
if entry.data[
|
|
"auth_implementation"
|
|
] == cloud.DOMAIN and not webhook_url.startswith("https://"):
|
|
_LOGGER.warning(
|
|
"Webhook not registered - "
|
|
"https and port 443 is required to register the webhook"
|
|
)
|
|
return
|
|
|
|
try:
|
|
webhook_register(
|
|
hass,
|
|
DOMAIN,
|
|
"Netatmo",
|
|
entry.data[CONF_WEBHOOK_ID],
|
|
async_handle_webhook,
|
|
)
|
|
|
|
async def handle_event(event: dict) -> None:
|
|
"""Handle webhook events."""
|
|
if event["data"][WEBHOOK_PUSH_TYPE] == WEBHOOK_ACTIVATION:
|
|
if activation_listener is not None:
|
|
activation_listener()
|
|
|
|
if activation_timeout is not None:
|
|
activation_timeout()
|
|
|
|
activation_listener = async_dispatcher_connect(
|
|
hass,
|
|
f"signal-{DOMAIN}-webhook-None",
|
|
handle_event,
|
|
)
|
|
|
|
activation_timeout = async_call_later(hass, 30, unregister_webhook)
|
|
|
|
await hass.data[DOMAIN][entry.entry_id][AUTH].async_addwebhook(webhook_url)
|
|
_LOGGER.info("Register Netatmo webhook: %s", webhook_url)
|
|
except pyatmo.ApiError as err:
|
|
_LOGGER.error("Error during webhook registration - %s", err)
|
|
|
|
entry.async_on_unload(
|
|
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP, unregister_webhook)
|
|
)
|
|
|
|
if hass.state == CoreState.running:
|
|
await register_webhook(None)
|
|
else:
|
|
hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STARTED, register_webhook)
|
|
|
|
hass.services.async_register(DOMAIN, "register_webhook", register_webhook)
|
|
hass.services.async_register(DOMAIN, "unregister_webhook", unregister_webhook)
|
|
|
|
entry.add_update_listener(async_config_entry_updated)
|
|
|
|
return True
|
|
|
|
|
|
async def async_config_entry_updated(hass: HomeAssistant, entry: ConfigEntry) -> None:
|
|
"""Handle signals of config entry being updated."""
|
|
async_dispatcher_send(hass, f"signal-{DOMAIN}-public-update-{entry.entry_id}")
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
data = hass.data[DOMAIN]
|
|
|
|
if CONF_WEBHOOK_ID in entry.data:
|
|
webhook_unregister(hass, entry.data[CONF_WEBHOOK_ID])
|
|
await data[entry.entry_id][AUTH].async_dropwebhook()
|
|
_LOGGER.info("Unregister Netatmo webhook")
|
|
|
|
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
|
|
|
if unload_ok and entry.entry_id in data:
|
|
data.pop(entry.entry_id)
|
|
|
|
return unload_ok
|
|
|
|
|
|
async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
|
|
"""Cleanup when entry is removed."""
|
|
if CONF_WEBHOOK_ID in entry.data and cloud.async_active_subscription(hass):
|
|
try:
|
|
_LOGGER.debug(
|
|
"Removing Netatmo cloudhook (%s)", entry.data[CONF_WEBHOOK_ID]
|
|
)
|
|
await cloud.async_delete_cloudhook(hass, entry.data[CONF_WEBHOOK_ID])
|
|
except cloud.CloudNotAvailable:
|
|
pass
|