entity.py

Go to the documentation of this file.

"""Base entity for Neato."""
 
from __future__ import annotations
 
from pybotvac import Robot
 
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
 
from .const import NEATO_DOMAIN
 
 
class NeatoEntity(Entity):
    """Base Neato entity."""
 
    _attr_has_entity_name = True
 
    def __init__(self, robot: Robot) -> None:
        """Initialize Neato entity."""
        self.robotrobot = robot
        self._attr_device_info: DeviceInfo = DeviceInfo(
            identifiers={(NEATO_DOMAIN, self.robotrobot.serial)},
            name=self.robotrobot.name,
        )