a little more defensive

2009-01-09 17:59:14 -05:00 · 2009-01-09 17:59:14 -05:00 · e7f5c2ec2a
commit e7f5c2ec2a
parent 3495cd482f
1 changed files with 6 additions and 2 deletions
--- a/mongo.py
+++ b/mongo.py
@ -75,8 +75,12 @@ class Mongo(object):

        to_send += data

-        # TODO be more robust
-        sent = self.__connection.send(to_send)
+        total_sent = 0
+        while total_sent < len(to_send):
+            sent = self.__connection.send(to_send[total_sent:])
+            if sent == 0:
+                raise ConnectionException("connection closed")
+            total_sent += sent

    def __cmp__(self, other):
        if isinstance(other, Mongo):