106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
"""
|
|
Support for Neato Connected Vaccums sensors.
|
|
|
|
For more details about this platform, please refer to the documentation at
|
|
https://home-assistant.io/components/sensor.neato/
|
|
"""
|
|
import logging
|
|
|
|
from homeassistant.helpers.entity import Entity
|
|
from homeassistant.components.neato import (
|
|
NEATO_ROBOTS, NEATO_LOGIN, ACTION, ERRORS, MODE, ALERTS)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
SENSOR_TYPE_STATUS = 'status'
|
|
SENSOR_TYPE_BATTERY = 'battery'
|
|
|
|
SENSOR_TYPES = {
|
|
SENSOR_TYPE_STATUS: ['Status'],
|
|
SENSOR_TYPE_BATTERY: ['Battery']
|
|
}
|
|
|
|
|
|
def setup_platform(hass, config, add_devices, discovery_info=None):
|
|
"""Setup the Neato sensor platform."""
|
|
if not hass.data['neato_robots']:
|
|
return False
|
|
|
|
dev = []
|
|
for robot in hass.data[NEATO_ROBOTS]:
|
|
for type_name in SENSOR_TYPES:
|
|
dev.append(NeatoConnectedSensor(hass, robot, type_name))
|
|
_LOGGER.debug('Adding sensors %s', dev)
|
|
add_devices(dev)
|
|
|
|
|
|
class NeatoConnectedSensor(Entity):
|
|
"""Neato Connected Sensor."""
|
|
|
|
def __init__(self, hass, robot, sensor_type):
|
|
"""Initialize the Neato Connected sensor."""
|
|
self.type = sensor_type
|
|
self.robot = robot
|
|
self.neato = hass.data[NEATO_LOGIN]
|
|
self._robot_name = self.robot.name + ' ' + SENSOR_TYPES[self.type][0]
|
|
self._state = self.robot.state
|
|
self._battery_state = None
|
|
self._status_state = None
|
|
|
|
def update(self):
|
|
"""Update the properties of sensor."""
|
|
_LOGGER.debug('Update of sensor')
|
|
self.neato.update_robots()
|
|
if not self._state:
|
|
return
|
|
self._state = self.robot.state
|
|
_LOGGER.debug('self._state=%s', self._state)
|
|
if self.type == SENSOR_TYPE_STATUS:
|
|
if self._state['state'] == 1:
|
|
if self._state['details']['isCharging']:
|
|
self._status_state = 'Charging'
|
|
elif (self._state['details']['isDocked'] and
|
|
not self._state['details']['isCharging']):
|
|
self._status_state = 'Docked'
|
|
else:
|
|
self._status_state = 'Stopped'
|
|
elif self._state['state'] == 2:
|
|
if ALERTS.get(self._state['error']) is None:
|
|
self._status_state = (
|
|
MODE.get(self._state['cleaning']['mode'])
|
|
+ ' ' + ACTION.get(self._state['action']))
|
|
else:
|
|
self._status_state = ALERTS.get(self._state['error'])
|
|
elif self._state['state'] == 3:
|
|
self._status_state = 'Paused'
|
|
elif self._state['state'] == 4:
|
|
self._status_state = ERRORS.get(self._state['error'])
|
|
if self.type == SENSOR_TYPE_BATTERY:
|
|
self._battery_state = self._state['details']['charge']
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return unit for the sensor."""
|
|
if self.type == SENSOR_TYPE_BATTERY:
|
|
return '%'
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return True if sensor data is available."""
|
|
if not self._state:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the sensor state."""
|
|
if self.type == SENSOR_TYPE_STATUS:
|
|
return self._status_state
|
|
if self.type == SENSOR_TYPE_BATTERY:
|
|
return self._battery_state
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return self._robot_name
|