Add google maps travel time sensor (#1987)
* Add travel time sensor * Throttle update of Time travel sensor * Rename travel time to google travel time, validate the api key
This commit is contained in:
parent
6901e5ea5e
commit
6cae7c0307
3 changed files with 121 additions and 0 deletions
|
@ -150,6 +150,7 @@ omit =
|
|||
homeassistant/components/sensor/eliqonline.py
|
||||
homeassistant/components/sensor/forecast.py
|
||||
homeassistant/components/sensor/glances.py
|
||||
homeassistant/components/sensor/google_travel_time.py
|
||||
homeassistant/components/sensor/gtfs.py
|
||||
homeassistant/components/sensor/loopenergy.py
|
||||
homeassistant/components/sensor/netatmo.py
|
||||
|
|
117
homeassistant/components/sensor/google_travel_time.py
Normal file
117
homeassistant/components/sensor/google_travel_time.py
Normal file
|
@ -0,0 +1,117 @@
|
|||
"""
|
||||
Support for Google travel time sensors.
|
||||
|
||||
For more details about this platform, please refer to the documentation at
|
||||
https://home-assistant.io/components/sensor.google_travel_time/
|
||||
"""
|
||||
from datetime import datetime
|
||||
from datetime import timedelta
|
||||
import logging
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.helpers.entity import Entity
|
||||
from homeassistant.const import CONF_API_KEY, TEMP_CELSIUS
|
||||
from homeassistant.util import Throttle
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
REQUIREMENTS = ['googlemaps==2.4.3']
|
||||
|
||||
# Return cached results if last update was less then this time ago
|
||||
MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=5)
|
||||
|
||||
CONF_ORIGIN = 'origin'
|
||||
CONF_DESTINATION = 'destination'
|
||||
CONF_TRAVEL_MODE = 'travel_mode'
|
||||
|
||||
PLATFORM_SCHEMA = vol.Schema({
|
||||
vol.Required('platform'): 'google_travel_time',
|
||||
vol.Required(CONF_API_KEY): vol.Coerce(str),
|
||||
vol.Required(CONF_ORIGIN): vol.Coerce(str),
|
||||
vol.Required(CONF_DESTINATION): vol.Coerce(str),
|
||||
vol.Optional(CONF_TRAVEL_MODE, default='driving'):
|
||||
vol.In(["driving", "walking", "bicycling", "transit"])
|
||||
})
|
||||
|
||||
|
||||
def setup_platform(hass, config, add_devices_callback, discovery_info=None):
|
||||
"""Setup the travel time platform."""
|
||||
# pylint: disable=too-many-locals
|
||||
|
||||
is_metric = (hass.config.temperature_unit == TEMP_CELSIUS)
|
||||
api_key = config.get(CONF_API_KEY)
|
||||
origin = config.get(CONF_ORIGIN)
|
||||
destination = config.get(CONF_DESTINATION)
|
||||
travel_mode = config.get(CONF_TRAVEL_MODE)
|
||||
|
||||
sensor = GoogleTravelTimeSensor(api_key, origin, destination,
|
||||
travel_mode, is_metric)
|
||||
|
||||
if sensor.valid_api_connection:
|
||||
add_devices_callback([sensor])
|
||||
|
||||
|
||||
class GoogleTravelTimeSensor(Entity):
|
||||
"""Representation of a tavel time sensor."""
|
||||
|
||||
# pylint: disable=too-many-arguments
|
||||
def __init__(self, api_key, origin, destination, travel_mode, is_metric):
|
||||
"""Initialize the sensor."""
|
||||
if is_metric:
|
||||
self._unit = 'metric'
|
||||
else:
|
||||
self._unit = 'imperial'
|
||||
self._origin = origin
|
||||
self._destination = destination
|
||||
self._travel_mode = travel_mode
|
||||
self._matrix = None
|
||||
self.valid_api_connection = True
|
||||
|
||||
import googlemaps
|
||||
self._client = googlemaps.Client(api_key, timeout=10)
|
||||
try:
|
||||
self.update()
|
||||
except googlemaps.exceptions.ApiError as exp:
|
||||
_LOGGER .error(exp)
|
||||
self.valid_api_connection = False
|
||||
return
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
"""Return the state of the sensor."""
|
||||
return self._matrix['rows'][0]['elements'][0]['duration']['value']/60.0
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Get the name of the sensor."""
|
||||
return "Google Travel time"
|
||||
|
||||
@property
|
||||
def device_state_attributes(self):
|
||||
"""Return the state attributes."""
|
||||
res = self._matrix.copy()
|
||||
del res['rows']
|
||||
_data = self._matrix['rows'][0]['elements'][0]
|
||||
if 'duration_in_traffic' in _data:
|
||||
res['duration_in_traffic'] = _data['duration_in_traffic']['text']
|
||||
if 'duration' in _data:
|
||||
res['duration'] = _data['duration']['text']
|
||||
if 'distance' in _data:
|
||||
res['distance'] = _data['distance']['text']
|
||||
return res
|
||||
|
||||
@property
|
||||
def unit_of_measurement(self):
|
||||
"""Return the unit this state is expressed in."""
|
||||
return "min"
|
||||
|
||||
@Throttle(MIN_TIME_BETWEEN_UPDATES)
|
||||
def update(self):
|
||||
"""Get the latest data from Google."""
|
||||
now = datetime.now()
|
||||
self._matrix = self._client.distance_matrix(self._origin,
|
||||
self._destination,
|
||||
mode=self._travel_mode,
|
||||
units=self._unit,
|
||||
departure_time=now,
|
||||
traffic_model="optimistic")
|
|
@ -62,6 +62,9 @@ fuzzywuzzy==0.8.0
|
|||
# homeassistant.components.notify.gntp
|
||||
gntp==1.0.3
|
||||
|
||||
# homeassistant.components.sensor.google_travel_time
|
||||
googlemaps==2.4.3
|
||||
|
||||
# homeassistant.components.mqtt.server
|
||||
hbmqtt==0.6.3
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue