| # Copyright (c) 2006- Facebook |
| # Distributed under the Thrift Software License |
| # |
| # See accompanying file LICENSE or visit the Thrift site at: |
| # http://developers.facebook.com/thrift/ |
| |
| from TTransport import * |
| from cStringIO import StringIO |
| |
| import urlparse |
| import httplib |
| import warnings |
| |
| class THttpClient(TTransportBase): |
| |
| """Http implementation of TTransport base.""" |
| |
| def __init__(self, uri_or_host, port=None, path=None): |
| """THttpClient supports two different types constructor parameters. |
| |
| THttpClient(host, port, path) - deprecated |
| THttpClient(uri) |
| |
| Only the second supports https.""" |
| |
| if port is not None: |
| warnings.warn("Please use the THttpClient('http://host:port/path') syntax", DeprecationWarning, stacklevel=2) |
| self.host = uri_or_host |
| self.port = port |
| assert path |
| self.path = path |
| self.scheme = 'http' |
| else: |
| parsed = urlparse.urlparse(uri_or_host) |
| self.scheme = parsed.scheme |
| assert self.scheme in ('http', 'https') |
| if self.scheme == 'http': |
| self.port = parsed.port or httplib.HTTP_PORT |
| elif self.scheme == 'https': |
| self.port = parsed.port or httplib.HTTPS_PORT |
| self.host = parsed.hostname |
| self.path = parsed.path |
| self.__wbuf = StringIO() |
| self.__http = None |
| |
| def open(self): |
| if self.scheme == 'http': |
| self.__http = httplib.HTTP(self.host, self.port) |
| else: |
| self.__http = httplib.HTTPS(self.host, self.port) |
| |
| def close(self): |
| self.__http.close() |
| self.__http = None |
| |
| def isOpen(self): |
| return self.__http != None |
| |
| def read(self, sz): |
| return self.__http.file.read(sz) |
| |
| def write(self, buf): |
| self.__wbuf.write(buf) |
| |
| def flush(self): |
| if self.isOpen(): |
| self.close() |
| self.open(); |
| |
| # Pull data out of buffer |
| data = self.__wbuf.getvalue() |
| self.__wbuf = StringIO() |
| |
| # HTTP request |
| self.__http.putrequest('POST', self.path) |
| |
| # Write headers |
| self.__http.putheader('Host', self.host) |
| self.__http.putheader('Content-Type', 'application/x-thrift') |
| self.__http.putheader('Content-Length', str(len(data))) |
| self.__http.endheaders() |
| |
| # Write payload |
| self.__http.send(data) |
| |
| # Get reply to flush the request |
| self.code, self.message, self.headers = self.__http.getreply() |