Added headers parameter to InfluxDBClient

nathanielatom · web-flow · commit 94f4addf3e5d · 2019-05-02T21:53:30.000-04:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -61,6 +61,9 @@ class InfluxDBClient(object):
     :type proxies: dict
     :param path: path of InfluxDB on the server to connect, defaults to ''
     :type path: str
+    :param headers: headers to add to Requests, will add 'Content-Type'
+        and 'Accept' unless these are already present, defaults to {}
+    :type headers: dict
     """
 
     def __init__(self,
@@ -78,6 +81,7 @@ def __init__(self,
                  proxies=None,
                  pool_size=10,
                  path='',
+                 headers=None,
                  ):
         """Construct a new InfluxDBClient object."""
         self.__host = host
@@ -126,10 +130,11 @@ def __init__(self,
             self._port,
             self._path)
 
-        self._headers = {
-            'Content-Type': 'application/json',
-            'Accept': 'text/plain'
-        }
+        if headers is None:
+            headers = {}
+        headers.setdefault('Content-Type', 'application/json')
+        headers.setdefault('Accept', 'text/plain')
+        self._headers = headers
 
     @property
     def _baseurl(self):
@@ -303,7 +308,7 @@ def write(self, data, params=None, expected_response_code=204,
         :returns: True, if the write operation is successful
         :rtype: bool
         """
-        headers = self._headers
+        headers = self._headers.copy()
         headers['Content-Type'] = 'application/octet-stream'
 
         if params:
