Merge branch 'master' of https://github.com/alaptseu/python-graphql-client into alaptseu-master

Resolve minor conflicts with updates from alaptseu PR prisma-labs#10

Author: bruce

.gitignore

@@ -1,3 +1,5 @@
 .DS_Store
 *.egg-info
 *.pyc
+*.idea
+example2_run.py

README.md

@@ -43,6 +43,13 @@ An alternative http header name for the token can be set by passing in the alter
 client.inject_token('very-long-and-secure-token','x-api-key')
 ```
 
+If you need more options for headers use
+
+```py
+client.inject_headers({'your_custom_header_name' : 'your_custom_header_value' ,
+                       'your_custom_header_name_2' :'your_custom_header_value_2'})
+```
+
 ## License
 
 [MIT License](http://opensource.org/licenses/MIT)

graphqlclient/client.py

@@ -6,6 +6,7 @@ def __init__(self, endpoint):
         self.endpoint = endpoint
         self.token = None
         self.headername = None
+        self.headersDict = None
 
     def execute(self, query, variables=None):
         return self._send(query, variables)
@@ -14,6 +15,9 @@ def inject_token(self, token, headername='Authorization'):
         self.token = token
         self.headername = headername
 
+    def inject_headers(self, headersDict):
+        self.headersDict = headersDict
+
     def _send(self, query, variables):
         data = {'query': query,
                 'variables': variables}
@@ -22,6 +26,8 @@ def _send(self, query, variables):
 
         if self.token is not None:
             headers[self.headername] = '{}'.format(self.token)
+        if self.headersDict is not None:
+            headers.update(self.headersDict)
 
         req = urllib.request.Request(self.endpoint, json.dumps(data).encode('utf-8'), headers)