From 8aa136f7ed6e9bf95fc0324be61157eea879a3b5 Mon Sep 17 00:00:00 2001 From: Lars Lydersen Date: Mon, 7 Jan 2019 20:03:22 +0100 Subject: [PATCH] Added support for spot cleaning that was introduced in pybotvac 0.12. --- homeassistant/components/vacuum/neato.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/vacuum/neato.py b/homeassistant/components/vacuum/neato.py index 0c97f99df99..8abf2b2779a 100644 --- a/homeassistant/components/vacuum/neato.py +++ b/homeassistant/components/vacuum/neato.py @@ -13,7 +13,7 @@ from homeassistant.components.vacuum import ( SUPPORT_STATE, SUPPORT_STOP, SUPPORT_START, STATE_IDLE, STATE_PAUSED, STATE_CLEANING, STATE_DOCKED, STATE_RETURNING, STATE_ERROR, SUPPORT_MAP, ATTR_STATUS, ATTR_BATTERY_LEVEL, ATTR_BATTERY_ICON, - SUPPORT_LOCATE) + SUPPORT_LOCATE, SUPPORT_CLEAN_SPOT) from homeassistant.components.neato import ( NEATO_ROBOTS, NEATO_LOGIN, NEATO_MAP_DATA, ACTION, ERRORS, MODE, ALERTS) @@ -24,7 +24,7 @@ DEPENDENCIES = ['neato'] SCAN_INTERVAL = timedelta(minutes=5) SUPPORT_NEATO = SUPPORT_BATTERY | SUPPORT_PAUSE | SUPPORT_RETURN_HOME | \ - SUPPORT_STOP | SUPPORT_START | \ + SUPPORT_STOP | SUPPORT_START | SUPPORT_CLEAN_SPOT\ SUPPORT_STATE | SUPPORT_MAP | SUPPORT_LOCATE ATTR_CLEAN_START = 'clean_start' @@ -222,3 +222,7 @@ class NeatoConnectedVacuum(StateVacuumDevice): def locate(self, **kwargs): """Locate the robot by making it emit a sound.""" self.robot.locate() + + def clean_spot(self, **kwargs): + """Run a spot cleaning starting from the base.""" + self.robot.start_spot_cleaning()