evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Set botvac availability (#17350)

Browse source 
* Set botvac availability

* Lint

* Reduce availability calls per review comments
This commit is contained in:
Daniel Shokouhi 2018-10-11 23:40:45 -07:00 • committed by Martin Hjelmare
parent 0ea5a73e8d
commit 1c23a36f46
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
homeassistant/components/vacuum/neato.py
View file

@ -64,6 +64,8 @@ class NeatoConnectedVacuum(StateVacuumDevice):
self.clean_battery_end = None self.clean_battery_end = None
self.clean_suspension_charge_count = None self.clean_suspension_charge_count = None
self.clean_suspension_time = None self.clean_suspension_time = None
self._available = False
self._battery_level = None
def update(self): def update(self):
"""Update the states of Neato Vacuums.""" """Update the states of Neato Vacuums."""
@ -71,12 +73,12 @@ class NeatoConnectedVacuum(StateVacuumDevice):
self.neato.update_robots() self.neato.update_robots()
try: try:
self._state = self.robot.state self._state = self.robot.state
self._available = True
except (requests.exceptions.ConnectionError, except (requests.exceptions.ConnectionError,
requests.exceptions.HTTPError) as ex: requests.exceptions.HTTPError) as ex:
_LOGGER.warning("Neato connection error: %s", ex) _LOGGER.warning("Neato connection error: %s", ex)
self._state = None self._state = None
self._clean_state = STATE_ERROR self._available = False
self._status_state = 'Robot Offline'
return return
_LOGGER.debug('self._state=%s', self._state) _LOGGER.debug('self._state=%s', self._state)
if self._state['state'] == 1: if self._state['state'] == 1:
@ -127,6 +129,8 @@ class NeatoConnectedVacuum(StateVacuumDevice):
self.clean_battery_end = ( self.clean_battery_end = (
self._mapdata[self.robot.serial]['maps'][0]['run_charge_at_end']) self._mapdata[self.robot.serial]['maps'][0]['run_charge_at_end'])
self._battery_level = self._state['details']['charge']
@property @property
def name(self): def name(self):
"""Return the name of the device.""" """Return the name of the device."""
@ -140,7 +144,12 @@ class NeatoConnectedVacuum(StateVacuumDevice):
@property @property
def battery_level(self): def battery_level(self):
"""Return the battery level of the vacuum cleaner.""" """Return the battery level of the vacuum cleaner."""
return self._state['details']['charge'] return self._battery_level
@property
def available(self):
"""Return if the robot is available."""
return self._available
@property @property
def state(self): def state(self):