+# Copyright (C) The Arvados Authors. All rights reserved.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from __future__ import absolute_import
+
+from builtins import object
+import copy
import threading
-import api
-import keep
-import config
-
-class SafeApi(object):
- """Threadsafe wrapper for API object. This stores and returns a different api
- object per thread, because httplib2 which underlies apiclient is not
- threadsafe.
+
+import arvados
+import arvados.keep as keep
+import arvados.config as config
+
+class ThreadSafeApiCache(object):
+ """Threadsafe wrapper for API objects.
+
+ This stores and returns a different api object per thread, because httplib2
+ which underlies apiclient is not threadsafe.
+
"""
def __init__(self, apiconfig=None, keep_params={}):
- if not apiconfig:
- apiconfig = config
- self.host = apiconfig.get('ARVADOS_API_HOST')
- self.api_token = apiconfig.get('ARVADOS_API_TOKEN')
- self.insecure = apiconfig.flag_is_true('ARVADOS_API_HOST_INSECURE')
+ if apiconfig is None:
+ apiconfig = config.settings()
+ self.apiconfig = copy.copy(apiconfig)
self.local = threading.local()
self.keep = keep.KeepClient(api_client=self, **keep_params)
def localapi(self):
if 'api' not in self.local.__dict__:
- self.local.api = api.api('v1', False, self.host,
- self.api_token, self.insecure)
+ self.local.api = arvados.api_from_config('v1', apiconfig=self.apiconfig)
return self.local.api
def __getattr__(self, name):
# Proxy nonexistent attributes to the thread-local API client.
- try:
- return getattr(self.localapi(), name)
- except AttributeError:
- return super(SafeApi, self).__getattr__(name)
+ if name == "api_token":
+ return self.apiconfig['ARVADOS_API_TOKEN']
+ return getattr(self.localapi(), name)