106 lines
3.1 KiB
Python
106 lines
3.1 KiB
Python
"""The BleBox devices integration."""
|
|
import logging
|
|
|
|
from blebox_uniapi.error import Error
|
|
from blebox_uniapi.products import Products
|
|
from blebox_uniapi.session import ApiHost
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import CONF_HOST, CONF_PORT, Platform
|
|
from homeassistant.core import HomeAssistant, callback
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
from homeassistant.helpers.entity import DeviceInfo, Entity
|
|
|
|
from .const import DEFAULT_SETUP_TIMEOUT, DOMAIN, PRODUCT
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
PLATFORMS = [
|
|
Platform.COVER,
|
|
Platform.SENSOR,
|
|
Platform.SWITCH,
|
|
Platform.AIR_QUALITY,
|
|
Platform.LIGHT,
|
|
Platform.CLIMATE,
|
|
]
|
|
|
|
PARALLEL_UPDATES = 0
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up BleBox devices from a config entry."""
|
|
|
|
websession = async_get_clientsession(hass)
|
|
|
|
host = entry.data[CONF_HOST]
|
|
port = entry.data[CONF_PORT]
|
|
timeout = DEFAULT_SETUP_TIMEOUT
|
|
|
|
api_host = ApiHost(host, port, timeout, websession, hass.loop)
|
|
|
|
try:
|
|
product = await Products.async_from_host(api_host)
|
|
except Error as ex:
|
|
_LOGGER.error("Identify failed at %s:%d (%s)", api_host.host, api_host.port, ex)
|
|
raise ConfigEntryNotReady from ex
|
|
|
|
domain = hass.data.setdefault(DOMAIN, {})
|
|
domain_entry = domain.setdefault(entry.entry_id, {})
|
|
product = domain_entry.setdefault(PRODUCT, product)
|
|
|
|
hass.config_entries.async_setup_platforms(entry, PLATFORMS)
|
|
|
|
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
|
|
|
|
|
|
@callback
|
|
def create_blebox_entities(
|
|
hass, config_entry, async_add_entities, entity_klass, entity_type
|
|
):
|
|
"""Create entities from a BleBox product's features."""
|
|
|
|
product = hass.data[DOMAIN][config_entry.entry_id][PRODUCT]
|
|
|
|
entities = []
|
|
if entity_type in product.features:
|
|
for feature in product.features[entity_type]:
|
|
entities.append(entity_klass(feature))
|
|
|
|
async_add_entities(entities, True)
|
|
|
|
|
|
class BleBoxEntity(Entity):
|
|
"""Implements a common class for entities representing a BleBox feature."""
|
|
|
|
def __init__(self, feature):
|
|
"""Initialize a BleBox entity."""
|
|
self._feature = feature
|
|
self._attr_name = feature.full_name
|
|
self._attr_unique_id = feature.unique_id
|
|
product = feature.product
|
|
self._attr_device_info = DeviceInfo(
|
|
identifiers={(DOMAIN, product.unique_id)},
|
|
manufacturer=product.brand,
|
|
model=product.model,
|
|
name=product.name,
|
|
sw_version=product.firmware_version,
|
|
)
|
|
|
|
async def async_update(self):
|
|
"""Update the entity state."""
|
|
try:
|
|
await self._feature.async_update()
|
|
except Error as ex:
|
|
_LOGGER.error("Updating '%s' failed: %s", self.name, ex)
|