handle connection errors during post as well (for hydrate)

DocNow · Jul 3, 2015 · 20cb45a · 20cb45a
1 parent 891bd10
commit 20cb45a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/twarc.py b/twarc.py
@@ -320,7 +320,12 @@ def get(self, *args, **kwargs):
 
     @rate_limit
     def post(self, *args, **kwargs):
-        return self.client.post(*args, **kwargs)
+        try:
+            return self.client.post(*args, **kwargs)
+        except requests.exceptions.ConnectionError as e:
+            logging.error("caught connection error %s", e)
+            self._connect()
+            return self.post(*args, **kwargs)
 
     def _connect(self):
         logging.info("creating http session")