Merge pull request #8 from AcrossTheCloud/master

Allow the authorization header name to vary

Author: ssshah86

README.md

@@ -32,9 +32,15 @@ print(result)
 
 Authorization tokens can be added to the request using the client's `inject_token` method:
 
-```
+```py
 client.inject_token('very-long-and-secure-token')
 ```
+which defaults to http header name 'Authorization'. An alternative http header name for the token can be set by passing in the alternative header name, e.g. for 'x-api-key':
+
+```py
+client.inject_token('very-long-and-secure-token','x-api-key')
+```
+
 
 ## License
 

graphqlclient/client.py

@@ -5,12 +5,14 @@ class GraphQLClient:
     def __init__(self, endpoint):
         self.endpoint = endpoint
         self.token = None
+        self.headername = None
 
     def execute(self, query, variables=None):
         return self._send(query, variables)
 
-    def inject_token(self, token):
+    def inject_token(self, token, headername='Authorization'):
         self.token = token
+        self.headername = headername
 
     def _send(self, query, variables):
         data = {'query': query,
@@ -19,7 +21,7 @@ def _send(self, query, variables):
                    'Content-Type': 'application/json'}
 
         if self.token is not None:
-            headers['Authorization'] = '{}'.format(self.token)
+            headers[self.headername] = '{}'.format(self.token)
 
         req = urllib.request.Request(self.endpoint, json.dumps(data).encode('utf-8'), headers)