From fa392e91adae6c03adcfd1f7c03c0729dbd15b25 Mon Sep 17 00:00:00 2001 From: Roger Meier Date: Wed, 11 Apr 2012 22:15:15 +0000 Subject: [PATCH] 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 --- lib/py/src/transport/THttpClient.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/lib/py/src/transport/THttpClient.py b/lib/py/src/transport/THttpClient.py index ad94d112..293d9304 100644 --- 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 -- 2.17.1