entity.py

Go to the documentation of this file.

"""The powerview integration base entity."""
 
import logging
 
from aiopvapi.resources.shade import BaseShade, ShadePosition
 
import homeassistant.helpers.device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import DOMAIN, MANUFACTURER
from .coordinator import PowerviewShadeUpdateCoordinator
from .model import PowerviewDeviceInfo
from .shade_data import PowerviewShadeData
 
_LOGGER = logging.getLogger(__name__)
 
 
class HDEntity(CoordinatorEntity[PowerviewShadeUpdateCoordinator]):
    """Base class for hunter douglas entities."""
 
    _attr_has_entity_name = True
 
    def __init__(
        self,
        coordinator: PowerviewShadeUpdateCoordinator,
        device_info: PowerviewDeviceInfo,
        room_name: str,
        powerview_id: str,
    ) -> None:
        """Initialize the entity."""
        super().__init__(coordinator)
        self._room_name_room_name = room_name
        self._attr_unique_id_attr_unique_id = f"{device_info.serial_number}_{powerview_id}"
        self._device_info_device_info = device_info
        self._configuration_url_configuration_url = self.coordinator.hub.url
 
    @property
    def data(self) -> PowerviewShadeData:
        """Return the PowerviewShadeData."""
        return self.coordinator.data
 
    @property
    def device_info(self) -> DeviceInfo:
        """Return the device_info of the device."""
        return DeviceInfo(
            connections={(dr.CONNECTION_NETWORK_MAC, self._device_info_device_info.mac_address)},
            identifiers={(DOMAIN, self._device_info_device_info.serial_number)},
            manufacturer=MANUFACTURER,
            model=self._device_info_device_info.model,
            name=self._device_info_device_info.name,
            sw_version=self._device_info_device_info.firmware,
            configuration_url=self._configuration_url_configuration_url,
        )
 
 
class ShadeEntity(HDEntity):
    """Base class for hunter douglas shade entities."""
 
    def __init__(
        self,
        coordinator: PowerviewShadeUpdateCoordinator,
        device_info: PowerviewDeviceInfo,
        room_name: str,
        shade: BaseShade,
        shade_name: str,
    ) -> None:
        """Initialize the shade."""
        super().__init__(coordinator, device_info, room_name, shade.id)
        self._shade_name_shade_name = shade_name
        self._shade_shade = shade
        self._is_hard_wired_is_hard_wired = not shade.is_battery_powered()
        self._configuration_url_configuration_url_configuration_url = shade.url
 
    @property
    def positions(self) -> ShadePosition:
        """Return the PowerviewShadeData."""
        return self.datadatadata.get_shade_position(self._shade_shade.id)
 
    @property
    def device_info(self) -> DeviceInfo:
        """Return the device_info of the device."""
        return DeviceInfo(
            identifiers={(DOMAIN, self._shade_shade.id)},
            name=self._shade_name_shade_name,
            suggested_area=self._room_name_room_name,
            manufacturer=MANUFACTURER,
            model=self._shade_shade.type_name,
            sw_version=self._shade_shade.firmware,
            via_device=(DOMAIN, self._device_info_device_info.serial_number),
            configuration_url=self._configuration_url_configuration_url_configuration_url,
        )