ed0568c655
* Report non-awaited/non-locked config entry platform forwards Its currently possible for config entries to be reloaded while their platforms are being forwarded if platform forwards are not awaited or done after the config entry is setup since the lock will not be held in this case. In https://developers.home-assistant.io/blog/2022/07/08/config_entry_forwards we advised to await platform forwards to ensure this does not happen, however for sleeping devices and late discovered devices, platform forwards may happen later. If config platform forwards are happening during setup, they should be awaited If config entry platform forwards are not happening during setup, instead async_late_forward_entry_setups should be used which will hold the lock to prevent the config entry from being unloaded while its platforms are being setup * Report non-awaited/non-locked config entry platform forwards Its currently possible for config entries to be reloaded while their platforms are being forwarded if platform forwards are not awaited or done after the config entry is setup since the lock will not be held in this case. In https://developers.home-assistant.io/blog/2022/07/08/config_entry_forwards we advised to await platform forwards to ensure this does not happen, however for sleeping devices and late discovered devices, platform forwards may happen later. If config platform forwards are happening during setup, they should be awaited If config entry platform forwards are not happening during setup, instead async_late_forward_entry_setups should be used which will hold the lock to prevent the config entry from being unloaded while its platforms are being setup * run with error on to find them * cert_exp, hold lock * cert_exp, hold lock * shelly async_late_forward_entry_setups * compact * compact * found another * patch up mobileapp * patch up hue tests * patch up smartthings * fix mqtt * fix esphome * zwave_js * mqtt * rework * fixes * fix mocking * fix mocking * do not call async_forward_entry_setup directly * docstrings * docstrings * docstrings * add comments * doc strings * fixed all in core, turn off strict * coverage * coverage * missing * coverage
145 lines
4.8 KiB
Python
145 lines
4.8 KiB
Python
"""VeSync integration."""
|
|
|
|
import logging
|
|
|
|
from pyvesync import VeSync
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME, Platform
|
|
from homeassistant.core import HomeAssistant, ServiceCall
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_send
|
|
|
|
from .common import async_process_devices
|
|
from .const import (
|
|
DOMAIN,
|
|
SERVICE_UPDATE_DEVS,
|
|
VS_DISCOVERY,
|
|
VS_FANS,
|
|
VS_LIGHTS,
|
|
VS_MANAGER,
|
|
VS_SENSORS,
|
|
VS_SWITCHES,
|
|
)
|
|
|
|
PLATFORMS = [Platform.FAN, Platform.LIGHT, Platform.SENSOR, Platform.SWITCH]
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONFIG_SCHEMA = cv.removed(DOMAIN, raise_if_present=False)
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
|
|
"""Set up Vesync as config entry."""
|
|
username = config_entry.data[CONF_USERNAME]
|
|
password = config_entry.data[CONF_PASSWORD]
|
|
|
|
time_zone = str(hass.config.time_zone)
|
|
|
|
manager = VeSync(username, password, time_zone)
|
|
|
|
login = await hass.async_add_executor_job(manager.login)
|
|
|
|
if not login:
|
|
_LOGGER.error("Unable to login to the VeSync server")
|
|
return False
|
|
|
|
device_dict = await async_process_devices(hass, manager)
|
|
|
|
forward_setups = hass.config_entries.async_forward_entry_setups
|
|
|
|
hass.data[DOMAIN] = {}
|
|
hass.data[DOMAIN][VS_MANAGER] = manager
|
|
|
|
switches = hass.data[DOMAIN][VS_SWITCHES] = []
|
|
fans = hass.data[DOMAIN][VS_FANS] = []
|
|
lights = hass.data[DOMAIN][VS_LIGHTS] = []
|
|
sensors = hass.data[DOMAIN][VS_SENSORS] = []
|
|
platforms = []
|
|
|
|
if device_dict[VS_SWITCHES]:
|
|
switches.extend(device_dict[VS_SWITCHES])
|
|
platforms.append(Platform.SWITCH)
|
|
|
|
if device_dict[VS_FANS]:
|
|
fans.extend(device_dict[VS_FANS])
|
|
platforms.append(Platform.FAN)
|
|
|
|
if device_dict[VS_LIGHTS]:
|
|
lights.extend(device_dict[VS_LIGHTS])
|
|
platforms.append(Platform.LIGHT)
|
|
|
|
if device_dict[VS_SENSORS]:
|
|
sensors.extend(device_dict[VS_SENSORS])
|
|
platforms.append(Platform.SENSOR)
|
|
|
|
await hass.config_entries.async_forward_entry_setups(config_entry, platforms)
|
|
|
|
async def async_new_device_discovery(service: ServiceCall) -> None:
|
|
"""Discover if new devices should be added."""
|
|
manager = hass.data[DOMAIN][VS_MANAGER]
|
|
switches = hass.data[DOMAIN][VS_SWITCHES]
|
|
fans = hass.data[DOMAIN][VS_FANS]
|
|
lights = hass.data[DOMAIN][VS_LIGHTS]
|
|
sensors = hass.data[DOMAIN][VS_SENSORS]
|
|
|
|
dev_dict = await async_process_devices(hass, manager)
|
|
switch_devs = dev_dict.get(VS_SWITCHES, [])
|
|
fan_devs = dev_dict.get(VS_FANS, [])
|
|
light_devs = dev_dict.get(VS_LIGHTS, [])
|
|
sensor_devs = dev_dict.get(VS_SENSORS, [])
|
|
|
|
switch_set = set(switch_devs)
|
|
new_switches = list(switch_set.difference(switches))
|
|
if new_switches and switches:
|
|
switches.extend(new_switches)
|
|
async_dispatcher_send(hass, VS_DISCOVERY.format(VS_SWITCHES), new_switches)
|
|
return
|
|
if new_switches and not switches:
|
|
switches.extend(new_switches)
|
|
hass.async_create_task(forward_setups(config_entry, [Platform.SWITCH]))
|
|
|
|
fan_set = set(fan_devs)
|
|
new_fans = list(fan_set.difference(fans))
|
|
if new_fans and fans:
|
|
fans.extend(new_fans)
|
|
async_dispatcher_send(hass, VS_DISCOVERY.format(VS_FANS), new_fans)
|
|
return
|
|
if new_fans and not fans:
|
|
fans.extend(new_fans)
|
|
hass.async_create_task(forward_setups(config_entry, [Platform.FAN]))
|
|
|
|
light_set = set(light_devs)
|
|
new_lights = list(light_set.difference(lights))
|
|
if new_lights and lights:
|
|
lights.extend(new_lights)
|
|
async_dispatcher_send(hass, VS_DISCOVERY.format(VS_LIGHTS), new_lights)
|
|
return
|
|
if new_lights and not lights:
|
|
lights.extend(new_lights)
|
|
hass.async_create_task(forward_setups(config_entry, [Platform.LIGHT]))
|
|
|
|
sensor_set = set(sensor_devs)
|
|
new_sensors = list(sensor_set.difference(sensors))
|
|
if new_sensors and sensors:
|
|
sensors.extend(new_sensors)
|
|
async_dispatcher_send(hass, VS_DISCOVERY.format(VS_SENSORS), new_sensors)
|
|
return
|
|
if new_sensors and not sensors:
|
|
sensors.extend(new_sensors)
|
|
hass.async_create_task(forward_setups(config_entry, [Platform.SENSOR]))
|
|
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_UPDATE_DEVS, async_new_device_discovery
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
|
if unload_ok:
|
|
hass.data[DOMAIN].pop(entry.entry_id)
|
|
|
|
return unload_ok
|