Source Code for Module arvados.safeapi

 1  # Copyright (C) The Arvados Authors. All rights reserved. 
 2  # 
 3  # SPDX-License-Identifier: Apache-2.0 
 4   
 5  from __future__ import absolute_import 
 6   
 7  from builtins import object 
 8  import copy 
 9  import threading 
10   
11  import arvados 
12  import arvados.keep as keep 
13  import arvados.config as config 
14   
15 -class ThreadSafeApiCache(object): 
16      """Threadsafe wrapper for API objects. 
17   
18      This stores and returns a different api object per thread, because httplib2 
19      which underlies apiclient is not threadsafe. 
20   
21      """ 
22   
23 -    def __init__(self, apiconfig=None, keep_params={}, api_params={}): 
24          if apiconfig is None: 
25              apiconfig = config.settings() 
26          self.apiconfig = copy.copy(apiconfig) 
27          self.api_params = api_params 
28          self.local = threading.local() 
29   
30          # Initialize an API object for this thread before creating 
31          # KeepClient, this will report if ARVADOS_API_HOST or 
32          # ARVADOS_API_TOKEN are missing. 
33          self.localapi() 
34   
35          self.keep = keep.KeepClient(api_client=self, **keep_params) 
36   
37 -    def localapi(self): 
38          if 'api' not in self.local.__dict__: 
39              self.local.api = arvados.api_from_config('v1', apiconfig=self.apiconfig, 
40                                                       **self.api_params) 
41          return self.local.api 
42   
43 -    def __getattr__(self, name): 
44          # Proxy nonexistent attributes to the thread-local API client. 
45          if name == "api_token": 
46              return self.apiconfig['ARVADOS_API_TOKEN'] 
47          return getattr(self.localapi(), name) 
48