browser.py 2.22 KB
Newer Older
Hervé Werner's avatar
Hervé Werner committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
# -*- 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/>.

20
from __future__ import unicode_literals
Hervé Werner's avatar
Hervé Werner committed
21

22
import datetime
23 24

from weboob.browser.browsers import APIBrowser
Hervé Werner's avatar
Hervé Werner committed
25 26 27 28 29


__all__ = ['VelibBrowser']


30
class VelibBrowser(APIBrowser):
31 32 33
    ENCODING = 'utf-8'

    API_KEY = '2282a34b49cf45d8129cdf93d88762914cece88b'
34
    BASEURL = 'https://api.jcdecaux.com/vls/v1/'
35

36
    def __init__(self, api_key, *a, **kw):
37
        super(VelibBrowser, self).__init__(*a, **kw)
38
        self.api_key = api_key or VelibBrowser.API_KEY
Hervé Werner's avatar
Hervé Werner committed
39

40
    def do_get(self, path, **query):
41
        query['apiKey'] = self.api_key
42
        return self.request(path, params=query)
Hervé Werner's avatar
Hervé Werner committed
43

44 45 46 47 48 49 50 51 52 53 54
    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
Hervé Werner's avatar
Hervé Werner committed
55 56

    def get_station_infos(self, gauge):
57 58 59 60 61 62 63 64
        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)
65 66
        jgauge['latitude'] = '%s' % jgauge['position']['lat']
        jgauge['longitude'] = '%s' % jgauge['position']['lng']
67
        del jgauge['position']
68
        return jgauge