THRIFT-1560 Add setCustomHeaders() method to python thrift lib

Patch: Dave Watson

git-svn-id: https://svn.apache.org/repos/asf/thrift/trunk@1325041 13f79535-47bb-0310-9956-ffa450edef68

diff --git a/lib/py/src/transport/THttpClient.py b/lib/py/src/transport/THttpClient.py
index ad94d11..293d930 100644 (file)
--- a/lib/py/src/transport/THttpClient.py
+++ b/lib/py/src/transport/THttpClient.py
@@ -63,6 +63,7 @@ class THttpClient(TTransportBase):
     self.__wbuf = StringIO()
     self.__http = None
     self.__timeout = None
+    self.__custom_headers = None
 
   def open(self):
     if self.scheme == 'http':
@@ -86,6 +87,9 @@ class THttpClient(TTransportBase):
     else:
       self.__timeout = ms / 1000.0
 
+  def setCustomHeaders(self, headers):
+    self.__custom_headers = headers
+
   def read(self, sz):
     return self.__http.file.read(sz)
 
@@ -117,6 +121,11 @@ class THttpClient(TTransportBase):
     self.__http.putheader('Host', self.host)
     self.__http.putheader('Content-Type', 'application/x-thrift')
     self.__http.putheader('Content-Length', str(len(data)))
+
+    if self.__custom_headers:
+        for key, val in self.__custom_headers.iteritems():
+            self.__http.putheader(key, val)
+
     self.__http.endheaders()
 
     # Write payload