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

display price information

Browse source
This commit is contained in:
Romain Bignon 2013-12-15 14:02:49 +01:00
commit d8ccc81f93
2 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
modules/voyagessncf/backend.py
View file

@ -79,8 +79,16 @@ class VoyagesSNCFBackend(BaseBackend, ICapTravel):
self._populate_stations()
pattern = pattern.lower()
already = set()
# First stations whose name starts with pattern...
for _id, name in enumerate(self.STATIONS):
if name.lower().startswith(pattern):
already.add(_id)
yield Station(_id, unicode(name))
# ...then ones whose name contains pattern.
for _id, name in enumerate(self.STATIONS):
if pattern in name.lower() and not _id in already:
yield Station(_id, unicode(name))
def iter_station_departures(self, station_id, arrival_id=None, date=None):
@ -105,4 +113,5 @@ class VoyagesSNCFBackend(BaseBackend, ICapTravel):
departure.arrival_time = d['arrival_time']
departure.price = d['price']
departure.currency = d['currency']
departure.information = d['price_info']
yield departure