@@ -28,24 +28,28 @@ class ApiClient(object):
28
28
headerValue: a header value to pass when making calls to the API
29
29
"""
30
30
def __init__ (self , host = None , headerName = None , headerValue = None ):
31
- self .headerName = headerName
32
- self .headerValue = headerValue
31
+ self .defaultHeaders = {}
32
+ if (headerName is not None ):
33
+ self .defaultHeaders [headerName ] = headerValue
33
34
self .host = host
34
35
self .cookie = None
35
36
self .boundary = '' .join (random .choice (string .ascii_letters + string .digits ) for _ in range (30 ))
36
37
38
+ def setDefaultHeader (self , headerName , headerValue ):
39
+ self .defaultHeaders [headerName ] = headerValue
40
+
37
41
def callAPI (self , resourcePath , method , queryParams , postData ,
38
42
headerParams = None , files = None ):
39
43
40
44
url = self .host + resourcePath
45
+
46
+ mergedHeaderParams = self .defaultHeaders .copy ()
47
+ mergedHeaderParams .update (headerParams )
41
48
headers = {}
42
- if headerParams :
43
- for param , value in headerParams .iteritems ():
49
+ if mergedHeaderParams :
50
+ for param , value in mergedHeaderParams .iteritems ():
44
51
headers [param ] = ApiClient .sanitizeForSerialization (value )
45
52
46
- if self .headerName :
47
- headers [self .headerName ] = ApiClient .sanitizeForSerialization (self .headerValue )
48
-
49
53
if self .cookie :
50
54
headers ['Cookie' ] = ApiClient .sanitizeForSerialization (self .cookie )
51
55
0 commit comments