25 lines
603 B
Python
25 lines
603 B
Python
"""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.robot = robot
|
|
self._attr_device_info: DeviceInfo = DeviceInfo(
|
|
identifiers={(NEATO_DOMAIN, self.robot.serial)},
|
|
name=self.robot.name,
|
|
)
|