@@ -31,11 +31,10 @@ def __init__(self, apiKey = None, apiAddress = None, caCert = None, bearerToken=
31
31
self .requestSession = requests .Session ()
32
32
if apiKey is None and 'ALGORITHMIA_API_KEY' in os .environ :
33
33
apiKey = os .environ ['ALGORITHMIA_API_KEY' ]
34
- if apiKey is None :
35
- if bearerToken is None and 'ALGORITHMIA_BEARER_TOKEN' in os .environ :
36
- bearerToken = os .environ ['ALGORITHMIA_BEARER_TOKEN' ]
37
- self .bearerToken = bearerToken
38
-
34
+ elif bearerToken is None and 'ALGORITHMIA_BEARER_TOKEN' in os .environ :
35
+ bearerToken = os .environ ['ALGORITHMIA_BEARER_TOKEN' ]
36
+
37
+ self .bearerToken = bearerToken
39
38
self .apiKey = apiKey
40
39
if apiAddress is not None :
41
40
self .apiAddress = apiAddress
@@ -254,42 +253,52 @@ def getHelper(self, url, **query_parameters):
254
253
headers = {}
255
254
if self .apiKey is not None :
256
255
headers ['Authorization' ] = self .apiKey
257
- else :
256
+ elif self . bearerToken is not None :
258
257
headers ['Authorization' ] = 'Bearer ' + self .bearerToken
258
+ else :
259
+ raise Exception ("No authentication provided" )
259
260
return self .requestSession .get (self .apiAddress + url , headers = headers , params = query_parameters )
260
261
261
262
def getStreamHelper (self , url , ** query_parameters ):
262
263
headers = {}
263
264
if self .apiKey is not None :
264
265
headers ['Authorization' ] = self .apiKey
265
- else :
266
+ elif self . bearerToken is not None :
266
267
headers ['Authorization' ] = 'Bearer ' + self .bearerToken
268
+ else :
269
+ raise Exception ("No authentication provided" )
267
270
return self .requestSession .get (self .apiAddress + url , headers = headers , params = query_parameters , stream = True )
268
271
269
272
def patchHelper (self , url , params ):
270
273
headers = {'content-type' : 'application/json' }
271
274
if self .apiKey is not None :
272
275
headers ['Authorization' ] = self .apiKey
273
- else :
276
+ elif self . bearerToken is not None :
274
277
headers ['Authorization' ] = 'Bearer ' + self .bearerToken
278
+ else :
279
+ raise Exception ("No authentication provided" )
275
280
return self .requestSession .patch (self .apiAddress + url , headers = headers , data = json .dumps (params ))
276
281
277
282
# Used internally to get http head result
278
283
def headHelper (self , url ):
279
284
headers = {}
280
285
if self .apiKey is not None :
281
286
headers ['Authorization' ] = self .apiKey
282
- else :
287
+ elif self . bearerToken is not None :
283
288
headers ['Authorization' ] = 'Bearer ' + self .bearerToken
289
+ else :
290
+ raise Exception ("No authentication provided" )
284
291
return self .requestSession .head (self .apiAddress + url , headers = headers )
285
292
286
293
# Used internally to http put a file
287
294
def putHelper (self , url , data ):
288
295
headers = {}
289
296
if self .apiKey is not None :
290
297
headers ['Authorization' ] = self .apiKey
291
- else :
298
+ elif self . bearerToken is not None :
292
299
headers ['Authorization' ] = 'Bearer ' + self .bearerToken
300
+ else :
301
+ raise Exception ("No authentication provided" )
293
302
if isJson (data ):
294
303
headers ['Content-Type' ] = 'application/json'
295
304
@@ -303,8 +312,10 @@ def deleteHelper(self, url):
303
312
headers = {}
304
313
if self .apiKey is not None :
305
314
headers ['Authorization' ] = self .apiKey
306
- else :
315
+ elif self . bearerToken is not None :
307
316
headers ['Authorization' ] = 'Bearer ' + self .bearerToken
317
+ else :
318
+ raise Exception ("No authentication provided" )
308
319
response = self .requestSession .delete (self .apiAddress + url , headers = headers )
309
320
if response .reason == "No Content" :
310
321
return response
0 commit comments