From 06c3f756b1b0a106d54589e9a07b39eb3273f8e2 Mon Sep 17 00:00:00 2001
From: Daniel Shokouhi <dshokouhi@gmail.com>
Date: Wed, 18 Jul 2018 03:19:38 -0700
Subject: [PATCH] Implement locate service for neato (#15467)

* Implement locate service for neato

* Hound
---
 homeassistant/components/neato.py        | 2 +-
 homeassistant/components/vacuum/neato.py | 9 +++++++--
 requirements_all.txt                     | 2 +-
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/neato.py b/homeassistant/components/neato.py
index fc407de0a6b..63b0f61bb7c 100644
--- a/homeassistant/components/neato.py
+++ b/homeassistant/components/neato.py
@@ -17,7 +17,7 @@ from homeassistant.util import Throttle
 
 _LOGGER = logging.getLogger(__name__)
 
-REQUIREMENTS = ['pybotvac==0.0.7']
+REQUIREMENTS = ['pybotvac==0.0.8']
 
 DOMAIN = 'neato'
 NEATO_ROBOTS = 'neato_robots'
diff --git a/homeassistant/components/vacuum/neato.py b/homeassistant/components/vacuum/neato.py
index 6289fed265d..224e763a097 100644
--- a/homeassistant/components/vacuum/neato.py
+++ b/homeassistant/components/vacuum/neato.py
@@ -12,7 +12,8 @@ from homeassistant.const import STATE_OFF, STATE_ON
 from homeassistant.components.vacuum import (
     VacuumDevice, SUPPORT_BATTERY, SUPPORT_PAUSE, SUPPORT_RETURN_HOME,
     SUPPORT_STATUS, SUPPORT_STOP, SUPPORT_TURN_OFF, SUPPORT_TURN_ON,
-    SUPPORT_MAP, ATTR_STATUS, ATTR_BATTERY_LEVEL, ATTR_BATTERY_ICON)
+    SUPPORT_MAP, ATTR_STATUS, ATTR_BATTERY_LEVEL, ATTR_BATTERY_ICON,
+    SUPPORT_LOCATE)
 from homeassistant.components.neato import (
     NEATO_ROBOTS, NEATO_LOGIN, NEATO_MAP_DATA, ACTION, ERRORS, MODE, ALERTS)
 
@@ -24,7 +25,7 @@ SCAN_INTERVAL = timedelta(minutes=5)
 
 SUPPORT_NEATO = SUPPORT_BATTERY | SUPPORT_PAUSE | SUPPORT_RETURN_HOME | \
                  SUPPORT_STOP | SUPPORT_TURN_OFF | SUPPORT_TURN_ON | \
-                 SUPPORT_STATUS | SUPPORT_MAP
+                 SUPPORT_STATUS | SUPPORT_MAP | SUPPORT_LOCATE
 
 ATTR_CLEAN_START = 'clean_start'
 ATTR_CLEAN_STOP = 'clean_stop'
@@ -211,3 +212,7 @@ class NeatoConnectedVacuum(VacuumDevice):
             self.robot.pause_cleaning()
         if self._state['state'] == 3:
             self.robot.resume_cleaning()
+
+    def locate(self, **kwargs):
+        """Locate the robot by making it emit a sound."""
+        self.robot.locate()
diff --git a/requirements_all.txt b/requirements_all.txt
index a39a2cada31..8ef15e4bd18 100644
--- a/requirements_all.txt
+++ b/requirements_all.txt
@@ -753,7 +753,7 @@ pyblackbird==0.5
 # pybluez==0.22
 
 # homeassistant.components.neato
-pybotvac==0.0.7
+pybotvac==0.0.8
 
 # homeassistant.components.cloudflare
 pycfdns==0.0.1
-- 
GitLab