Cp 7117/add retries to api calls

README.md

@@ -28,7 +28,11 @@ pip3 install quandl
 |---|---|---|
 | api_key | Your access key | `tEsTkEy123456789` | Used to identify who you are and provide full access. |
 | api_version | The API version you wish to use | 2015-04-09 | Can be used to test your code against the latest version without committing to it. |
-
+| use_retries | Whether API calls which return statuses in `RETRY_STATUS_CODES` should be automatically retried | True
+| number_of_retries | Maximum number of retries that should be attempted. Only used if `use_retries` is True | 8
+| retry_backoff_factor | Determines the amount of time in seconds that should be waited before attempting another retry. Note that this factor is exponential so a `retry_backoff_factor` of 0.1 will cause waits of [0.1, 0.2, 0.4, 0.8, etc]. Only used if `use_retries` is True | 0.1
+| RETRY_STATUS_CODES | A list of HTTP status codes which will trigger a retry to occur. Only used if `use_retries` is True| [500, 501, 502]
+
 ```python
 import quandl
 quandl.ApiConfig.api_key = 'tEsTkEy123456789'

quandl/api_config.py

@@ -1,9 +1,16 @@
 class ApiConfig:
     api_key = None
-    api_base = 'https://www.quandl.com/api/v3'
+    api_protocol = 'https://'
+    api_base = '{}www.quandl.com/api/v3'.format(api_protocol)
     api_version = None
     page_limit = 100
 
+    use_retries = True
+    number_of_retries = 5
+    retry_backoff_factor = 0.4
+    max_wait_between_retries = 8
+    RETRY_STATUS_CODES = list(range(500, 512))
+
 
 def save_key(apikey, filename=None):
     if filename is None:

quandl/connection.py

@@ -1,6 +1,8 @@
 import re
 
 import requests
+from urllib3.util.retry import Retry
+from requests.adapters import HTTPAdapter
 
 from .util import Util
 from .version import VERSION
@@ -37,9 +39,10 @@ def request(cls, http_verb, url, **options):
 
     @classmethod
     def execute_request(cls, http_verb, url, **options):
+        session = cls.get_session()
+
         try:
-            func = getattr(requests, http_verb)
-            response = func(url, **options)
+            response = session.request(method=http_verb, url=url, **options)
             if response.status_code < 200 or response.status_code >= 300:
                 cls.handle_api_error(response)
             else:
@@ -49,6 +52,30 @@ def execute_request(cls, http_verb, url, **options):
                 cls.handle_api_error(e.response)
             raise e
 
+    @classmethod
+    def get_session(cls):
+        session = requests.Session()
+        adapter = HTTPAdapter(max_retries=cls.get_retries())
+        session.mount(ApiConfig.api_protocol, adapter)
+
+        return session
+
+    @classmethod
+    def get_retries(cls):
+        if not ApiConfig.use_retries:
+            return Retry(total=0)
+
+        Retry.BACKOFF_MAX = ApiConfig.max_wait_between_retries
+        retries = Retry(total=ApiConfig.number_of_retries,
+                        connect=ApiConfig.number_of_retries,
+                        read=ApiConfig.number_of_retries,
+                        status_forcelist=ApiConfig.RETRY_STATUS_CODES,
+                        backoff_factor=ApiConfig.retry_backoff_factor,
+                        raise_on_status=False,
+                        respect_retry_after_header=False)
+
+        return retries
+
     @classmethod
     def parse(cls, response):
         try:

test/test_connection.py

@@ -4,17 +4,18 @@
     QuandlError, LimitExceededError, InternalServerError,
     AuthenticationError, ForbiddenError, InvalidRequestError,
     NotFoundError, ServiceUnavailableError)
-import unittest
+from test.test_retries import ModifyRetrySettingsTestCase
 from test.helpers.httpretty_extension import httpretty
 import json
 from mock import patch, call
 from quandl.version import VERSION
 
 
-class ConnectionTest(unittest.TestCase):
+class ConnectionTest(ModifyRetrySettingsTestCase):
 
     @httpretty.activate
-    def test_quandl_exceptions(self):
+    def test_quandl_exceptions_no_retries(self):
+        ApiConfig.use_retries = False
         quandl_errors = [('QELx04', 429, LimitExceededError),
                          ('QEMx01', 500, InternalServerError),
                          ('QEAx01', 400, AuthenticationError),
@@ -38,6 +39,7 @@ def test_quandl_exceptions(self):
 
     @httpretty.activate
     def test_parse_error(self):
+        ApiConfig.retry_backoff_factor = 0
         httpretty.register_uri(httpretty.GET,
                                "https://www.quandl.com/api/v3/databases",
                                body="not json", status=500)
@@ -46,6 +48,7 @@ def test_parse_error(self):
 
     @httpretty.activate
     def test_non_quandl_error(self):
+        ApiConfig.retry_backoff_factor = 0
         httpretty.register_uri(httpretty.GET,
                                "https://www.quandl.com/api/v3/databases",
                                body=json.dumps(

test/test_database.py

@@ -14,6 +14,7 @@
 from quandl.api_config import ApiConfig
 from quandl.model.database import Database
 from quandl.connection import Connection
+from test.test_retries import ModifyRetrySettingsTestCase
 from mock import patch, call, mock_open
 from test.factories.database import DatabaseFactory
 from test.factories.meta import MetaFactory
@@ -110,10 +111,9 @@ def test_databases_has_more(self):
         self.assertTrue(results.has_more_results())
 
 
-class BulkDownloadDatabaseTest(unittest.TestCase):
+class BulkDownloadDatabaseTest(ModifyRetrySettingsTestCase):
 
-    @classmethod
-    def setUpClass(cls):
+    def setUp(self):
         httpretty.enable()
         httpretty.register_uri(httpretty.GET,
                                re.compile(
@@ -125,17 +125,15 @@ def setUpClass(cls):
         httpretty.register_uri(httpretty.GET,
                                re.compile('https://www.blah.com/'), body='{}')
 
-    @classmethod
-    def tearDownClass(cls):
-        httpretty.disable()
-        httpretty.reset()
-
-    def setUp(self):
         database = {'database': DatabaseFactory.build(database_code='NSE')}
         self.database = Database(database['database']['database_code'], database['database'])
         ApiConfig.api_key = 'api_token'
         ApiConfig.api_version = '2015-04-09'
 
+    def tearDown(self):
+        httpretty.disable()
+        httpretty.reset()
+
     def test_get_bulk_downnload_url_with_download_type(self):
         url = self.database.bulk_download_url(params={'download_type': 'partial'})
         parsed_url = urlparse(url)
@@ -179,12 +177,15 @@ def test_bulk_download_raises_exception_when_no_path(self):
             QuandlError, lambda: self.database.bulk_download_to_file(None))
 
     def test_bulk_download_raises_exception_when_error_response(self):
+        ApiConfig.retry_backoff_factor = 0
+        httpretty.reset()
         httpretty.register_uri(httpretty.GET,
                                re.compile(
                                    'https://www.quandl.com/api/v3/databases/*'),
                                body=json.dumps(
                                    {'quandl_error':
                                     {'code': 'QEMx01', 'message': 'something went wrong'}}),
                                status=500)
+
         self.assertRaises(
             InternalServerError, lambda: self.database.bulk_download_to_file('.'))

test/test_datatable.py

@@ -11,11 +11,12 @@
 from quandl.model.datatable import Datatable
 from mock import patch, call, mock_open
 from test.factories.datatable import DatatableFactory
+from test.test_retries import ModifyRetrySettingsTestCase
 from quandl.api_config import ApiConfig
 from quandl.errors.quandl_error import (InternalServerError, QuandlError)
 
 
-class GetDatatableDatasetTest(unittest.TestCase):
+class GetDatatableDatasetTest(ModifyRetrySettingsTestCase):
 
     @classmethod
     def setUpClass(cls):
@@ -114,12 +115,17 @@ def test_bulk_download_raises_exception_when_no_path(self):
                 QuandlError, lambda: self.datatable.download_file(None))
 
     def test_bulk_download_table_raises_exception_when_error_response(self):
+        httpretty.reset()
+        ApiConfig.number_of_retries = 2
+        error_responses = [httpretty.Response(
+            body=json.dumps({'quandl_error': {'code': 'QEMx01',
+                                              'message': 'something went wrong'}}),
+            status=500)]
+
         httpretty.register_uri(httpretty.GET,
                                re.compile(
                                    'https://www.quandl.com/api/v3/datatables/*'),
-                               body=json.dumps(
-                                   {'quandl_error':
-                                    {'code': 'QEMx01', 'message': 'something went wrong'}}),
-                               status=500)
+                               responses=error_responses)
+
         self.assertRaises(
             InternalServerError, lambda: self.datatable.download_file('.'))

test/test_retries.py

@@ -0,0 +1,112 @@
+import unittest
+import json
+
+from quandl.connection import Connection
+from quandl.api_config import ApiConfig
+from test.factories.datatable import DatatableFactory
+from test.helpers.httpretty_extension import httpretty
+from quandl.errors.quandl_error import InternalServerError
+
+
+class ModifyRetrySettingsTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.default_use_retries = ApiConfig.use_retries
+        self.default_number_of_retries = ApiConfig.number_of_retries
+        self.default_retry_backoff_factor = ApiConfig.retry_backoff_factor
+        self.default_max_wait_between_retries = ApiConfig.max_wait_between_retries
+        self.DEFAULT_RETRY_STATUS_CODES = ApiConfig.RETRY_STATUS_CODES
+
+    def tearDown(self):
+        ApiConfig.use_retries = self.default_use_retries
+        ApiConfig.number_of_retries = self.default_number_of_retries
+        ApiConfig.retry_backoff_factor = self.default_retry_backoff_factor
+        ApiConfig.max_wait_between_retries = self.default_max_wait_between_retries
+        ApiConfig.RETRY_STATUS_CODES = self.DEFAULT_RETRY_STATUS_CODES
+
+
+class TestRetries(ModifyRetrySettingsTestCase):
+
+    def setUp(self):
+        ApiConfig.use_retries = True
+        super(TestRetries, self).setUp()
+
+    @classmethod
+    def setUpClass(cls):
+        cls.datatable = {'datatable': DatatableFactory.build(
+            vendor_code='ZACKS',
+            datatable_code='FC')}
+
+        cls.error_response = httpretty.Response(
+            body=json.dumps({'quandl_error': {'code': 'QEMx01',
+                                              'message': 'something went wrong'}}),
+            status=500)
+        cls.success_response = httpretty.Response(body=json.dumps(cls.datatable), status=200)
+
+    def test_modifying_use_retries(self):
+        ApiConfig.use_retries = False
+
+        retries = Connection.get_session().get_adapter(ApiConfig.api_protocol).max_retries
+        self.assertEqual(retries.total, 0)
+
+    def test_modifying_number_of_retries(self):
+        ApiConfig.number_of_retries = 3000
+
+        retries = Connection.get_session().get_adapter(ApiConfig.api_protocol).max_retries
+
+        self.assertEqual(retries.total, ApiConfig.number_of_retries)
+        self.assertEqual(retries.connect, ApiConfig.number_of_retries)
+        self.assertEqual(retries.read, ApiConfig.number_of_retries)
+
+    def test_modifying_retry_backoff_factor(self):
+        ApiConfig.retry_backoff_factor = 3000
+
+        retries = Connection.get_session().get_adapter(ApiConfig.api_protocol).max_retries
+        self.assertEqual(retries.backoff_factor, ApiConfig.retry_backoff_factor)
+
+    def test_modifying_retry_status_codes(self):
+        ApiConfig.RETRY_STATUS_CODES = [1, 2, 3]
+
+        retries = Connection.get_session().get_adapter(ApiConfig.api_protocol).max_retries
+        self.assertEqual(retries.status_forcelist, ApiConfig.RETRY_STATUS_CODES)
+
+    def test_modifying_max_wait_between_retries(self):
+        ApiConfig.max_wait_between_retries = 3000
+
+        retries = Connection.get_session().get_adapter(ApiConfig.api_protocol).max_retries
+        self.assertEqual(retries.BACKOFF_MAX, ApiConfig.max_wait_between_retries)
+
+    @httpretty.activate
+    def test_correct_response_returned_if_retries_succeed(self):
+        ApiConfig.number_of_retries = 3
+        ApiConfig.RETRY_STATUS_CODES = [self.error_response.status]
+
+        mock_responses = [self.error_response] + [self.error_response] + [self.success_response]
+        httpretty.register_uri(httpretty.GET,
+                               "https://www.quandl.com/api/v3/databases",
+                               responses=mock_responses)
+
+        response = Connection.request('get', 'databases')
+        self.assertEqual(response.json(), self.datatable)
+        self.assertEqual(response.status_code, self.success_response.status)
+
+    @httpretty.activate
+    def test_correct_response_exception_raised_if_retries_fail(self):
+        ApiConfig.number_of_retries = 2
+        ApiConfig.RETRY_STATUS_CODES = [self.error_response.status]
+        mock_responses = [self.error_response] * 3
+        httpretty.register_uri(httpretty.GET,
+                               "https://www.quandl.com/api/v3/databases",
+                               responses=mock_responses)
+
+        self.assertRaises(InternalServerError, Connection.request, 'get', 'databases')
+
+    @httpretty.activate
+    def test_correct_response_exception_raised_for_errors_not_in_retry_status_codes(self):
+        ApiConfig.RETRY_STATUS_CODES = []
+        mock_responses = [self.error_response]
+        httpretty.register_uri(httpretty.GET,
+                               "https://www.quandl.com/api/v3/databases",
+                               responses=mock_responses)
+
+        self.assertRaises(InternalServerError, Connection.request, 'get', 'databases')