537 lines
20 KiB
Python
537 lines
20 KiB
Python
"""Support for the Fibaro devices."""
|
|
from __future__ import annotations
|
|
|
|
from collections import defaultdict
|
|
from collections.abc import Mapping
|
|
import logging
|
|
from typing import Any
|
|
|
|
from pyfibaro.fibaro_client import FibaroClient
|
|
from pyfibaro.fibaro_device import DeviceModel
|
|
from pyfibaro.fibaro_scene import SceneModel
|
|
from requests.exceptions import HTTPError
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
ATTR_ARMED,
|
|
ATTR_BATTERY_LEVEL,
|
|
CONF_PASSWORD,
|
|
CONF_URL,
|
|
CONF_USERNAME,
|
|
Platform,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import (
|
|
ConfigEntryAuthFailed,
|
|
ConfigEntryNotReady,
|
|
HomeAssistantError,
|
|
)
|
|
from homeassistant.helpers import device_registry as dr
|
|
from homeassistant.helpers.entity import DeviceInfo, Entity
|
|
from homeassistant.util import slugify
|
|
|
|
from .const import CONF_IMPORT_PLUGINS, DOMAIN
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
FIBARO_CONTROLLER = "fibaro_controller"
|
|
FIBARO_DEVICES = "fibaro_devices"
|
|
PLATFORMS = [
|
|
Platform.BINARY_SENSOR,
|
|
Platform.CLIMATE,
|
|
Platform.COVER,
|
|
Platform.LIGHT,
|
|
Platform.SCENE,
|
|
Platform.SENSOR,
|
|
Platform.LOCK,
|
|
Platform.SWITCH,
|
|
]
|
|
|
|
FIBARO_TYPEMAP = {
|
|
"com.fibaro.multilevelSensor": Platform.SENSOR,
|
|
"com.fibaro.binarySwitch": Platform.SWITCH,
|
|
"com.fibaro.multilevelSwitch": Platform.SWITCH,
|
|
"com.fibaro.FGD212": Platform.LIGHT,
|
|
"com.fibaro.FGR": Platform.COVER,
|
|
"com.fibaro.doorSensor": Platform.BINARY_SENSOR,
|
|
"com.fibaro.doorWindowSensor": Platform.BINARY_SENSOR,
|
|
"com.fibaro.FGMS001": Platform.BINARY_SENSOR,
|
|
"com.fibaro.heatDetector": Platform.BINARY_SENSOR,
|
|
"com.fibaro.lifeDangerSensor": Platform.BINARY_SENSOR,
|
|
"com.fibaro.smokeSensor": Platform.BINARY_SENSOR,
|
|
"com.fibaro.remoteSwitch": Platform.SWITCH,
|
|
"com.fibaro.sensor": Platform.SENSOR,
|
|
"com.fibaro.colorController": Platform.LIGHT,
|
|
"com.fibaro.securitySensor": Platform.BINARY_SENSOR,
|
|
"com.fibaro.hvac": Platform.CLIMATE,
|
|
"com.fibaro.hvacSystem": Platform.CLIMATE,
|
|
"com.fibaro.setpoint": Platform.CLIMATE,
|
|
"com.fibaro.FGT001": Platform.CLIMATE,
|
|
"com.fibaro.thermostatDanfoss": Platform.CLIMATE,
|
|
"com.fibaro.doorLock": Platform.LOCK,
|
|
"com.fibaro.binarySensor": Platform.BINARY_SENSOR,
|
|
"com.fibaro.accelerometer": Platform.BINARY_SENSOR,
|
|
}
|
|
|
|
|
|
class FibaroController:
|
|
"""Initiate Fibaro Controller Class."""
|
|
|
|
def __init__(self, config: Mapping[str, Any]) -> None:
|
|
"""Initialize the Fibaro controller."""
|
|
|
|
# The FibaroClient uses the correct API version automatically
|
|
self._client = FibaroClient(config[CONF_URL])
|
|
self._client.set_authentication(config[CONF_USERNAME], config[CONF_PASSWORD])
|
|
|
|
# Whether to import devices from plugins
|
|
self._import_plugins = config[CONF_IMPORT_PLUGINS]
|
|
self._room_map = None # Mapping roomId to room object
|
|
self._device_map = None # Mapping deviceId to device object
|
|
self.fibaro_devices: dict[Platform, list] = defaultdict(
|
|
list
|
|
) # List of devices by entity platform
|
|
self._callbacks: dict[Any, Any] = {} # Update value callbacks by deviceId
|
|
self.hub_serial: str # Unique serial number of the hub
|
|
self.hub_name: str # The friendly name of the hub
|
|
self.hub_software_version: str
|
|
self.hub_api_url: str = config[CONF_URL]
|
|
# Device infos by fibaro device id
|
|
self._device_infos: dict[int, DeviceInfo] = {}
|
|
|
|
def connect(self):
|
|
"""Start the communication with the Fibaro controller."""
|
|
|
|
connected = self._client.connect()
|
|
info = self._client.read_info()
|
|
self.hub_serial = info.serial_number
|
|
self.hub_name = info.hc_name
|
|
self.hub_software_version = info.current_version
|
|
|
|
if connected is False:
|
|
_LOGGER.error(
|
|
"Invalid login for Fibaro HC. Please check username and password"
|
|
)
|
|
return False
|
|
|
|
self._room_map = {room.fibaro_id: room for room in self._client.read_rooms()}
|
|
self._read_devices()
|
|
self._read_scenes()
|
|
return True
|
|
|
|
def connect_with_error_handling(self) -> None:
|
|
"""Translate connect errors to easily differentiate auth and connect failures.
|
|
|
|
When there is a better error handling in the used library this can be improved.
|
|
"""
|
|
try:
|
|
connected = self.connect()
|
|
if not connected:
|
|
raise FibaroConnectFailed("Connect status is false")
|
|
except HTTPError as http_ex:
|
|
if http_ex.response.status_code == 403:
|
|
raise FibaroAuthFailed from http_ex
|
|
|
|
raise FibaroConnectFailed from http_ex
|
|
except Exception as ex:
|
|
raise FibaroConnectFailed from ex
|
|
|
|
def enable_state_handler(self):
|
|
"""Start StateHandler thread for monitoring updates."""
|
|
self._client.register_update_handler(self._on_state_change)
|
|
|
|
def disable_state_handler(self):
|
|
"""Stop StateHandler thread used for monitoring updates."""
|
|
self._client.unregister_update_handler()
|
|
|
|
def _on_state_change(self, state):
|
|
"""Handle change report received from the HomeCenter."""
|
|
callback_set = set()
|
|
for change in state.get("changes", []):
|
|
try:
|
|
dev_id = change.pop("id")
|
|
if dev_id not in self._device_map:
|
|
continue
|
|
device = self._device_map[dev_id]
|
|
for property_name, value in change.items():
|
|
if property_name == "log":
|
|
if value and value != "transfer OK":
|
|
_LOGGER.debug("LOG %s: %s", device.friendly_name, value)
|
|
continue
|
|
if property_name == "logTemp":
|
|
continue
|
|
if property_name in device.properties:
|
|
device.properties[property_name] = value
|
|
_LOGGER.debug(
|
|
"<- %s.%s = %s", device.ha_id, property_name, str(value)
|
|
)
|
|
else:
|
|
_LOGGER.warning("%s.%s not found", device.ha_id, property_name)
|
|
if dev_id in self._callbacks:
|
|
callback_set.add(dev_id)
|
|
except (ValueError, KeyError):
|
|
pass
|
|
for item in callback_set:
|
|
for callback in self._callbacks[item]:
|
|
callback()
|
|
|
|
def register(self, device_id, callback):
|
|
"""Register device with a callback for updates."""
|
|
self._callbacks.setdefault(device_id, [])
|
|
self._callbacks[device_id].append(callback)
|
|
|
|
def get_children(self, device_id):
|
|
"""Get a list of child devices."""
|
|
return [
|
|
device
|
|
for device in self._device_map.values()
|
|
if device.parent_fibaro_id == device_id
|
|
]
|
|
|
|
def get_children2(self, device_id, endpoint_id):
|
|
"""Get a list of child devices for the same endpoint."""
|
|
return [
|
|
device
|
|
for device in self._device_map.values()
|
|
if device.parent_fibaro_id == device_id
|
|
and (not device.has_endpoint_id or device.endpoint_id == endpoint_id)
|
|
]
|
|
|
|
def get_siblings(self, device):
|
|
"""Get the siblings of a device."""
|
|
if device.has_endpoint_id:
|
|
return self.get_children2(
|
|
self._device_map[device.fibaro_id].parent_fibaro_id,
|
|
self._device_map[device.fibaro_id].endpoint_id,
|
|
)
|
|
return self.get_children(self._device_map[device.fibaro_id].parent_fibaro_id)
|
|
|
|
@staticmethod
|
|
def _map_device_to_platform(device: Any) -> Platform | None:
|
|
"""Map device to HA device type."""
|
|
# Use our lookup table to identify device type
|
|
platform: Platform | None = None
|
|
if device.type:
|
|
platform = FIBARO_TYPEMAP.get(device.type)
|
|
if platform is None and device.base_type:
|
|
platform = FIBARO_TYPEMAP.get(device.base_type)
|
|
|
|
# We can also identify device type by its capabilities
|
|
if platform is None:
|
|
if "setBrightness" in device.actions:
|
|
platform = Platform.LIGHT
|
|
elif "turnOn" in device.actions:
|
|
platform = Platform.SWITCH
|
|
elif "open" in device.actions:
|
|
platform = Platform.COVER
|
|
elif "secure" in device.actions:
|
|
platform = Platform.LOCK
|
|
elif device.value.has_value:
|
|
if device.value.is_bool_value:
|
|
platform = Platform.BINARY_SENSOR
|
|
else:
|
|
platform = Platform.SENSOR
|
|
|
|
# Switches that control lights should show up as lights
|
|
if platform == Platform.SWITCH and device.properties.get("isLight", False):
|
|
platform = Platform.LIGHT
|
|
return platform
|
|
|
|
def _create_device_info(
|
|
self, device: DeviceModel, devices: list[DeviceModel]
|
|
) -> None:
|
|
"""Create the device info. Unrooted entities are directly shown below the home center."""
|
|
|
|
# The home center is always id 1 (z-wave primary controller)
|
|
if device.parent_fibaro_id <= 1:
|
|
return
|
|
|
|
master_entity: Any | None = None
|
|
if device.parent_fibaro_id == 1:
|
|
master_entity = device
|
|
else:
|
|
for parent in devices:
|
|
if parent.fibaro_id == device.parent_fibaro_id:
|
|
master_entity = parent
|
|
if master_entity is None:
|
|
_LOGGER.error("Parent with id %s not found", device.parent_fibaro_id)
|
|
return
|
|
|
|
if "zwaveCompany" in master_entity.properties:
|
|
manufacturer = master_entity.properties.get("zwaveCompany")
|
|
else:
|
|
manufacturer = None
|
|
|
|
self._device_infos[master_entity.fibaro_id] = DeviceInfo(
|
|
identifiers={(DOMAIN, master_entity.fibaro_id)},
|
|
manufacturer=manufacturer,
|
|
name=master_entity.name,
|
|
via_device=(DOMAIN, self.hub_serial),
|
|
)
|
|
|
|
def get_device_info(self, device: Any) -> DeviceInfo:
|
|
"""Get the device info by fibaro device id."""
|
|
if device.fibaro_id in self._device_infos:
|
|
return self._device_infos[device.fibaro_id]
|
|
if device.parent_fibaro_id in self._device_infos:
|
|
return self._device_infos[device.parent_fibaro_id]
|
|
return DeviceInfo(identifiers={(DOMAIN, self.hub_serial)})
|
|
|
|
def _read_scenes(self):
|
|
scenes = self._client.read_scenes()
|
|
for device in scenes:
|
|
device.fibaro_controller = self
|
|
if device.room_id == 0:
|
|
room_name = "Unknown"
|
|
else:
|
|
room_name = self._room_map[device.room_id].name
|
|
device.room_name = room_name
|
|
device.friendly_name = f"{room_name} {device.name}"
|
|
device.ha_id = (
|
|
f"scene_{slugify(room_name)}_{slugify(device.name)}_{device.fibaro_id}"
|
|
)
|
|
device.unique_id_str = (
|
|
f"{slugify(self.hub_serial)}.scene.{device.fibaro_id}"
|
|
)
|
|
self.fibaro_devices[Platform.SCENE].append(device)
|
|
_LOGGER.debug("%s scene -> %s", device.ha_id, device)
|
|
|
|
def _read_devices(self):
|
|
"""Read and process the device list."""
|
|
devices = self._client.read_devices()
|
|
self._device_map = {}
|
|
last_climate_parent = None
|
|
last_endpoint = None
|
|
for device in devices:
|
|
try:
|
|
device.fibaro_controller = self
|
|
if device.room_id == 0:
|
|
room_name = "Unknown"
|
|
else:
|
|
room_name = self._room_map[device.room_id].name
|
|
device.room_name = room_name
|
|
device.friendly_name = f"{room_name} {device.name}"
|
|
device.ha_id = (
|
|
f"{slugify(room_name)}_{slugify(device.name)}_{device.fibaro_id}"
|
|
)
|
|
if device.enabled and (not device.is_plugin or self._import_plugins):
|
|
device.mapped_platform = self._map_device_to_platform(device)
|
|
else:
|
|
device.mapped_platform = None
|
|
if (platform := device.mapped_platform) is None:
|
|
continue
|
|
device.unique_id_str = f"{slugify(self.hub_serial)}.{device.fibaro_id}"
|
|
self._create_device_info(device, devices)
|
|
self._device_map[device.fibaro_id] = device
|
|
_LOGGER.debug(
|
|
"%s (%s, %s) -> %s %s",
|
|
device.ha_id,
|
|
device.type,
|
|
device.base_type,
|
|
platform,
|
|
str(device),
|
|
)
|
|
if platform != Platform.CLIMATE:
|
|
self.fibaro_devices[platform].append(device)
|
|
continue
|
|
# We group climate devices into groups with the same
|
|
# endPointID belonging to the same parent device.
|
|
if device.has_endpoint_id:
|
|
_LOGGER.debug(
|
|
"climate device: %s, endPointId: %s",
|
|
device.ha_id,
|
|
device.endpoint_id,
|
|
)
|
|
else:
|
|
_LOGGER.debug("climate device: %s, no endPointId", device.ha_id)
|
|
# If a sibling of this device has been added, skip this one
|
|
# otherwise add the first visible device in the group
|
|
# which is a hack, but solves a problem with FGT having
|
|
# hidden compatibility devices before the real device
|
|
if last_climate_parent != device.parent_fibaro_id or (
|
|
device.has_endpoint_id and last_endpoint != device.endpoint_id
|
|
):
|
|
_LOGGER.debug("Handle separately")
|
|
self.fibaro_devices[platform].append(device)
|
|
last_climate_parent = device.parent_fibaro_id
|
|
last_endpoint = device.endpoint_id
|
|
else:
|
|
_LOGGER.debug("not handling separately")
|
|
except (KeyError, ValueError):
|
|
pass
|
|
|
|
|
|
def _init_controller(data: Mapping[str, Any]) -> FibaroController:
|
|
"""Validate the user input allows us to connect to fibaro."""
|
|
controller = FibaroController(data)
|
|
controller.connect_with_error_handling()
|
|
return controller
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up the Fibaro Component.
|
|
|
|
The unique id of the config entry is the serial number of the home center.
|
|
"""
|
|
try:
|
|
controller = await hass.async_add_executor_job(_init_controller, entry.data)
|
|
except FibaroConnectFailed as connect_ex:
|
|
raise ConfigEntryNotReady(
|
|
f"Could not connect to controller at {entry.data[CONF_URL]}"
|
|
) from connect_ex
|
|
except FibaroAuthFailed as auth_ex:
|
|
raise ConfigEntryAuthFailed from auth_ex
|
|
|
|
data: dict[str, Any] = {}
|
|
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = data
|
|
data[FIBARO_CONTROLLER] = controller
|
|
devices = data[FIBARO_DEVICES] = {}
|
|
for platform in PLATFORMS:
|
|
devices[platform] = [*controller.fibaro_devices[platform]]
|
|
|
|
# register the hub device info separately as the hub has sometimes no entities
|
|
device_registry = dr.async_get(hass)
|
|
device_registry.async_get_or_create(
|
|
config_entry_id=entry.entry_id,
|
|
identifiers={(DOMAIN, controller.hub_serial)},
|
|
manufacturer="Fibaro",
|
|
name=controller.hub_name,
|
|
model=controller.hub_serial,
|
|
sw_version=controller.hub_software_version,
|
|
configuration_url=controller.hub_api_url.removesuffix("/api/"),
|
|
)
|
|
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
|
|
controller.enable_state_handler()
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
_LOGGER.debug("Shutting down Fibaro connection")
|
|
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
|
|
|
hass.data[DOMAIN][entry.entry_id][FIBARO_CONTROLLER].disable_state_handler()
|
|
hass.data[DOMAIN].pop(entry.entry_id)
|
|
|
|
return unload_ok
|
|
|
|
|
|
class FibaroDevice(Entity):
|
|
"""Representation of a Fibaro device entity."""
|
|
|
|
_attr_should_poll = False
|
|
|
|
def __init__(self, fibaro_device: DeviceModel | SceneModel) -> None:
|
|
"""Initialize the device."""
|
|
self.fibaro_device = fibaro_device
|
|
self.controller = fibaro_device.fibaro_controller
|
|
self.ha_id = fibaro_device.ha_id
|
|
self._attr_name = fibaro_device.friendly_name
|
|
self._attr_unique_id = fibaro_device.unique_id_str
|
|
|
|
if isinstance(fibaro_device, DeviceModel):
|
|
self._attr_device_info = self.controller.get_device_info(fibaro_device)
|
|
# propagate hidden attribute set in fibaro home center to HA
|
|
if not fibaro_device.visible:
|
|
self._attr_entity_registry_visible_default = False
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Call when entity is added to hass."""
|
|
self.controller.register(self.fibaro_device.fibaro_id, self._update_callback)
|
|
|
|
def _update_callback(self):
|
|
"""Update the state."""
|
|
self.schedule_update_ha_state(True)
|
|
|
|
@property
|
|
def level(self):
|
|
"""Get the level of Fibaro device."""
|
|
if self.fibaro_device.value.has_value:
|
|
return self.fibaro_device.value.int_value()
|
|
return None
|
|
|
|
@property
|
|
def level2(self):
|
|
"""Get the tilt level of Fibaro device."""
|
|
if self.fibaro_device.value_2.has_value:
|
|
return self.fibaro_device.value_2.int_value()
|
|
return None
|
|
|
|
def dont_know_message(self, action):
|
|
"""Make a warning in case we don't know how to perform an action."""
|
|
_LOGGER.warning(
|
|
"Not sure how to setValue: %s (available actions: %s)",
|
|
str(self.ha_id),
|
|
str(self.fibaro_device.actions),
|
|
)
|
|
|
|
def set_level(self, level):
|
|
"""Set the level of Fibaro device."""
|
|
self.action("setValue", level)
|
|
if self.fibaro_device.value.has_value:
|
|
self.fibaro_device.properties["value"] = level
|
|
if self.fibaro_device.has_brightness:
|
|
self.fibaro_device.properties["brightness"] = level
|
|
|
|
def set_level2(self, level):
|
|
"""Set the level2 of Fibaro device."""
|
|
self.action("setValue2", level)
|
|
if self.fibaro_device.value_2.has_value:
|
|
self.fibaro_device.properties["value2"] = level
|
|
|
|
def call_turn_on(self):
|
|
"""Turn on the Fibaro device."""
|
|
self.action("turnOn")
|
|
|
|
def call_turn_off(self):
|
|
"""Turn off the Fibaro device."""
|
|
self.action("turnOff")
|
|
|
|
def call_set_color(self, red, green, blue, white):
|
|
"""Set the color of Fibaro device."""
|
|
red = int(max(0, min(255, red)))
|
|
green = int(max(0, min(255, green)))
|
|
blue = int(max(0, min(255, blue)))
|
|
white = int(max(0, min(255, white)))
|
|
color_str = f"{red},{green},{blue},{white}"
|
|
self.fibaro_device.properties["color"] = color_str
|
|
self.action("setColor", str(red), str(green), str(blue), str(white))
|
|
|
|
def action(self, cmd, *args):
|
|
"""Perform an action on the Fibaro HC."""
|
|
if cmd in self.fibaro_device.actions:
|
|
self.fibaro_device.execute_action(cmd, args)
|
|
_LOGGER.debug("-> %s.%s%s called", str(self.ha_id), str(cmd), str(args))
|
|
else:
|
|
self.dont_know_message(cmd)
|
|
|
|
@property
|
|
def current_binary_state(self):
|
|
"""Return the current binary state."""
|
|
return self.fibaro_device.value.bool_value(False)
|
|
|
|
@property
|
|
def extra_state_attributes(self):
|
|
"""Return the state attributes of the device."""
|
|
attr = {"fibaro_id": self.fibaro_device.fibaro_id}
|
|
|
|
if isinstance(self.fibaro_device, DeviceModel):
|
|
if self.fibaro_device.has_battery_level:
|
|
attr[ATTR_BATTERY_LEVEL] = self.fibaro_device.battery_level
|
|
if self.fibaro_device.has_armed:
|
|
attr[ATTR_ARMED] = self.fibaro_device.armed
|
|
|
|
return attr
|
|
|
|
|
|
class FibaroConnectFailed(HomeAssistantError):
|
|
"""Error to indicate we cannot connect to fibaro home center."""
|
|
|
|
|
|
class FibaroAuthFailed(HomeAssistantError):
|
|
"""Error to indicate that authentication failed on fibaro home center."""
|