mirror of
https://github.com/home-assistant/core.git
synced 2025-05-25 08:17:09 +00:00
b4bac0f7a0
* Remove unnecessary exception re-wraps * Preserve exception chains on re-raise We slap "from cause" to almost all possible cases here. In some cases it could conceivably be better to do "from None" if we really want to hide the cause. However those should be in the minority, and "from cause" should be an improvement over the corresponding raise without a "from" in all cases anyway. The only case where we raise from None here is in plex, where the exception for an original invalid SSL cert is not the root cause for failure to validate a newly fetched one. Follow local convention on exception variable names if there is a consistent one, otherwise `err` to match with majority of codebase. * Fix mistaken re-wrap in homematicip_cloud/hap.py Missed the difference between HmipConnectionError and HmipcConnectionError. * Do not hide original error on plex new cert validation error Original is not the cause for the new one, but showing old in the traceback is useful nevertheless.
142 lines
4.0 KiB
Python
142 lines
4.0 KiB
Python
"""Integration with the Rachio Iro sprinkler system controller."""
|
|
import asyncio
|
|
import logging
|
|
import secrets
|
|
|
|
from rachiopy import Rachio
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
|
|
from homeassistant.const import CONF_API_KEY
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.helpers import config_validation as cv
|
|
|
|
from .const import (
|
|
CONF_CLOUDHOOK_URL,
|
|
CONF_MANUAL_RUN_MINS,
|
|
CONF_WEBHOOK_ID,
|
|
DEFAULT_MANUAL_RUN_MINS,
|
|
DOMAIN,
|
|
RACHIO_API_EXCEPTIONS,
|
|
)
|
|
from .device import RachioPerson
|
|
from .webhooks import (
|
|
async_get_or_create_registered_webhook_id_and_url,
|
|
async_register_webhook,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SUPPORTED_DOMAINS = ["switch", "binary_sensor"]
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
{
|
|
vol.Required(CONF_API_KEY): cv.string,
|
|
vol.Optional(
|
|
CONF_MANUAL_RUN_MINS, default=DEFAULT_MANUAL_RUN_MINS
|
|
): cv.positive_int,
|
|
}
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config: dict):
|
|
"""Set up the rachio component from YAML."""
|
|
|
|
conf = config.get(DOMAIN)
|
|
hass.data.setdefault(DOMAIN, {})
|
|
|
|
if not conf:
|
|
return True
|
|
|
|
hass.async_create_task(
|
|
hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": SOURCE_IMPORT}, data=conf
|
|
)
|
|
)
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
|
|
"""Unload a config entry."""
|
|
unload_ok = all(
|
|
await asyncio.gather(
|
|
*[
|
|
hass.config_entries.async_forward_entry_unload(entry, component)
|
|
for component in SUPPORTED_DOMAINS
|
|
]
|
|
)
|
|
)
|
|
|
|
if unload_ok:
|
|
hass.data[DOMAIN].pop(entry.entry_id)
|
|
|
|
return unload_ok
|
|
|
|
|
|
async def async_remove_entry(hass, entry):
|
|
"""Remove a rachio config entry."""
|
|
if CONF_CLOUDHOOK_URL in entry.data:
|
|
await hass.components.cloud.async_delete_cloudhook(entry.data[CONF_WEBHOOK_ID])
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
|
|
"""Set up the Rachio config entry."""
|
|
|
|
config = entry.data
|
|
options = entry.options
|
|
|
|
# CONF_MANUAL_RUN_MINS can only come from a yaml import
|
|
if not options.get(CONF_MANUAL_RUN_MINS) and config.get(CONF_MANUAL_RUN_MINS):
|
|
options_copy = options.copy()
|
|
options_copy[CONF_MANUAL_RUN_MINS] = config[CONF_MANUAL_RUN_MINS]
|
|
hass.config_entries.async_update_entry(entry, options=options_copy)
|
|
|
|
# Configure API
|
|
api_key = config[CONF_API_KEY]
|
|
rachio = Rachio(api_key)
|
|
|
|
# Get the URL of this server
|
|
rachio.webhook_auth = secrets.token_hex()
|
|
webhook_id, webhook_url = await async_get_or_create_registered_webhook_id_and_url(
|
|
hass, entry
|
|
)
|
|
rachio.webhook_url = webhook_url
|
|
|
|
person = RachioPerson(rachio, entry)
|
|
|
|
# Get the API user
|
|
try:
|
|
await hass.async_add_executor_job(person.setup, hass)
|
|
# Yes we really do get all these exceptions (hopefully rachiopy switches to requests)
|
|
# and there is not a reasonable timeout here so it can block for a long time
|
|
except RACHIO_API_EXCEPTIONS as error:
|
|
_LOGGER.error("Could not reach the Rachio API: %s", error)
|
|
raise ConfigEntryNotReady from error
|
|
|
|
# Check for Rachio controller devices
|
|
if not person.controllers:
|
|
_LOGGER.error("No Rachio devices found in account %s", person.username)
|
|
return False
|
|
_LOGGER.info(
|
|
"%d Rachio device(s) found; The url %s must be accessible from the internet in order to receive updates",
|
|
len(person.controllers),
|
|
webhook_url,
|
|
)
|
|
|
|
# Enable component
|
|
hass.data[DOMAIN][entry.entry_id] = person
|
|
async_register_webhook(hass, webhook_id, entry.entry_id)
|
|
|
|
for component in SUPPORTED_DOMAINS:
|
|
hass.async_create_task(
|
|
hass.config_entries.async_forward_entry_setup(entry, component)
|
|
)
|
|
|
|
return True
|