nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

if station ID is not found, try to look for it as a name and take the first result

Browse source 
It is possible to do both:

traveloob> departures 2656@voyagessncf 2128@voyagessncf

and:

traveloob> departures paris lyon
This commit is contained in:
Romain Bignon 2014-01-14 23:21:31 +01:00
commit b319b92a15
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
modules/voyagessncf/backend.py
View file

@ -104,7 +104,11 @@ class VoyagesSNCFBackend(BaseBackend, ICapTravel):
station = self.STATIONS[int(station_id)] station = self.STATIONS[int(station_id)]
arrival = self.STATIONS[int(arrival_id)] arrival = self.STATIONS[int(arrival_id)]
except (IndexError, ValueError): except (IndexError, ValueError):
raise UserError('Unknown station') try:
station = list(self.iter_station_search(station_id))[0].name
arrival = list(self.iter_station_search(arrival_id))[0].name
except IndexError:
raise UserError('Unknown station')
with self.browser: with self.browser:
for i, d in enumerate(self.browser.iter_departures(station, arrival, date, for i, d in enumerate(self.browser.iter_departures(station, arrival, date,