Merge pull request #49 from pshv/master

Added retries parameter to avoid requests' connection drops

Author: Dzmitry Humianiuk

reportportal_client/service.py

@@ -19,6 +19,8 @@
 import uuid
 import logging
 
+from requests.adapters import HTTPAdapter
+
 from .errors import ResponseError, EntryCreatedError, OperationCompletionError
 
 logger = logging.getLogger(__name__)
@@ -103,7 +105,7 @@ class ReportPortalService(object):
     """Service class with report portal event callbacks."""
 
     def __init__(self, endpoint, project, token, api_base="api/v1",
-                 is_skipped_an_issue=True, verify_ssl=True):
+                 is_skipped_an_issue=True, verify_ssl=True, retries=None):
         """Init the service class.
 
         Args:
@@ -126,6 +128,9 @@ def __init__(self, endpoint, project, token, api_base="api/v1",
                                  self.project)
 
         self.session = requests.Session()
+        if retries:
+            self.session.mount('https://', HTTPAdapter(max_retries=retries))
+            self.session.mount('http://', HTTPAdapter(max_retries=retries))
         self.session.headers["Authorization"] = "bearer {0}".format(self.token)
         self.stack = [None]
         self.launch_id = None

reportportal_client/service_async.py

@@ -134,7 +134,7 @@ class ReportPortalServiceAsync(object):
     def __init__(self, endpoint, project, token, api_base="api/v1",
                  error_handler=None, log_batch_size=20,
                  is_skipped_an_issue=True,
-                 verify_ssl=True, queue_get_timeout=5):
+                 verify_ssl=True, queue_get_timeout=5, retries=None):
         """Init the service class.
 
         Args:
@@ -155,7 +155,8 @@ def __init__(self, endpoint, project, token, api_base="api/v1",
             endpoint, project, token,
             api_base,
             is_skipped_an_issue,
-            verify_ssl)
+            verify_ssl,
+            retries)
         self.log_batch = []
         self.supported_methods = ["start_launch", "finish_launch",
                                   "start_test_item", "finish_test_item", "log"]