mirror of
https://github.com/home-assistant/core.git
synced 2026-04-17 23:53:49 +01:00
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
"""Support for Neato botvac connected vacuum cleaners."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import timedelta
|
|
import logging
|
|
from typing import Any
|
|
|
|
from pybotvac import Account
|
|
from urllib3.response import HTTPResponse
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.util import Throttle
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class NeatoHub:
|
|
"""A My Neato hub wrapper class."""
|
|
|
|
def __init__(self, hass: HomeAssistant, neato: Account) -> None:
|
|
"""Initialize the Neato hub."""
|
|
self._hass = hass
|
|
self.my_neato: Account = neato
|
|
self.robots: set[Any] = set()
|
|
self.persistent_maps: dict[str, Any] = {}
|
|
self.map_data: dict[str, Any] = {}
|
|
|
|
@Throttle(timedelta(minutes=1))
|
|
def update_robots(self) -> None:
|
|
"""Update the robot states."""
|
|
_LOGGER.debug("Running HUB.update_robots %s", self.robots)
|
|
self.robots = self.my_neato.robots
|
|
self.persistent_maps = self.my_neato.persistent_maps
|
|
self.map_data = self.my_neato.maps
|
|
|
|
def download_map(self, url: str) -> HTTPResponse:
|
|
"""Download a new map image."""
|
|
map_image_data: HTTPResponse = self.my_neato.get_map_image(url)
|
|
return map_image_data
|
|
|
|
async def async_update_entry_unique_id(self, entry: ConfigEntry) -> str:
|
|
"""Update entry for unique_id."""
|
|
|
|
await self._hass.async_add_executor_job(self.my_neato.refresh_userdata)
|
|
unique_id: str = self.my_neato.unique_id
|
|
|
|
if entry.unique_id == unique_id:
|
|
return unique_id
|
|
|
|
_LOGGER.debug("Updating user unique_id for previous config entry")
|
|
self._hass.config_entries.async_update_entry(entry, unique_id=unique_id)
|
|
return unique_id
|