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

Add timeout for requests

Browse source
This commit is contained in:
Fabian Affolter 2015-09-10 21:20:34 +02:00
parent f21d97d5a2
commit 53b43dc4db
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
homeassistant/components/sensor/swiss_public_transport.py
View file

@ -53,7 +53,8 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
try: try:
for location in [config.get('from', None), config.get('to', None)]: for location in [config.get('from', None), config.get('to', None)]:
# transport.opendata.ch doesn't play nice with requests.Session # transport.opendata.ch doesn't play nice with requests.Session
result = get(_RESOURCE + 'locations?query=%s' % location) result = get(_RESOURCE + 'locations?query=%s' % location,
timeout=10)
journey.append(result.json()['stations'][0]['name']) journey.append(result.json()['stations'][0]['name'])
except KeyError: except KeyError:
_LOGGER.exception( _LOGGER.exception(
@ -115,8 +116,8 @@ class PublicTransportData(object):
'from=' + self.start + '&' + 'from=' + self.start + '&' +
'to=' + self.destination + '&' + 'to=' + self.destination + '&' +
'fields[]=connections/from/departureTimestamp/&' + 'fields[]=connections/from/departureTimestamp/&' +
'fields[]=connections/') 'fields[]=connections/',
timeout=10)
connections = response.json()['connections'][:2] connections = response.json()['connections'][:2]
try: try: