hass-core/homeassistant/components/neato/camera.py

"""Support for loading picture from Neato."""
from datetime import timedelta
import logging

from pybotvac.exceptions import NeatoRobotException

from homeassistant.components.camera import Camera

from .const import (
    NEATO_DOMAIN,
    NEATO_MAP_DATA,
    NEATO_ROBOTS,
    NEATO_LOGIN,
    SCAN_INTERVAL_MINUTES,
)

_LOGGER = logging.getLogger(__name__)

SCAN_INTERVAL = timedelta(minutes=SCAN_INTERVAL_MINUTES)


async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
    """Set up the Neato Camera."""
    pass


async def async_setup_entry(hass, entry, async_add_entities):
    """Set up Neato camera with config entry."""
    dev = []
    for robot in hass.data[NEATO_ROBOTS]:
        if "maps" in robot.traits:
            dev.append(NeatoCleaningMap(hass, robot))

    if not dev:
        return

    _LOGGER.debug("Adding robots for cleaning maps %s", dev)
    async_add_entities(dev, True)


class NeatoCleaningMap(Camera):
    """Neato cleaning map for last clean."""

    def __init__(self, hass, robot):
        """Initialize Neato cleaning map."""
        super().__init__()
        self.robot = robot
        self.neato = hass.data[NEATO_LOGIN] if NEATO_LOGIN in hass.data else None
        self._robot_name = f"{self.robot.name} Cleaning Map"
        self._robot_serial = self.robot.serial
        self._image_url = None
        self._image = None

    def camera_image(self):
        """Return image response."""
        self.update()
        return self._image

    def update(self):
        """Check the contents of the map list."""
        if self.neato is None:
            _LOGGER.error("Error while updating camera")
            self._image = None
            self._image_url = None
            return

        _LOGGER.debug("Running camera update")
        try:
            self.neato.update_robots()

            image_url = None
            map_data = self.hass.data[NEATO_MAP_DATA][self._robot_serial]["maps"][0]
            image_url = map_data["url"]
            if image_url == self._image_url:
                _LOGGER.debug("The map image_url is the same as old")
                return

            image = self.neato.download_map(image_url)
            self._image = image.read()
            self._image_url = image_url

        except NeatoRobotException as ex:
            _LOGGER.error("Neato camera connection error: %s", ex)
            self._image = None
            self._image_url = None

    @property
    def name(self):
        """Return the name of this camera."""
        return self._robot_name

    @property
    def unique_id(self):
        """Return unique ID."""
        return self._robot_serial

    @property
    def device_info(self):
        """Device info for neato robot."""
        return {"identifiers": {(NEATO_DOMAIN, self._robot_serial)}}