If we have duration_in_traffic use that as the state, otherwise use duration

This commit is contained in:
Robbie Trencheny 2016-05-23 14:05:12 -07:00
parent b3afb386b7
commit c96f73d1be

View file

@ -136,10 +136,11 @@ class GoogleTravelTimeSensor(Entity):
@property @property
def state(self): def state(self):
"""Return the state of the sensor.""" """Return the state of the sensor."""
try: _data = self._matrix['rows'][0]['elements'][0]
res = self._matrix['rows'][0]['elements'][0]['duration']['value'] if 'duration_in_traffic' in _data:
return round(res/60) return round(_data['duration_in_traffic']['value']/60)
except KeyError: if 'duration' in _data:
return round(_data['duration']['value']/60)
return None return None
@property @property