added argument to client constructor to set request timeout

Author: LiamNorman

sailthru/sailthru_client.py

@@ -55,10 +55,11 @@ class SailthruClient(object):
         client = SailthruClient(api_key, api_secret)
     """
 
-    def __init__(self, api_key, secret, api_url=None):
+    def __init__(self, api_key, secret, api_url=None, request_timeout=None):
         self.api_key = api_key
         self.secret = secret
         self.api_url = api_url if api_url else 'https://api.sailthru.com'
+        self.request_timeout = request_timeout
         self.last_rate_limit_info = {}
 
     def send(self, template, email, _vars=None, options=None, schedule_time=None, limit=None):
@@ -762,7 +763,7 @@ def _api_request(self, action, data, request_type, headers=None):
     def _http_request(self, action, data, method, file_data=None, headers=None):
         url = self.api_url + '/' + action
         file_data = file_data or {}
-        response = sailthru_http_request(url, data, method, file_data, headers)
+        response = sailthru_http_request(url, data, method, file_data, headers, self.request_timeout)
         if (action in self.last_rate_limit_info):
             self.last_rate_limit_info[action][method] = response.get_rate_limit_headers()
         else:

sailthru/sailthru_http.py

@@ -27,21 +27,22 @@ def flatten(hash_table, brackets=True):
         return f
     return flatten(hash_table, False)
 
-def sailthru_http_request(url, data, method, file_data=None, headers=None):
+def sailthru_http_request(url, data, method, file_data=None, headers=None, request_timeout=None):
     """
     Perform an HTTP GET / POST / DELETE request
     """
     data = flatten_nested_hash(data)
     method = method.upper()
     params, data = (None, data) if method == 'POST' else (data, None)
     sailthru_headers = {'User-Agent': 'Sailthru API Python Client %s; Python Version: %s' % ('2.3.4', platform.python_version())}
+    request_timeout = request_timeout if request_timeout is not None else 10
     if headers and isinstance(headers, dict):
         for key, value in sailthru_headers.items():
             headers[key] = value
     else:
         headers = sailthru_headers
     try:
-        response = requests.request(method, url, params=params, data=data, files=file_data, headers=headers, timeout=10)
+        response = requests.request(method, url, params=params, data=data, files=file_data, headers=headers, timeout=request_timeout)
         return SailthruResponse(response)
     except requests.HTTPError as e:
         raise SailthruClientError(str(e))