core/homeassistant/components/romy/entity.py

25 lines
754 B
Python

"""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.romy = coordinator.romy
self._attr_device_info = DeviceInfo(
identifiers={(DOMAIN, self.romy.unique_id)},
manufacturer="ROMY",
name=self.romy.name,
model=self.romy.model,
)