Added a try/except to retry sending the file is the connection was reset by peer

1 files changed, 9 insertions, 3 deletions

diff --git a/S3.py b/S3.py
index 2456732..a8c6f1f 100644
--- a/S3.py
+++ b/S3.py
@@ -272,7 +272,13 @@ class AWSAuthConnection:
             # add auth header
             self._add_aws_auth_header(final_headers, method, bucket, key, query_args)
 
-            connection.request(method, path, data, final_headers)
+            try:
+                connection.request(method, path, data, final_headers)
+            except:
+                # Sometimes the connection is reset by peer. If that happens
+                # just try it again and we'll see what happens.
+                connection.request(method, path, data, final_headers)
+
             resp = connection.getresponse()
             if resp.status < 300 or resp.status >= 400:
                 return resp
@@ -475,7 +481,7 @@ class ListBucketResponse(Response):
 class ListAllMyBucketsResponse(Response):
     def __init__(self, http_response):
         Response.__init__(self, http_response)
-        if http_response.status < 300: 
+        if http_response.status < 300:
             handler = ListAllMyBucketsHandler()
             xml.sax.parseString(self.body, handler)
             self.entries = handler.entries
@@ -501,7 +507,7 @@ class GetResponse(Response):
 class LocationResponse(Response):
     def __init__(self, http_response):
         Response.__init__(self, http_response)
-        if http_response.status < 300: 
+        if http_response.status < 300:
             handler = LocationHandler()
             xml.sax.parseString(self.body, handler)
             self.location = handler.location