# -*- coding: utf-8 -*-
# Copyright(C) 2010-2011 Julien Hébert, Romain Bignon
# Copyright(C) 2014 Benjamin Carton
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see .
import re
from weboob.browser.pages import JsonPage, HTMLPage
from weboob.browser.elements import TableElement, ItemElement, DictElement, method
from weboob.capabilities.travel import Station, Departure, RoadStep
from weboob.capabilities import NotAvailable
from weboob.browser.filters.standard import CleanText, Filter, Time, Env, Regexp, Duration,\
Format, Join, DateTime
from weboob.browser.filters.json import Dict
from weboob.browser.filters.html import Link, TableCell
class RoadMapDuration(Duration):
_regexp = re.compile(r'(?P\d+)')
kwargs = {'minutes': 'mn'}
class DepartureTypeFilter(Filter):
def filter(self, el):
result = []
for img in el[0].getiterator(tag='img'):
result.append(img.attrib['alt'])
return u' '.join(result)
class Child(Filter):
def filter(self, el):
return list(el[0].iterchildren())
class RoadMapPage(HTMLPage):
def request_roadmap(self, station, arrival, departure_date, arrival_date):
form = self.get_form('//form[@id="form_rechercheitineraire"]')
form['depart'] = '%s' % station.name.replace(' ', '+')
form['coordDepart'] = station._coord
form['typeDepart'] = station._type_point
form['arrivee'] = '%s' % arrival.name.replace(' ', '+')
form['coordArrivee'] = arrival._coord
form['typeArrivee'] = arrival._type_point
if departure_date:
form['jour'] = departure_date.strftime('%d/%m/%Y')
form['horaire'] = departure_date.strftime('%H:%M')
form['sens'] = 1
elif arrival_date:
form['jour'] = arrival_date.strftime('%d/%m/%Y')
form['horaire'] = arrival_date.strftime('%H:%M')
form['sens'] = -1
form.submit()
def is_ambiguous(self):
return self.doc.xpath('//span[has-class("errormsg")]')
def fix_ambiguity(self):
form = self.get_form('//form[@id="cRechercheItineraire"]')
if self.doc.xpath('//select[@id="gare_arrivee_ambigu"]'):
form['coordArrivee'] = self.doc.xpath('//select[@id="gare_arrivee_ambigu"]/option[@cat="STOP_AREA"]/@value')[0]
if self.doc.xpath('//select[@id="gare_depart_ambigu"]'):
form['coordDepart'] = self.doc.xpath('//select[@id="gare_depart_ambigu"]/option[@cat="STOP_AREA"]/@value')[0]
form.submit()
def get_roadmap(self):
roadstep = None
for step in self.doc.xpath('(//ol[@class="trajet_feuilleDeRoute transport"])[1]/li'):
if step.attrib and 'class' in step.attrib and step.attrib['class'] == 'odd':
if roadstep:
roadstep.end_time = Time(CleanText('./div/div[has-class("temps")]'))(step)
roadstep.arrival = CleanText('./div/div/div/div[@class="step_infos clearfix"]', default=None)(step)
yield roadstep
roadstep = RoadStep()
roadstep.start_time = Time(CleanText('./div/div[has-class("temps")]'))(step)
roadstep.departure = CleanText('./div/div/div/div[@class="step_infos clearfix"]', default=None)(step)
if not step.attrib:
roadstep.line = CleanText('./div/div/div/div/div/div[@class="transport"]', default=None)(step) or\
CleanText('./div/div/div/div[@class="step_infos clearfix"]', default=None)(step) or\
Join('\n', './div/div/div/div/div/ul/li/text()')(step)
roadstep.duration = RoadMapDuration(CleanText('./div/div[has-class("temps")]'))(step)
del roadstep
class HorairesPage(HTMLPage):
def get_departures(self, station, arrival, date):
for table in self.doc.xpath('//table[@class="trajet_horaires trajet_etapes"]'):
lignes = table.xpath('./tr[@class="ligne"]/th')
arrives = table.xpath('./tr[@class="arrivee"]/td')
departs = table.xpath('./tr[@class="depart"]/td')
items = zip(lignes, arrives, departs)
for item in items:
departure = Departure()
departure.id = Regexp(Link('./div/a'), '.*?vehicleJourneyExternalCode=(.*?)&.*?')(item[1])
departure.departure_station = station
departure.arrival_station = arrival
hour, minute = CleanText('./div/a')(item[1]).split('h')
departure.time = date.replace(hour=int(hour), minute=int(minute))
hour, minute = CleanText('./div/a')(item[2]).split('h')
departure.arrival_time = date.replace(hour=int(hour), minute=int(minute))
departure.information = CleanText('.')(item[0])
departure.type = DepartureTypeFilter(item)(self)
yield departure
class StationsPage(JsonPage):
@method
class get_stations(DictElement):
item_xpath = 'gares'
class item(ItemElement):
klass = Station
MapCategorieToTypePoint = {"StopArea": "STOP_AREA",
"City": "CITY",
"Site": "SITE_SEUL",
"Address": "ADRESSE"}
def condition(self):
if self.env['only_station']:
return Dict('entryPointType')(self.el) == 'StopArea' and Dict('reseau')(self.el)[0]
return True
obj_name = CleanText(Dict('gare'))
obj_id = CleanText(Dict('gare'), replace=[(' ', '-')])
obj__coord = Format('%s_%s', Dict('coordLambertX'), Dict('coordLambertY'))
def obj__type_point(self):
key = Dict('entryPointType', default=None)(self)
if key:
return self.MapCategorieToTypePoint[key]
class DeparturesPage2(HTMLPage):
def get_potential_arrivals(self):
arrivals = {}
for el in self.doc.xpath('//select[@id="gare_arrive_ambigu"]/option'):
arrivals[el.text] = el.attrib['value']
return arrivals
def get_station_id(self):
form = self.get_form('//form[@id="cfichehoraire"]')
return form['departExternalCode']
def init_departure(self, station):
form = self.get_form('//form[@id="cfichehoraire"]')
form['depart'] = station
form.submit()
def get_departures(self, arrival, date):
form = self.get_form('//form[@id="cfichehoraire"]')
form['arrive'] = arrival
if date:
form['jourHoraire'] = date.day
form['moiHoraire'] = '%s|%s' % (date.month, date.year)
form['heureHoraire'] = date.hour
form['minuteHoraire'] = date.minute
self.logger.debug(form)
form.submit()
class DeparturesPage(HTMLPage):
@method
class get_departures(TableElement):
head_xpath = u'//table[@class="etat_trafic"][1]/thead/tr/th[@scope="col"]/text()'
item_xpath = u'//table[@class="etat_trafic"]/tr'
col_type = u'Ligne'
col_info = u'Nom du train'
col_time = u'Heure de départ'
col_arrival = u'Destination'
col_plateform = u'Voie/quai'
col_id = u'Gares desservies'
class item(ItemElement):
klass = Departure
def condition(self):
return len(self.el.xpath('./td')) >= 6
obj_time = TableCell('time') & CleanText & DateTime | NotAvailable
obj_type = DepartureTypeFilter(TableCell('type'))
obj_departure_station = CleanText(Env('station'))
obj_arrival_station = CleanText(TableCell('arrival'))
obj_information = TableCell('time') & CleanText & Regexp(pattern='([^\d:]+)') | u''
obj_plateform = CleanText(TableCell('plateform'))
obj_id = Regexp(Link(Child(TableCell('id'))), '.*?numeroTrain=(.*?)&.*?')