weboob-devel/modules/velib/browser.py

66 lines
2.1 KiB
Python

# -*- coding: utf-8 -*-
# Copyright(C) 2013 dud
#
# 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 datetime
from weboob.tools.browser import BaseBrowser
__all__ = ['VelibBrowser']
class VelibBrowser(BaseBrowser):
ENCODING = 'utf-8'
API_KEY = '2282a34b49cf45d8129cdf93d88762914cece88b'
BASE_URL = 'https://api.jcdecaux.com/vls/v1/'
def __init__(self, *a, **kw):
kw['parser'] = 'json'
BaseBrowser.__init__(self, *a, **kw)
def do_get(self, path, **query):
qs = '&'.join('%s=%s' % kv for kv in query.items())
if qs:
qs = '&' + qs
url = '%s%s?apiKey=%s%s' % (self.BASE_URL, path, self.API_KEY, qs)
return self.get_document(self.openurl(url))
def get_contracts_list(self):
return self.do_get('contracts')
def get_station_list(self, contract=None):
if contract:
doc = self.do_get('stations', contract=contract)
else:
doc = self.do_get('stations')
for jgauge in doc:
self._transform(jgauge)
return doc
def get_station_infos(self, gauge):
station_id, contract = gauge.split('.', 1)
doc = self.do_get('stations/%s' % station_id, contract=contract)
return self._transform(doc)
def _transform(self, jgauge):
jgauge['id'] = '%s.%s' % (jgauge['number'], jgauge['contract_name'])
jgauge['city'] = jgauge['contract_name']
jgauge['last_update'] = datetime.datetime.fromtimestamp(jgauge['last_update'] / 1000)
return jgauge