mirror of
https://github.com/home-assistant/core.git
synced 2025-07-18 18:57:06 +00:00
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
@ -150,6 +150,7 @@ omit =
|
|||||||
homeassistant/components/sensor/eliqonline.py
|
homeassistant/components/sensor/eliqonline.py
|
||||||
homeassistant/components/sensor/forecast.py
|
homeassistant/components/sensor/forecast.py
|
||||||
homeassistant/components/sensor/glances.py
|
homeassistant/components/sensor/glances.py
|
||||||
|
homeassistant/components/sensor/google_travel_time.py
|
||||||
homeassistant/components/sensor/gtfs.py
|
homeassistant/components/sensor/gtfs.py
|
||||||
homeassistant/components/sensor/loopenergy.py
|
homeassistant/components/sensor/loopenergy.py
|
||||||
homeassistant/components/sensor/netatmo.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
|
# homeassistant.components.notify.gntp
|
||||||
gntp==1.0.3
|
gntp==1.0.3
|
||||||
|
|
||||||
|
# homeassistant.components.sensor.google_travel_time
|
||||||
|
googlemaps==2.4.3
|
||||||
|
|
||||||
# homeassistant.components.mqtt.server
|
# homeassistant.components.mqtt.server
|
||||||
hbmqtt==0.6.3
|
hbmqtt==0.6.3
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user