entity.py

Go to the documentation of this file.

"""SimpleFin Base Entity."""
 
from simplefin4py import Account
 
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity import EntityDescription
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import DOMAIN
from .coordinator import SimpleFinDataUpdateCoordinator
 
 
class SimpleFinEntity(CoordinatorEntity[SimpleFinDataUpdateCoordinator]):
    """Define a generic class for SimpleFIN entities."""
 
    _attr_attribution = "Data provided by SimpleFIN API"
    _attr_has_entity_name = True
 
    def __init__(
        self,
        coordinator: SimpleFinDataUpdateCoordinator,
        description: EntityDescription,
        account: Account,
    ) -> None:
        """Class initializer."""
        super().__init__(coordinator)
        self.entity_descriptionentity_description = description
        self._account_id_account_id = account.id
 
        self._attr_unique_id_attr_unique_id = f"account_{account.id}_{description.key}"
 
        self._attr_device_info_attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, account.id)},
            name=f"{account.org.name} {account.name}",
            entry_type=DeviceEntryType.SERVICE,
            manufacturer=account.org.name,
            model=account.name,
        )
 
    @property
    def account_data(self) -> Account:
        """Return the account data."""
        return self.coordinator.data.get_account_for_id(self._account_id_account_id)