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()