add parameter 'date' to ICapTravel.iter_station_departures

This commit is contained in:
Romain Bignon 2013-12-14 15:47:41 +01:00
commit c36cfdc811
3 changed files with 5 additions and 3 deletions

View file

@ -39,7 +39,7 @@ class CanalTPBackend(BaseBackend, ICapTravel):
for _id, name in self.browser.iter_station_search(pattern):
yield Station(_id, name)
def iter_station_departures(self, station_id, arrival_id=None):
def iter_station_departures(self, station_id, arrival_id=None, date=None):
for i, d in enumerate(self.browser.iter_station_departures(station_id, arrival_id)):
departure = Departure(i, d['type'], d['time'])
departure.departure_station = d['departure']

View file

@ -42,7 +42,7 @@ class TransilienBackend(BaseBackend, ICapTravel):
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, date=None):
with self.browser:
for i, d in enumerate(self.browser.iter_station_departures(station_id, arrival_id)):
departure = Departure(i, d['type'], d['time'])