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

support roadmaps in ICapTravel (closes #680)

Browse source
This commit is contained in:
Romain Bignon 2011-06-28 17:04:27 +02:00
commit 6fa7a49349
1 changed files with 22 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

24
weboob/capabilities/travel.py
View file

@ -18,7 +18,7 @@
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
from datetime import time, datetime
from datetime import time, datetime, timedelta
from .base import IBaseCap, CapBaseObject
@ -34,7 +34,6 @@ class Station(CapBaseObject):
def __repr__(self):
return "<Station id=%r name=%r>" % (self.id, self.name)
class Departure(CapBaseObject):
def __init__(self, id, _type, _time):
CapBaseObject.__init__(self, id)
@ -51,6 +50,17 @@ class Departure(CapBaseObject):
return u"<Departure id=%r type=%r time=%r departure=%r arrival=%r>" % (
self.id, self.type, self.time.strftime('%H:%M'), self.departure_station, self.arrival_station)
class RoadStep(CapBaseObject):
def __init__(self, id):
CapBaseObject.__init__(self, id)
self.add_field('line', basestring)
self.add_field('start_time', time)
self.add_field('end_time', time)
self.add_field('departure', unicode)
self.add_field('arrival', unicode)
self.add_field('duration', timedelta)
class ICapTravel(IBaseCap):
def iter_station_search(self, pattern):
"""
@ -70,3 +80,13 @@ class ICapTravel(IBaseCap):
@return [iter] result of Departure objects
"""
raise NotImplementedError()
def iter_roadmap(self, departure, arrival):
"""
Get a roadmap.
@param departure [str] name of departure station
@param arrival [str] name of arrival station
@return [iter(RoadStep)] steps of roadmap
"""
raise NotImplementedError()