d766aae436
Co-authored-by: copilot-swe-agent[bot] <[email protected]> Co-authored-by: frenck <[email protected]>
23 lines
555 B
Python
23 lines
555 B
Python
"""Base entity for Neato."""
|
|
|
|
from pybotvac import Robot
|
|
|
|
from homeassistant.helpers.device_registry import DeviceInfo
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
from .const import 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={(DOMAIN, self.robot.serial)},
|
|
name=self.robot.name,
|
|
)
|