X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/688319dca9cbbb27452a0062943bb38b6935254e..ff6785340ccbe4436bc0ee3b81cf084b3456a15d:/sdk/python/arvados/api.py diff --git a/sdk/python/arvados/api.py b/sdk/python/arvados/api.py index b652db77d1..315fc74a71 100644 --- a/sdk/python/arvados/api.py +++ b/sdk/python/arvados/api.py @@ -14,6 +14,7 @@ import logging import os import re import socket +import sys import time import types @@ -32,6 +33,9 @@ RETRY_DELAY_INITIAL = 2 RETRY_DELAY_BACKOFF = 2 RETRY_COUNT = 2 +if sys.version_info >= (3,): + httplib2.SSLHandshakeError = None + class OrderedJsonModel(apiclient.model.JsonModel): """Model class for JSON that preserves the contents' order. @@ -153,7 +157,7 @@ def http_cache(data_type): return cache.SafeHTTPCache(path, max_age=60*60*24*2) def api(version=None, cache=True, host=None, token=None, insecure=False, - request_id=None, **kwargs): + request_id=None, timeout=5*60, **kwargs): """Return an apiclient Resources object for an Arvados instance. :version: @@ -173,6 +177,9 @@ def api(version=None, cache=True, host=None, token=None, insecure=False, :insecure: If True, ignore SSL certificate validation errors. + :timeout: + A timeout value for http requests. + :request_id: Default X-Request-Id header value for outgoing requests that don't already provide one. If None or omitted, generate a random @@ -225,12 +232,16 @@ def api(version=None, cache=True, host=None, token=None, insecure=False, http_kwargs['disable_ssl_certificate_validation'] = True kwargs['http'] = httplib2.Http(**http_kwargs) + if kwargs['http'].timeout is None: + kwargs['http'].timeout = timeout + kwargs['http'] = _patch_http_request(kwargs['http'], token) svc = apiclient_discovery.build('arvados', version, cache_discovery=False, **kwargs) svc.api_token = token svc.insecure = insecure svc.request_id = request_id + svc.config = lambda: util.get_config_once(svc) kwargs['http'].max_request_size = svc._rootDesc.get('maxRequestSize', 0) kwargs['http'].cache = None kwargs['http']._request_id = lambda: svc.request_id or util.new_request_id()