entity.py

Go to the documentation of this file.

"""Base entity for Monzo."""
 
from __future__ import annotations
 
from collections.abc import Callable
from typing import Any
 
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import DOMAIN
from .coordinator import MonzoCoordinator, MonzoData
 
 
class MonzoBaseEntity(CoordinatorEntity[MonzoCoordinator]):
    """Common base for Monzo entities."""
 
    _attr_attribution = "Data provided by Monzo"
    _attr_has_entity_name = True
 
    def __init__(
        self,
        coordinator: MonzoCoordinator,
        index: int,
        device_model: str,
        data_accessor: Callable[[MonzoData], list[dict[str, Any]]],
    ) -> None:
        """Initialize sensor."""
        super().__init__(coordinator)
        self.indexindex = index
        self._data_accessor_data_accessor = data_accessor
 
        self._attr_device_info_attr_device_info = DeviceInfo(
            entry_type=DeviceEntryType.SERVICE,
            identifiers={(DOMAIN, str(self.datadatadata["id"]))},
            manufacturer="Monzo",
            model=device_model,
            name=self.datadatadata["name"],
        )
 
    @property
    def data(self) -> dict[str, Any]:
        """Shortcut to access coordinator data for the entity."""
        return self._data_accessor_data_accessor(self.coordinator.data)[self.indexindex]