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

aibility to use search stations

Browse source
This commit is contained in:
Romain Bignon 2010-03-24 00:32:25 +01:00
commit 8e394c2203
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
weboob/backends/transilien/backend.py
View file

@ -22,6 +22,7 @@ from weboob.backend import Backend
from weboob.capabilities.travel import ICapTravel, Station, Departure from weboob.capabilities.travel import ICapTravel, Station, Departure
from .browser import Transilien from .browser import Transilien
from .stations import STATIONS
class TransilienBackend(Backend, ICapTravel): class TransilienBackend(Backend, ICapTravel):
MAINTAINER = u'Julien Hébert' MAINTAINER = u'Julien Hébert'
@ -32,9 +33,10 @@ class TransilienBackend(Backend, ICapTravel):
Backend.__init__(self, weboob) Backend.__init__(self, weboob)
def iter_station_search(self, pattern): def iter_station_search(self, pattern):
transilien = Transilien() pattern = pattern.lower()
for _id, name in transilien.iter_station_search(pattern): for _id, name in STATIONS.iteritems():
yield Station(_id, name) if name.lower().find(pattern) >= 0:
yield Station(_id, name)
def iter_station_departures(self, station_id, arrival_id=None): def iter_station_departures(self, station_id, arrival_id=None):
transilien = Transilien() transilien = Transilien()