entity.py

Go to the documentation of this file.

"""Base entity for ROMY."""
 
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
from .const import DOMAIN
from .coordinator import RomyVacuumCoordinator
 
 
class RomyEntity(CoordinatorEntity[RomyVacuumCoordinator]):
    """Base ROMY entity."""
 
    _attr_has_entity_name = True
 
    def __init__(self, coordinator: RomyVacuumCoordinator) -> None:
        """Initialize ROMY entity."""
        super().__init__(coordinator)
        self.romyromyromy = coordinator.romy
        self._attr_device_info_attr_device_info = DeviceInfo(
            identifiers={(DOMAIN, self.romyromyromy.unique_id)},
            manufacturer="ROMY",
            name=self.romyromyromy.name,
            model=self.romyromyromy.model,
        )