4ce6d00a22
* Add support for async device discovery * FIx missing dispatcher cleanup breaking integration reload * Update homeassistant/components/gree/climate.py Co-authored-by: Erik Montnemery <erik@montnemery.com> * Update homeassistant/components/gree/switch.py Co-authored-by: Erik Montnemery <erik@montnemery.com> * Update homeassistant/components/gree/bridge.py Co-authored-by: Erik Montnemery <erik@montnemery.com> * Working on feedback * Improving load/unload tests * Update homeassistant/components/gree/__init__.py Co-authored-by: Erik Montnemery <erik@montnemery.com> * Working on more feedback * Add tests covering async discovery scenarios * Remove unnecessary shutdown * Update homeassistant/components/gree/__init__.py Co-authored-by: Erik Montnemery <erik@montnemery.com> * Code refactor from reviews Co-authored-by: Erik Montnemery <erik@montnemery.com>
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
"""The Gree Climate integration."""
|
|
import asyncio
|
|
from datetime import timedelta
|
|
import logging
|
|
|
|
from homeassistant.components.climate import DOMAIN as CLIMATE_DOMAIN
|
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.event import async_track_time_interval
|
|
|
|
from .bridge import DiscoveryService
|
|
from .const import (
|
|
COORDINATORS,
|
|
DATA_DISCOVERY_INTERVAL,
|
|
DATA_DISCOVERY_SERVICE,
|
|
DISCOVERY_SCAN_INTERVAL,
|
|
DISPATCHERS,
|
|
DOMAIN,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config: dict):
|
|
"""Set up the Gree Climate component."""
|
|
hass.data[DOMAIN] = {}
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
|
|
"""Set up Gree Climate from a config entry."""
|
|
gree_discovery = DiscoveryService(hass)
|
|
hass.data[DATA_DISCOVERY_SERVICE] = gree_discovery
|
|
|
|
hass.data[DOMAIN].setdefault(DISPATCHERS, [])
|
|
|
|
hass.async_create_task(
|
|
hass.config_entries.async_forward_entry_setup(entry, CLIMATE_DOMAIN)
|
|
)
|
|
hass.async_create_task(
|
|
hass.config_entries.async_forward_entry_setup(entry, SWITCH_DOMAIN)
|
|
)
|
|
|
|
async def _async_scan_update(_=None):
|
|
await gree_discovery.discovery.scan()
|
|
|
|
_LOGGER.debug("Scanning network for Gree devices")
|
|
await _async_scan_update()
|
|
|
|
hass.data[DOMAIN][DATA_DISCOVERY_INTERVAL] = async_track_time_interval(
|
|
hass, _async_scan_update, timedelta(seconds=DISCOVERY_SCAN_INTERVAL)
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
|
|
"""Unload a config entry."""
|
|
if hass.data[DOMAIN].get(DISPATCHERS) is not None:
|
|
for cleanup in hass.data[DOMAIN][DISPATCHERS]:
|
|
cleanup()
|
|
|
|
if hass.data[DOMAIN].get(DATA_DISCOVERY_INTERVAL) is not None:
|
|
hass.data[DOMAIN].pop(DATA_DISCOVERY_INTERVAL)()
|
|
|
|
if hass.data.get(DATA_DISCOVERY_SERVICE) is not None:
|
|
hass.data.pop(DATA_DISCOVERY_SERVICE)
|
|
|
|
results = asyncio.gather(
|
|
hass.config_entries.async_forward_entry_unload(entry, CLIMATE_DOMAIN),
|
|
hass.config_entries.async_forward_entry_unload(entry, SWITCH_DOMAIN),
|
|
)
|
|
|
|
unload_ok = all(await results)
|
|
if unload_ok:
|
|
hass.data[DOMAIN].pop(COORDINATORS, None)
|
|
hass.data[DOMAIN].pop(DISPATCHERS, None)
|
|
|
|
return unload_ok
|