entity.py

Go to the documentation of this file.

"""Generic platform."""
 
from __future__ import annotations
 
from devolo_plc_api.device_api import (
    ConnectedStationInfo,
    NeighborAPInfo,
    WifiGuestAccessGet,
)
from devolo_plc_api.plcnet_api import DataRate, LogicalNetwork
 
from homeassistant.const import ATTR_CONNECTIONS
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC, DeviceInfo
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from . import DevoloHomeNetworkConfigEntry
from .const import DOMAIN
from .coordinator import DevoloDataUpdateCoordinator
 
type _DataType = (
    LogicalNetwork
    | DataRate
    | list[ConnectedStationInfo]
    | list[NeighborAPInfo]
    | WifiGuestAccessGet
    | bool
    | int
)
 
 
class DevoloEntity(Entity):
    """Representation of a devolo home network device."""
 
    _attr_has_entity_name = True
 
    def __init__(
        self,
        entry: DevoloHomeNetworkConfigEntry,
    ) -> None:
        """Initialize a devolo home network device."""
        self.devicedevice = entry.runtime_data.device
        self.entryentry = entry
 
        self._attr_device_info_attr_device_info = DeviceInfo(
            configuration_url=f"http://{self.device.ip}",
            identifiers={(DOMAIN, str(self.devicedevice.serial_number))},
            manufacturer="devolo",
            model=self.devicedevice.product,
            model_id=self.devicedevice.mt_number,
            serial_number=self.devicedevice.serial_number,
            sw_version=self.devicedevice.firmware_version,
        )
        if self.devicedevice.mac:
            self._attr_device_info_attr_device_info[ATTR_CONNECTIONS] = {
                (CONNECTION_NETWORK_MAC, self.devicedevice.mac)
            }
        self._attr_translation_key_attr_translation_key = self.entity_description.key
        self._attr_unique_id_attr_unique_id = (
            f"{self.device.serial_number}_{self.entity_description.key}"
        )
 
 
class DevoloCoordinatorEntity[_DataT: _DataType](
    CoordinatorEntity[DevoloDataUpdateCoordinator[_DataT]], DevoloEntity
):
    """Representation of a coordinated devolo home network device."""
 
    def __init__(
        self,
        entry: DevoloHomeNetworkConfigEntry,
        coordinator: DevoloDataUpdateCoordinator[_DataT],
    ) -> None:
        """Initialize a devolo home network device."""
        super().__init__(coordinator)
        DevoloEntity.__init__(self, entry)