__init__.py

Go to the documentation of this file.

"""The gogogate2 component."""
 
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_DEVICE, Platform
from homeassistant.core import HomeAssistant
 
from .common import get_data_update_coordinator
from .const import DEVICE_TYPE_GOGOGATE2
 
PLATFORMS = [Platform.COVER, Platform.SENSOR]
 
 
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
    """Do setup of Gogogate2."""
 
    # Update the config entry.
    config_updates = {}
    if CONF_DEVICE not in entry.data:
        config_updates = {
            **entry.data,
            CONF_DEVICE: DEVICE_TYPE_GOGOGATE2,
        }
 
    if config_updates:
        hass.config_entries.async_update_entry(entry, data=config_updates)
 
    data_update_coordinator = get_data_update_coordinator(hass, entry)
    await data_update_coordinator.async_config_entry_first_refresh()
 
    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
 
    return True
 
 
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
    """Unload Gogogate2 config entry."""
    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)