@@ -24,16 +24,15 @@ class Client(object):
24
24
requestSession = None
25
25
bearerToken = None
26
26
27
-
28
- def __init__ (self , apiKey = None , apiAddress = None , caCert = None , bearerToken = None ):
27
+ def __init__ (self , apiKey = None , apiAddress = None , caCert = None , bearerToken = None ):
29
28
# Override apiKey with environment variable
30
29
config = None
31
30
self .requestSession = requests .Session ()
32
31
if apiKey is None and 'ALGORITHMIA_API_KEY' in os .environ :
33
32
apiKey = os .environ ['ALGORITHMIA_API_KEY' ]
34
33
elif bearerToken is None and 'ALGORITHMIA_BEARER_TOKEN' in os .environ :
35
34
bearerToken = os .environ ['ALGORITHMIA_BEARER_TOKEN' ]
36
-
35
+
37
36
self .bearerToken = bearerToken
38
37
self .apiKey = apiKey
39
38
if apiAddress is not None :
@@ -224,8 +223,10 @@ def postJsonHelper(self, url, input_object, parse_response_as_json=True, **query
224
223
headers = {}
225
224
if self .apiKey is not None :
226
225
headers ['Authorization' ] = self .apiKey
227
- else :
228
- headers ['Authorization' ] = "Bearer " + self .bearerToken
226
+ elif self .bearerToken is not None :
227
+ headers ['Authorization' ] = 'Bearer ' + self .bearerToken
228
+ else :
229
+ raise Exception ("No authentication provided" )
229
230
230
231
input_json = None
231
232
if input_object is None :
@@ -253,8 +254,8 @@ def getHelper(self, url, **query_parameters):
253
254
headers = {}
254
255
if self .apiKey is not None :
255
256
headers ['Authorization' ] = self .apiKey
256
- elif self .bearerToken is not None :
257
- headers ['Authorization' ] = 'Bearer ' + self .bearerToken
257
+ elif self .bearerToken is not None :
258
+ headers ['Authorization' ] = 'Bearer ' + self .bearerToken
258
259
else :
259
260
raise Exception ("No authentication provided" )
260
261
return self .requestSession .get (self .apiAddress + url , headers = headers , params = query_parameters )
@@ -263,8 +264,8 @@ def getStreamHelper(self, url, **query_parameters):
263
264
headers = {}
264
265
if self .apiKey is not None :
265
266
headers ['Authorization' ] = self .apiKey
266
- elif self .bearerToken is not None :
267
- headers ['Authorization' ] = 'Bearer ' + self .bearerToken
267
+ elif self .bearerToken is not None :
268
+ headers ['Authorization' ] = 'Bearer ' + self .bearerToken
268
269
else :
269
270
raise Exception ("No authentication provided" )
270
271
return self .requestSession .get (self .apiAddress + url , headers = headers , params = query_parameters , stream = True )
@@ -273,8 +274,8 @@ def patchHelper(self, url, params):
273
274
headers = {'content-type' : 'application/json' }
274
275
if self .apiKey is not None :
275
276
headers ['Authorization' ] = self .apiKey
276
- elif self .bearerToken is not None :
277
- headers ['Authorization' ] = 'Bearer ' + self .bearerToken
277
+ elif self .bearerToken is not None :
278
+ headers ['Authorization' ] = 'Bearer ' + self .bearerToken
278
279
else :
279
280
raise Exception ("No authentication provided" )
280
281
return self .requestSession .patch (self .apiAddress + url , headers = headers , data = json .dumps (params ))
@@ -284,8 +285,8 @@ def headHelper(self, url):
284
285
headers = {}
285
286
if self .apiKey is not None :
286
287
headers ['Authorization' ] = self .apiKey
287
- elif self .bearerToken is not None :
288
- headers ['Authorization' ] = 'Bearer ' + self .bearerToken
288
+ elif self .bearerToken is not None :
289
+ headers ['Authorization' ] = 'Bearer ' + self .bearerToken
289
290
else :
290
291
raise Exception ("No authentication provided" )
291
292
return self .requestSession .head (self .apiAddress + url , headers = headers )
@@ -295,8 +296,8 @@ def putHelper(self, url, data):
295
296
headers = {}
296
297
if self .apiKey is not None :
297
298
headers ['Authorization' ] = self .apiKey
298
- elif self .bearerToken is not None :
299
- headers ['Authorization' ] = 'Bearer ' + self .bearerToken
299
+ elif self .bearerToken is not None :
300
+ headers ['Authorization' ] = 'Bearer ' + self .bearerToken
300
301
else :
301
302
raise Exception ("No authentication provided" )
302
303
if isJson (data ):
@@ -312,8 +313,8 @@ def deleteHelper(self, url):
312
313
headers = {}
313
314
if self .apiKey is not None :
314
315
headers ['Authorization' ] = self .apiKey
315
- elif self .bearerToken is not None :
316
- headers ['Authorization' ] = 'Bearer ' + self .bearerToken
316
+ elif self .bearerToken is not None :
317
+ headers ['Authorization' ] = 'Bearer ' + self .bearerToken
317
318
else :
318
319
raise Exception ("No authentication provided" )
319
320
response = self .requestSession .delete (self .apiAddress + url , headers = headers )
@@ -375,11 +376,12 @@ def freeze(self, manifest_path, manifest_output_dir="."):
375
376
required_files [i ]['md5_checksum' ] = md5_checksum
376
377
lock_md5_checksum = md5_for_str (str (manifest_file ))
377
378
manifest_file ['lock_checksum' ] = lock_md5_checksum
378
- with open (manifest_output_dir + '/' + 'model_manifest.json.freeze' , 'w' ) as f :
379
+ with open (manifest_output_dir + '/' + 'model_manifest.json.freeze' , 'w' ) as f :
379
380
json .dump (manifest_file , f )
380
381
else :
381
382
print ("Expected to find a model_manifest.json file, none was discovered in working directory" )
382
383
384
+
383
385
def isJson (myjson ):
384
386
try :
385
387
json_object = json .loads (myjson )
0 commit comments