entity.py

Go to the documentation of this file.

"""Support for the Hive devices and services."""
 
from __future__ import annotations
 
from typing import Any
 
from apyhiveapi import Hive
 
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity
 
from .const import DOMAIN
 
 
class HiveEntity(Entity):
    """Initiate Hive Base Class."""
 
    def __init__(self, hive: Hive, hive_device: dict[str, Any]) -> None:
        """Initialize the instance."""
        self.hivehive = hive
        self.devicedevice = hive_device
        self._attr_name_attr_name = self.devicedevice["haName"]
        self._attr_unique_id_attr_unique_id = f'{self.device["hiveID"]}-{self.device["hiveType"]}'
        self._attr_device_info_attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, self.devicedevice["device_id"])},
            model=self.devicedevice["deviceData"]["model"],
            manufacturer=self.devicedevice["deviceData"]["manufacturer"],
            name=self.devicedevice["device_name"],
            sw_version=self.devicedevice["deviceData"]["version"],
            via_device=(DOMAIN, self.devicedevice["parentDevice"]),
        )
        self.attributes: dict[str, Any] = {}
 
    async def async_added_to_hass(self) -> None:
        """When entity is added to Home Assistant."""
        self.async_on_removeasync_on_remove(
            async_dispatcher_connect(self.hasshass, DOMAIN, self.async_write_ha_stateasync_write_ha_state)
        )