Starting python exception handling cleanup
Reviewed By: aditya
git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@665013 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/lib/py/src/transport/TSocket.py b/lib/py/src/transport/TSocket.py
index c8f9e36..ee429cd 100644
--- a/lib/py/src/transport/TSocket.py
+++ b/lib/py/src/transport/TSocket.py
@@ -22,8 +22,11 @@
self.handle.settimeout(ms/1000.00)
def open(self):
- self.handle = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
- self.handle.connect((self.host, self.port))
+ try:
+ self.handle = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+ self.handle.connect((self.host, self.port))
+ except socket.error, e:
+ raise TTransportException(TTransportException.NOT_OPEN, e.message)
def close(self):
if self.handle != None:
@@ -42,7 +45,7 @@
while sent < have:
plus = self.handle.send(buff)
if plus == 0:
- raise TTransportException('sent 0 bytes')
+ raise TTransportException('TSocket sent 0 bytes')
sent += plus
buff = buff[plus:]