entity.py

Go to the documentation of this file.

"""The aurora component."""
 
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import ATTRIBUTION, DOMAIN
from .coordinator import AuroraDataUpdateCoordinator
 
 
class AuroraEntity(CoordinatorEntity[AuroraDataUpdateCoordinator]):
    """Implementation of the base Aurora Entity."""
 
    _attr_attribution = ATTRIBUTION
    _attr_has_entity_name = True
 
    def __init__(
        self,
        coordinator: AuroraDataUpdateCoordinator,
        translation_key: str,
    ) -> None:
        """Initialize the Aurora Entity."""
 
        super().__init__(coordinator=coordinator)
 
        self._attr_translation_key_attr_translation_key = translation_key
        self._attr_unique_id_attr_unique_id = f"{coordinator.latitude}_{coordinator.longitude}"
        self._attr_device_info_attr_device_info = DeviceInfo(
            entry_type=DeviceEntryType.SERVICE,
            identifiers={(DOMAIN, self._attr_unique_id_attr_unique_id)},
            manufacturer="NOAA",
            model="Aurora Visibility Sensor",
        )