core/homeassistant/components/kiosker/entity.py

"""Base entity for Kiosker."""

from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity

from .const import DOMAIN
from .coordinator import KioskerDataUpdateCoordinator


class KioskerEntity(CoordinatorEntity[KioskerDataUpdateCoordinator]):
    """Base class for Kiosker entities."""

    _attr_has_entity_name = True

    def __init__(
        self,
        coordinator: KioskerDataUpdateCoordinator,
        description: EntityDescription,
    ) -> None:
        """Initialize the entity."""
        super().__init__(coordinator)

        self.entity_description = description

        status = coordinator.data.status
        device_id = status.device_id
        model = status.model
        app_name = status.app_name
        app_version = status.app_version
        os_version = status.os_version

        # Use uppercased truncated device ID for display purposes (device name, titles)
        device_id_short_display = device_id[:8].upper()

        # Set device info
        self._attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, device_id)},
            name=(f"Kiosker {device_id_short_display}"),
            sw_version=(f"{app_name} {app_version}"),
            hw_version=(
                None
                if model is None
                else model
                if os_version is None
                else f"{model} ({os_version})"
            ),
            serial_number=device_id,
        )

        self._attr_unique_id = f"{device_id}_{description.key}"