Merge pull request #482 from fomars/master

hotfix DataUploader: retry over ProtocolError
yandex · Dec 5, 2017 · 99d3eb5 · 99d3eb5
2 parents 10f1041 + ae0a4d3
commit 99d3eb5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/yandextank/plugins/DataUploader/client.py b/yandextank/plugins/DataUploader/client.py
@@ -11,6 +11,7 @@
 import logging
 
 from requests.exceptions import ConnectionError, Timeout
+from urllib3.exceptions import ProtocolError
 
 requests.packages.urllib3.disable_warnings()
 logger = logging.getLogger(__name__)  # pylint: disable=C0103
@@ -188,7 +189,7 @@ def __make_api_request(
             try:
                 response = self.__send_single_request(request, ids.next(), trace=trace)
                 return response_callback(response)
-            except (Timeout, ConnectionError):
+            except (Timeout, ConnectionError, ProtocolError):
                 logger.warn(traceback.format_exc())
                 try:
                     timeout = next(network_timeouts)