From 1fe17774eb1059bc784a5c3be46b208a2fc0a63c Mon Sep 17 00:00:00 2001 From: David Reiss Date: Sat, 31 Jan 2009 21:59:27 +0000 Subject: [PATCH] python: Add TFileObjectTransport TFileObjectTransport is a Thrift transport that wraps a Python-style file-like object. This is necessary to add methods like isOpen and readAll. git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@739637 13f79535-47bb-0310-9956-ffa450edef68 --- lib/py/src/transport/TTransport.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/lib/py/src/transport/TTransport.py b/lib/py/src/transport/TTransport.py index 67f97ba5..a83ac256 100644 --- a/lib/py/src/transport/TTransport.py +++ b/lib/py/src/transport/TTransport.py @@ -289,3 +289,25 @@ class TFramedTransport(TTransportBase, CReadableTransport): prefix += self.__rbuf.getvalue() self.__rbuf = StringIO(prefix) return self.__rbuf + + +class TFileObjectTransport(TTransportBase): + """Wraps a file-like object to make it work as a Thrift transport.""" + + def __init__(self, fileobj): + self.fileobj = fileobj + + def isOpen(self): + return True + + def close(self): + self.fileobj.close() + + def read(self, sz): + return self.fileobj.read(sz) + + def write(self, buf): + self.fileobj.write(buf) + + def flush(self): + self.fileobj.flush() -- 2.17.1