|
15 | 15 |
|
16 | 16 | import logging
|
17 | 17 | import requests
|
18 |
| -from requests.adapters import HTTPAdapter, Retry |
| 18 | +from requests.adapters import HTTPAdapter, Retry, DEFAULT_RETRIES |
19 | 19 |
|
20 | 20 | from ._local import set_current
|
21 | 21 | from .core.rp_requests import (
|
@@ -100,16 +100,16 @@ def __init__(self,
|
100 | 100 | self.step_reporter = StepReporter(self)
|
101 | 101 | self._item_stack = []
|
102 | 102 | self.mode = mode
|
103 |
| - if retries: |
104 |
| - retry_strategy = Retry( |
105 |
| - total=retries, |
106 |
| - backoff_factor=0.1, |
107 |
| - status_forcelist=[429, 500, 502, 503, 504] |
108 |
| - ) |
109 |
| - self.session.mount('https://', HTTPAdapter( |
110 |
| - max_retries=retry_strategy, pool_maxsize=max_pool_size)) |
111 |
| - self.session.mount('http://', HTTPAdapter( |
112 |
| - max_retries=retry_strategy, pool_maxsize=max_pool_size)) |
| 103 | + |
| 104 | + retry_strategy = Retry( |
| 105 | + total=retries, |
| 106 | + backoff_factor=0.1, |
| 107 | + status_forcelist=[429, 500, 502, 503, 504] |
| 108 | + ) if retries else DEFAULT_RETRIES |
| 109 | + self.session.mount('https://', HTTPAdapter( |
| 110 | + max_retries=retry_strategy, pool_maxsize=max_pool_size)) |
| 111 | + self.session.mount('http://', HTTPAdapter( |
| 112 | + max_retries=retry_strategy, pool_maxsize=max_pool_size)) |
113 | 113 | self.session.headers['Authorization'] = 'bearer {0}'.format(self.token)
|
114 | 114 |
|
115 | 115 | self._log_manager = LogManager(
|
|
0 commit comments