Fixed bug in TCPTransport.send

Author: Sam Clements

riemann_client/__init__.py

@@ -1,4 +1,4 @@
 """A Python Riemann client and command line tool"""
 
-__version__ = '3.0.0'
+__version__ = '3.0.1'
 __author__ = 'Sam Clements <sam.clements@datasift.com>'

riemann_client/transport.py

@@ -9,6 +9,14 @@
 import riemann_client.riemann_pb2
 
 
+def socket_recvall(socket, length, bufsize=4096):
+    """Recives bytes from a socket until the buffer is the requested length"""
+    data = ""
+    while len(data) < length:
+        data += socket.recv(bufsize)
+    return data
+
+
 class RiemannError(Exception):
     """Error class for errors recived from the Riemann server"""
     pass
@@ -71,16 +79,9 @@ def send(self, message):
 
         length = struct.unpack('!I', self.socket.recv(4))[0]
         response = riemann_client.riemann_pb2.Msg()
-        response.ParseFromString(self.socket_recvall(socket, length))
+        response.ParseFromString(socket_recvall(self.socket, length))
 
         if not response.ok:
             raise RiemannError(response.error)
 
         return response
-
-    @staticmethod
-    def socket_recvall(socket, length, bufsize=4096):
-        data = ""
-        while len(data) < length:
-            data += socket.recv(bufsize)
-        return data

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name = "riemann-client",
-    version = '3.0.0',
+    version = '3.0.1',
 
     author = "Sam Clements",
     author_email = "sam.clements@datasift.com",