Source code for woob.capabilities.travel
# Copyright(C) 2010-2011 Romain Bignon, Julien Hebert
#
# This file is part of woob.
#
# woob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# woob is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with woob. If not, see <http://www.gnu.org/licenses/>.
import datetime
from .base import BaseObject, Capability, DecimalField, StringField, UserError
from .date import DateField, DeltaField, TimeField
__all__ = ["Station", "Departure", "RoadStep", "RoadmapError", "RoadmapFilters", "CapTravel"]
[docs]class Station(BaseObject):
"""
Describes a station.
"""
name = StringField("Name of station")
def __init__(self, id=None, name=None, url=None):
super().__init__(id, url)
self.name = name
def __repr__(self):
return f"<Station id={self.id!r} name={self.name!r}>"
[docs]class Departure(BaseObject):
"""
Describes a departure.
"""
type = StringField("Type of train")
time = TimeField("Departure time")
departure_station = StringField("Departure station")
arrival_station = StringField("Destination of the train")
arrival_time = TimeField("Arrival time")
late = TimeField("Optional late", default=datetime.time())
information = StringField("Informations")
plateform = StringField("Where the train will leave")
price = DecimalField("Price of ticket")
currency = StringField("Currency", default=None)
def __init__(self, id=None, _type=None, _time=None, url=None):
super().__init__(id, url)
self.type = _type
self.time = _time
def __repr__(self):
return "<Departure id={!r} type={!r} time={!r} departure={!r} arrival={!r}>".format(
self.id,
self.type,
self.time.strftime("%H:%M"),
self.departure_station,
self.arrival_station,
)
[docs]class RoadStep(BaseObject):
"""
A step on a roadmap.
"""
line = StringField("When line")
start_time = TimeField("Start of step")
end_time = TimeField("End of step")
departure = StringField("Departure station")
arrival = StringField("Arrival station")
duration = DeltaField("Duration of this step")
[docs]class RoadmapError(UserError):
"""
Raised when the roadmap is unable to be calculated.
"""
[docs]class RoadmapFilters(BaseObject):
"""
Filters to get a roadmap.
"""
departure_time = DateField("Wanted departure time")
arrival_time = DateField("Wanted arrival time")
def __init__(self, id="", url=None):
super().__init__(id, url)
[docs]class CapTravel(Capability):
"""
Travel websites.
"""
[docs] def iter_station_search(self, pattern):
"""
Iterates on search results of stations.
:param pattern: the search pattern
:type pattern: str
:rtype: iter[:class:`Station`]
"""
raise NotImplementedError()
[docs] def iter_station_departures(self, station_id, arrival_id=None, date=None):
"""
Iterate on departures.
:param station_id: the station ID
: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()
[docs] def iter_roadmap(self, departure, arrival, filters):
"""
Get a roadmap.
:param departure: name of departure station
:type departure: str
:param arrival: name of arrival station
:type arrival: str
:param filters: filters on search
:type filters: :class:`RoadmapFilters`
:rtype: iter[:class:`RoadStep`]
"""
raise NotImplementedError()