From 2b353fa266c13b77b2c77af797dc28c8380b9443 Mon Sep 17 00:00:00 2001 From: RaNaN Date: Thu, 10 Feb 2011 23:43:10 +0100 Subject: more fixes --- module/remote/thriftbackend/Protocol.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 module/remote/thriftbackend/Protocol.py (limited to 'module/remote/thriftbackend/Protocol.py') diff --git a/module/remote/thriftbackend/Protocol.py b/module/remote/thriftbackend/Protocol.py new file mode 100644 index 000000000..c42d01459 --- /dev/null +++ b/module/remote/thriftbackend/Protocol.py @@ -0,0 +1,30 @@ +# -*- coding: utf-8 -*- + +from thrift.protocol import TBinaryProtocol + +class Protocol(TBinaryProtocol.TBinaryProtocol): + def writeString(self, str): + try: + str = str.encode("utf8", "ignore") + except Exception, e: + pass + + self.writeI32(len(str)) + self.trans.write(str) + + def readString(self): + len = self.readI32() + str = self.trans.readAll(len) + try: + str = str.decode("utf8", "ignore") + except: + pass + + return str + + +class ProtocolFactory(TBinaryProtocol.TBinaryProtocolFactory): + + def getProtocol(self, trans): + prot = Protocol(trans, self.strictRead, self.strictWrite) + return prot \ No newline at end of file -- cgit v1.2.3