From c36cfdc811b0f1889ecf60316520f75a7125d9cd Mon Sep 17 00:00:00 2001 From: Romain Bignon Date: Sat, 14 Dec 2013 15:47:41 +0100 Subject: [PATCH] add parameter 'date' to ICapTravel.iter_station_departures --- modules/canaltp/backend.py | 2 +- modules/transilien/backend.py | 2 +- weboob/capabilities/travel.py | 4 +++- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/modules/canaltp/backend.py b/modules/canaltp/backend.py index 7d5c3f90..525bda1c 100644 --- a/modules/canaltp/backend.py +++ b/modules/canaltp/backend.py @@ -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'] diff --git a/modules/transilien/backend.py b/modules/transilien/backend.py index f7e202b9..1dff7376 100644 --- a/modules/transilien/backend.py +++ b/modules/transilien/backend.py @@ -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']) diff --git a/weboob/capabilities/travel.py b/weboob/capabilities/travel.py index 8f8cd756..9893dc81 100644 --- a/weboob/capabilities/travel.py +++ b/weboob/capabilities/travel.py @@ -107,7 +107,7 @@ class ICapTravel(IBaseCap): """ raise NotImplementedError() - def iter_station_departures(self, station_id, arrival_id=None): + def iter_station_departures(self, station_id, arrival_id=None, date=None): """ Iterate on departures. @@ -115,6 +115,8 @@ class ICapTravel(IBaseCap): :type station_id: str :param arrival_id: optionnal arrival station ID :type arrival_id: str + :param date: optional date + :type date: datetime.datetime :rtype: iter[:class:`Departure`] """ raise NotImplementedError()