mirror of
https://github.com/Matysh/houseplan-card
synced 2026-07-31 16:38:31 +00:00
Matysh
18b8589438
The public stand cannot run any Tier-A map integration (no radios, no robots), which left the whole vacuum section of docs/TESTING.md untestable outside the owner's home. demo_robot fakes exactly the surface the card consumes: a Tasshack-shaped map sensor (dict vacuum_position, rooms named after the plan rooms so auto-calibration has something to match, flipped Y so the mirror default is meaningful) and a vacuum that drives a serpentine route through every room in ~3.5 minutes and docks itself.
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
"""The vacuum entity of the stand simulator."""
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from homeassistant.components.vacuum import (
|
|
StateVacuumEntity,
|
|
VacuumActivity,
|
|
VacuumEntityFeature,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from . import DOMAIN, RobotSim
|
|
|
|
ACTIVITY = {
|
|
"docked": VacuumActivity.DOCKED,
|
|
"cleaning": VacuumActivity.CLEANING,
|
|
"paused": VacuumActivity.PAUSED,
|
|
"returning": VacuumActivity.RETURNING,
|
|
}
|
|
|
|
|
|
async def async_setup_platform(hass: HomeAssistant, config, add_entities, discovery_info=None) -> None:
|
|
add_entities([DemoRobotVacuum(hass.data[DOMAIN])])
|
|
|
|
|
|
class DemoRobotVacuum(StateVacuumEntity):
|
|
_attr_should_poll = False
|
|
_attr_name = "Робот-пылесос"
|
|
_attr_unique_id = "demo_robot_vacuum"
|
|
_attr_supported_features = (
|
|
VacuumEntityFeature.START
|
|
| VacuumEntityFeature.STOP
|
|
| VacuumEntityFeature.PAUSE
|
|
| VacuumEntityFeature.RETURN_HOME
|
|
| VacuumEntityFeature.STATE
|
|
)
|
|
|
|
def __init__(self, sim: RobotSim) -> None:
|
|
self._sim = sim
|
|
self.entity_id = "vacuum.demo_robot"
|
|
|
|
async def async_added_to_hass(self) -> None:
|
|
self.async_on_remove(self._sim.add_listener(self.async_write_ha_state))
|
|
|
|
@property
|
|
def activity(self) -> VacuumActivity:
|
|
return ACTIVITY.get(self._sim.state, VacuumActivity.IDLE)
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, Any]:
|
|
# the card resolves the active map name through the vacuum entity
|
|
return {"selected_map": "1"}
|
|
|
|
async def async_start(self) -> None:
|
|
self._sim.start()
|
|
|
|
async def async_pause(self) -> None:
|
|
self._sim.pause()
|
|
|
|
async def async_stop(self, **kwargs: Any) -> None:
|
|
self._sim.stop()
|
|
|
|
async def async_return_to_base(self, **kwargs: Any) -> None:
|
|
self._sim.return_to_base()
|