diff options
-rw-r--r-- | cdsapi/__init__.py | 5 | ||||
-rw-r--r-- | cdsapi/api.py | 11 | ||||
-rw-r--r-- | tests/test_api.py | 2 |
3 files changed, 14 insertions, 4 deletions
diff --git a/cdsapi/__init__.py b/cdsapi/__init__.py index e7cdf1b..c9e9d4e 100644 --- a/cdsapi/__init__.py +++ b/cdsapi/__init__.py @@ -16,7 +16,8 @@ # granted to it by virtue of its status as an intergovernmental organisation nor # does it submit to any jurisdiction. +from __future__ import absolute_import, division, print_function, unicode_literals -import cdsapi.api +from . import api -Client = cdsapi.api.Client +Client = api.Client diff --git a/cdsapi/api.py b/cdsapi/api.py index 0ada51b..a53ebf0 100644 --- a/cdsapi/api.py +++ b/cdsapi/api.py @@ -6,12 +6,15 @@ # granted to it by virtue of its status as an intergovernmental organisation nor # does it submit to any jurisdiction. -import requests +from __future__ import absolute_import, division, print_function, unicode_literals + import json import time import os import logging +import requests + def bytes_to_string(n): u = ['', 'K', 'M', 'G', 'T', 'P'] @@ -45,8 +48,12 @@ class Client(object): config = {} with open(dotrc) as f: for l in f.readlines(): + try: k, v = l.strip().split(':', 1) - config[k] = v.strip() + if k in ['url', 'key']: + config[k] = v.strip() + except: + print('The file ~/.cdsapirc is badly formatted (remove extra lines not required)') url = config.get('url') key = config.get('key') diff --git a/tests/test_api.py b/tests/test_api.py index f76a842..2403dca 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -1,4 +1,6 @@ +from __future__ import absolute_import, division, print_function, unicode_literals + from cdsapi import api |