svn commit: r1325041 - /thrift/trunk/lib/py/src/transport/THttpClient.py

[ro...@apache.org]

Author: roger
Date: Wed Apr 11 22:15:15 2012
New Revision: 1325041

URL: http://svn.apache.org/viewvc?rev=1325041&view=rev
Log:
THRIFT-1560 Add setCustomHeaders() method to python thrift lib
Patch: Dave Watson

Modified:
    thrift/trunk/lib/py/src/transport/THttpClient.py

Modified: thrift/trunk/lib/py/src/transport/THttpClient.py
URL: http://svn.apache.org/viewvc/thrift/trunk/lib/py/src/transport/THttpClient.py?rev=1325041&r1=1325040&r2=1325041&view=diff
==============================================================================
--- thrift/trunk/lib/py/src/transport/THttpClient.py (original)
+++ thrift/trunk/lib/py/src/transport/THttpClient.py Wed Apr 11 22:15:15 2012
@@ -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