support repositories to manage backends (closes #747)
This commit is contained in:
Romain Bignon
parent
ef16a5b726
commit
14a7a1d362
410 changed files with 1079 additions and 297 deletions
23
modules/canaltp/__init__.py
Normal file
23
modules/canaltp/__init__.py
Normal file
|
|
@ -0,0 +1,23 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright(C) 2010-2011 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/>.
|
||||
|
||||
|
||||
from .backend import CanalTPBackend
|
||||
|
||||
__all__ = ['CanalTPBackend']
|
||||
49
modules/canaltp/backend.py
Normal file
49
modules/canaltp/backend.py
Normal file
|
|
@ -0,0 +1,49 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright(C) 2010-2011 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/>.
|
||||
|
||||
from weboob.capabilities.travel import ICapTravel, Station, Departure
|
||||
from weboob.tools.backend import BaseBackend
|
||||
|
||||
from .browser import CanalTP
|
||||
|
||||
|
||||
__all__ = ['CanalTPBackend']
|
||||
|
||||
|
||||
class CanalTPBackend(BaseBackend, ICapTravel):
|
||||
NAME = 'canaltp'
|
||||
MAINTAINER = 'Romain Bignon'
|
||||
EMAIL = 'romain@weboob.org'
|
||||
VERSION = '0.a'
|
||||
LICENSE = 'AGPLv3+'
|
||||
DESCRIPTION = "French trains"
|
||||
BROWSER = CanalTP
|
||||
|
||||
def iter_station_search(self, pattern):
|
||||
for _id, name in self.browser.iter_station_search(pattern):
|
||||
yield Station(_id, name)
|
||||
|
||||
def iter_station_departures(self, station_id, arrival_id=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']
|
||||
departure.arrival_station = d['arrival']
|
||||
departure.late = d['late']
|
||||
departure.information = d['late_reason']
|
||||
yield departure
|
||||
76
modules/canaltp/browser.py
Normal file
76
modules/canaltp/browser.py
Normal file
|
|
@ -0,0 +1,76 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright(C) 2010-2011 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/>.
|
||||
|
||||
|
||||
from datetime import datetime, date, time
|
||||
|
||||
from weboob.tools.browser import BaseBrowser
|
||||
from weboob.tools.misc import to_unicode
|
||||
from weboob.tools.browser import BrokenPageError
|
||||
|
||||
|
||||
__all__ = ['CanalTP']
|
||||
|
||||
|
||||
class CanalTP(BaseBrowser):
|
||||
DOMAIN = 'widget.canaltp.fr'
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
BaseBrowser.__init__(self, '', **kwargs)
|
||||
|
||||
def iter_station_search(self, pattern):
|
||||
url = u'http://widget.canaltp.fr/Prochains_departs_15122009/dev/gare.php?txtrech=%s' % unicode(pattern)
|
||||
result = self.openurl(url.encode('utf-8')).read()
|
||||
for station in result.split('&'):
|
||||
try:
|
||||
_id, name = station.split('=')
|
||||
except ValueError:
|
||||
continue
|
||||
else:
|
||||
yield _id, to_unicode(name)
|
||||
|
||||
def iter_station_departures(self, station_id, arrival_id=None):
|
||||
url = u'http://widget.canaltp.fr/Prochains_departs_15122009/dev/index.php?gare=%s' % unicode(station_id)
|
||||
result = self.openurl(url.encode('utf-8')).read()
|
||||
result = result
|
||||
departure = ''
|
||||
for line in result.split('&'):
|
||||
if not '=' in line:
|
||||
raise BrokenPageError('Unable to parse result: %s' % line)
|
||||
key, value = line.split('=', 1)
|
||||
if key == 'nomgare':
|
||||
departure = value
|
||||
elif key.startswith('ligne'):
|
||||
_type, unknown, _time, arrival, served, late, late_reason = value.split(';', 6)
|
||||
yield {'type': to_unicode(_type),
|
||||
'time': datetime.combine(date.today(), time(*[int(x) for x in _time.split(':')])),
|
||||
'departure': to_unicode(departure),
|
||||
'arrival': to_unicode(arrival).strip(),
|
||||
'late': late and time(0, int(late.split()[0])) or time(),
|
||||
'late_reason': to_unicode(late_reason).replace('\n', '').strip()}
|
||||
|
||||
def home(self):
|
||||
pass
|
||||
|
||||
def login(self):
|
||||
pass
|
||||
|
||||
def is_logged(self):
|
||||
""" Do not need to be logged """
|
||||
return True
|
||||
BIN
modules/canaltp/favicon.png
Normal file
BIN
modules/canaltp/favicon.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 2.4 KiB |
30
modules/canaltp/test.py
Normal file
30
modules/canaltp/test.py
Normal file
|
|
@ -0,0 +1,30 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
|
||||
# Copyright(C) 2010-2011 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/>.
|
||||
|
||||
|
||||
from weboob.tools.test import BackendTest
|
||||
|
||||
class CanalTPTest(BackendTest):
|
||||
BACKEND = 'canaltp'
|
||||
|
||||
def test_canaltp(self):
|
||||
stations = list(self.backend.iter_station_search('defense'))
|
||||
self.assertTrue(len(stations) > 0)
|
||||
|
||||
list(self.backend.iter_station_departures(stations[0].id))
|
||||
Loading…
Add table
Add a link
Reference in a new issue