From 8e394c2203f79374e92870389afc14a96cd83709 Mon Sep 17 00:00:00 2001 From: Romain Bignon Date: Wed, 24 Mar 2010 00:32:25 +0100 Subject: [PATCH] aibility to use search stations --- weboob/backends/transilien/backend.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/weboob/backends/transilien/backend.py b/weboob/backends/transilien/backend.py index f26b0a8f..fdd7c702 100644 --- a/weboob/backends/transilien/backend.py +++ b/weboob/backends/transilien/backend.py @@ -22,6 +22,7 @@ from weboob.backend import Backend from weboob.capabilities.travel import ICapTravel, Station, Departure from .browser import Transilien +from .stations import STATIONS class TransilienBackend(Backend, ICapTravel): MAINTAINER = u'Julien Hébert' @@ -32,9 +33,10 @@ class TransilienBackend(Backend, ICapTravel): Backend.__init__(self, weboob) def iter_station_search(self, pattern): - transilien = Transilien() - for _id, name in transilien.iter_station_search(pattern): - yield Station(_id, name) + pattern = pattern.lower() + for _id, name in STATIONS.iteritems(): + if name.lower().find(pattern) >= 0: + yield Station(_id, name) def iter_station_departures(self, station_id, arrival_id=None): transilien = Transilien()