add module VoyagesSNCF (CapTravel)

modules/voyagessncf/pages.py

@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2013      Romain Bignon
+#
+# 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 <http://www.gnu.org/licenses/>.
+
+
+import re
+from decimal import Decimal
+from datetime import time, datetime
+
+from weboob.tools.browser import BasePage
+from weboob.tools.json import json
+from weboob.tools.mech import ClientForm
+from weboob.capabilities.base import UserError, Currency
+
+
+__all__ = ['CitiesPage', 'SearchPage']
+
+
+class CitiesPage(BasePage):
+    def get_stations(self):
+       result = json.loads(self.document[self.document.find('{'):-2])
+       return result['CITIES']
+
+class SearchPage(BasePage):
+    def search(self, departure, arrival, date):
+        self.browser.select_form(name='saisie')
+        self.browser['ORIGIN_CITY'] = departure.encode(self.browser.ENCODING)
+        self.browser['DESTINATION_CITY'] = arrival.encode(self.browser.ENCODING)
+
+        if date is None:
+            date = datetime.now()
+
+        self.browser['OUTWARD_DATE'] = date.strftime('%d/%m/%y')
+        self.browser['OUTWARD_TIME'] = [str(date.hour + 1)]
+        self.browser['PASSENGER_1'] = ['ADULT']
+        self.browser.controls.append(ClientForm.TextControl('text', 'nbAnimalsForTravel', {'value': ''}))
+        self.browser['nbAnimalsForTravel'] = '0'
+        self.browser.submit()
+
+class SearchErrorPage(BasePage):
+    def on_loaded(self):
+        p = self.document.getroot().cssselect('div.messagesError p')
+        if len(p) > 0:
+            message = p[0].text.strip()
+            raise UserError(message)
+
+class SearchInProgressPage(BasePage):
+    def on_loaded(self):
+        link = self.document.xpath('//a[@id="url_redirect_proposals"]')[0]
+        self.browser.location(link.attrib['href'])
+
+class ResultsPage(BasePage):
+    def get_value(self, div, name):
+        p = div.cssselect(name)[0]
+        sub = p.find('p')
+        if sub is not None:
+            txt = sub.tail.strip()
+            if txt == '':
+                p.remove(sub)
+            else:
+                return unicode(txt)
+
+        return unicode(self.parser.tocleanstring(p))
+
+    def parse_hour(self, div, name):
+        txt = self.get_value(div, name)
+        hour, minute = map(int, txt.split('h'))
+        return time(hour, minute)
+
+    def iter_results(self):
+        for div in self.document.getroot().cssselect('div.train_info'):
+            price = None
+            currency = None
+            for td in div.cssselect('td.price'):
+                txt = self.parser.tocleanstring(td)
+                p = Decimal(re.sub('([^\d\.]+)', '', txt))
+                currency = Currency.get_currency(txt)
+                if price is None or p < price:
+                    price = p
+
+            yield {'type': self.get_value(div, 'div.transporteur-txt'),
+                   'time': self.parse_hour(div, 'div.departure div.hour'),
+                   'departure': self.get_value(div, 'div.departure div.station'),
+                   'arrival': self.get_value(div, 'div.arrival div.station'),
+                   'arrival_time': self.parse_hour(div, 'div.arrival div.hour'),
+                   'price': price,
+                   'currency': currency,
+                  }