Added urllib3 < 2.X compatibility for the Retry object

Added matrix to test if it also works with urllib3 <2.x
Flared · Oct 21, 2024 · 98b2d50 · 98b2d50
1 parent 4c583f3
commit 98b2d50
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 9 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -12,12 +12,17 @@ jobs:
           '3.11',
           '3.12',
         ]
+        pip-override: [
+          '',
+          'pip install "urllib3<2"',
+        ]
     name: Python ${{ matrix.python-version }}
     steps:
     - uses: actions/checkout@v4
     - uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
         architecture: x64
+    - run: ${{ matrix.pip-override }}
     - run: make test
     - run: make lint
diff --git a/flareio/api_client.py b/flareio/api_client.py
@@ -47,18 +47,21 @@ def _create_session() -> requests.Session:
             ),
         )
 
+        retry = Retry(
+            total=5,
+            backoff_factor=2,
+            status_forcelist=[429, 502, 503, 504],
+            allowed_methods={"GET", "POST"},
+        )
+
+        # Support for urllib3 < 2.X
+        if hasattr(Retry, "backoff_max"):
+            retry.backoff_max = 15
+
         # Enable retries
         session.mount(
             "https://",
-            HTTPAdapter(
-                max_retries=Retry(
-                    total=5,
-                    backoff_factor=2,
-                    status_forcelist=[429, 502, 503, 504],
-                    allowed_methods={"GET", "POST"},
-                    backoff_max=15,
-                )
-            ),
+            HTTPAdapter(max_retries=retry),
         )
 
         return session