httpx-tenacity 0.1.0__tar.gz → 0.2.0__tar.gz

{httpx_tenacity-0.1.0 → httpx_tenacity-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: httpx-tenacity
-Version: 0.1.0
+Version: 0.2.0
 Summary: A lightweight package that provides tenacious (retrying) httpx transports.
 Project-URL: Homepage, https://github.com/Midnighter/httpx-tenacity
 Project-URL: Documentation, https://midnighter.github.io/httpx-tenacity/

{httpx_tenacity-0.1.0 → httpx_tenacity-0.2.0}/pyproject.toml

@@ -131,6 +131,9 @@ case-sensitive = true
 known-first-party = ["src", "httpx_tenacity"]
 lines-after-imports = 2
 
+[tool.ruff.lint.pylint]
+max-args = 6
+
 [tool.pydoclint]
 style = "google"
 arg-type-hints-in-docstring = false
@@ -142,6 +145,9 @@ exclude = "_version.py"
 # Hatch Environments
 ################################################################################
 
+[tool.hatch.env]
+requires = ["hatch-pip-compile"]
+
 [tool.hatch.envs.default]
 description = """
 Base environment with default configuration values.
@@ -149,7 +155,17 @@ Base environment with default configuration values.
 All environments inheriting from the default also inherit the configuration values set
 here.
 """
-installer = "uv"
+type = "pip-compile"
+lock-filename = "requirements/requirements-{env_name}.txt"
+pip-compile-constraint = "dev"
+pip-compile-hashes = true
+pip-compile-resolver = "uv"
+pip-compile-installer = "uv"
+
+[tool.hatch.envs.no-install]
+description = """A base environment for use cases that do not install the project."""
+skip-install = true
+features = []
 
 [tool.hatch.envs.dev]
 description = """Development environment."""
@@ -161,9 +177,8 @@ PYTHONASYNCIODEBUG = "1"
 
 [tool.hatch.envs.style]
 description = """Check the style of the codebase."""
+template = "no-install"
 dependency-groups = ["style"]
-detached = true
-installer = "uv"
 
 [tool.hatch.envs.style.scripts]
 docstrings = ["pydoclint src", "pydoclint tests"]
@@ -197,9 +212,8 @@ deploy = "mike deploy {args}"
 
 [tool.hatch.envs.distribution]
 description = """Test the source and wheel distribution of the package."""
+template = "no-install"
 dependencies = ["pip", "twine"]
-detached = true
-installer = "uv"
 builder = true
 
 [tool.hatch.envs.distribution.scripts]

{httpx_tenacity-0.1.0 → httpx_tenacity-0.2.0}/src/httpx_tenacity/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.0'
-__version_tuple__ = version_tuple = (0, 1, 0)
+__version__ = version = '0.2.0'
+__version_tuple__ = version_tuple = (0, 2, 0)
 
 __commit_id__ = commit_id = None

{httpx_tenacity-0.1.0 → httpx_tenacity-0.2.0}/src/httpx_tenacity/async_tenacious_transport.py

@@ -39,7 +39,6 @@ if TYPE_CHECKING:  # pragma: no cover
 
     from .types import HTTPXHTTPTransportKeywordArguments
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -71,6 +70,7 @@ class AsyncTenaciousTransport(httpx.AsyncBaseTransport):
         max_wait_seconds: float | timedelta = 60,
         min_wait_seconds: float | timedelta = 0.02,
         exponent_base: float = 2,
+        transport: httpx.AsyncBaseTransport | None = None,
         **kwargs: Unpack[HTTPXHTTPTransportKeywordArguments],
     ) -> AsyncTenaciousTransport:
         """
@@ -90,6 +90,8 @@ class AsyncTenaciousTransport(httpx.AsyncBaseTransport):
             max_wait_seconds: Maximum wait time between retries.
             min_wait_seconds: Minimum wait time between retries.
             exponent_base: The base for the exponential backoff.
+            transport: Pass an asynchronous transport instance to handle requests
+                instead of the default one.
             **kwargs: Additional keyword arguments are used in the construction of an
                 `httpx.AsyncHTTPTransport`.
 
@@ -99,7 +101,8 @@ class AsyncTenaciousTransport(httpx.AsyncBaseTransport):
         """
         return cls(
             retry=tenacity.AsyncRetrying(
-                retry=tenacity.retry_if_result(is_server_side_issue),
+                retry=tenacity.retry_if_exception_type(httpx.RequestError)
+                | tenacity.retry_if_result(is_server_side_issue),
                 stop=tenacity.stop_after_attempt(max_attempts),
                 wait=smart_wait(
                     multiplier=multiplier,
@@ -107,10 +110,12 @@ class AsyncTenaciousTransport(httpx.AsyncBaseTransport):
                     min=min_wait_seconds,
                     exp_base=exponent_base,
                 ),
-                before_sleep=tenacity.before_sleep_log(logger, logging.DEBUG),
+                before_sleep=tenacity.before_sleep_log(logger, logging.DEBUG),  # type: ignore[invalid-argument-type]
                 reraise=True,
             ),
-            transport=httpx.AsyncHTTPTransport(**kwargs),
+            transport=httpx.AsyncHTTPTransport(**kwargs)
+            if transport is None
+            else transport,
         )
 
     async def handle_async_request(

{httpx_tenacity-0.1.0 → httpx_tenacity-0.2.0}/src/httpx_tenacity/tenacious_transport.py

@@ -39,7 +39,6 @@ if TYPE_CHECKING:  # pragma: no cover
 
     from .types import HTTPXHTTPTransportKeywordArguments
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -71,6 +70,7 @@ class TenaciousTransport(httpx.BaseTransport):
         max_wait_seconds: float | timedelta = 60,
         min_wait_seconds: float | timedelta = 0.02,
         exponent_base: float = 2,
+        transport: httpx.BaseTransport | None = None,
         **kwargs: Unpack[HTTPXHTTPTransportKeywordArguments],
     ) -> TenaciousTransport:
         """
@@ -90,6 +90,8 @@ class TenaciousTransport(httpx.BaseTransport):
             max_wait_seconds: Maximum wait time between retries.
             min_wait_seconds: Minimum wait time between retries.
             exponent_base: The base for the exponential backoff.
+            transport: Pass a transport instance to handle requests instead of the
+                default one.
             **kwargs: Additional keyword arguments are used in the construction of an
                 `httpx.HTTPTransport`.
 
@@ -99,7 +101,8 @@ class TenaciousTransport(httpx.BaseTransport):
         """
         return cls(
             retry=tenacity.Retrying(
-                retry=tenacity.retry_if_result(is_server_side_issue),
+                retry=tenacity.retry_if_exception_type(httpx.RequestError)
+                | tenacity.retry_if_result(is_server_side_issue),
                 stop=tenacity.stop_after_attempt(max_attempts),
                 wait=smart_wait(
                     multiplier=multiplier,
@@ -107,10 +110,10 @@ class TenaciousTransport(httpx.BaseTransport):
                     min=min_wait_seconds,
                     exp_base=exponent_base,
                 ),
-                before_sleep=tenacity.before_sleep_log(logger, logging.DEBUG),
+                before_sleep=tenacity.before_sleep_log(logger, logging.DEBUG),  # type: ignore[invalid-argument-type]
                 reraise=True,
             ),
-            transport=httpx.HTTPTransport(**kwargs),
+            transport=httpx.HTTPTransport(**kwargs) if transport is None else transport,
         )
 
     def handle_request(

httpx_tenacity-0.2.0/src/httpx_tenacity/wait.py

@@ -0,0 +1,85 @@
+# Copyright (c) 2026 Moritz E. Beber
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may not
+# use this file except in compliance with the License. You may obtain a copy of
+# the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations under
+# the License.
+
+
+"""Provide custom wait functions."""
+
+from datetime import timedelta
+from typing import TYPE_CHECKING
+
+import httpx
+import tenacity
+from tenacity.wait import wait_random_exponential
+
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
+
+# We follow tenacity's convention here to use lower snake-case class names for the wait
+# strategies.
+class smart_wait(wait_random_exponential):  # noqa: N801
+    """
+    Define a wait strategy that smartly backs-off.
+
+    This strategy first checks HTTP responses for an expected status code (by
+    default 429) that indicates that the server is busy. If the status code
+    matches, it checks the response header (by default  'Retry-After') for a
+    wait time indication. Otherwise, it uses random exponential backoff.
+
+    """
+
+    def __init__(
+        self,
+        multiplier: float = 1,
+        max: timedelta | float = 60,  # noqa: A002
+        exp_base: float = 2,
+        min: timedelta | float = 0,  # noqa: A002
+        status_code: int | tuple[int, ...] = httpx.codes.TOO_MANY_REQUESTS,
+        header: str = "Retry-After",
+    ) -> None:
+        super().__init__(
+            multiplier=multiplier,
+            max=max,
+            exp_base=exp_base,
+            min=min,
+        )
+        self._status_code = status_code
+        self._check: Callable[[int], bool]
+        if isinstance(self._status_code, tuple):
+            self._check = lambda code: code in self._status_code  # type: ignore[operator]
+        else:
+            self._check = lambda code: code == self._status_code
+        self._header = header
+
+    def __call__(self, retry_state: tenacity.RetryCallState) -> float:
+        """
+        Return a wait time in seconds based on response headers or retry state.
+
+        When we have a successful outcome, i.e., we received an HTTP response
+        without encountering an exception, and the HTTP status code of the
+        response matches the specified status code, check if the response header
+        indicates a wait time; otherwise use the random exponential strategy to
+        determine the wait time.
+
+        """
+        if retry_state.outcome and not retry_state.outcome.failed:
+            response: httpx.Response = retry_state.outcome.result()
+            if (
+                self._check(response.status_code)
+                and (wait := response.headers.get(self._header)) is not None
+            ):
+                return float(wait)
+
+        return super().__call__(retry_state=retry_state)

httpx_tenacity-0.1.0/src/httpx_tenacity/wait.py

@@ -1,47 +0,0 @@
-# Copyright (c) 2026 Moritz E. Beber
-#
-# Licensed under the Apache License, Version 2.0 (the "License"); you may not
-# use this file except in compliance with the License. You may obtain a copy of
-# the License at
-#
-#     https://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
-# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
-# License for the specific language governing permissions and limitations under
-# the License.
-
-
-"""Provide custom wait functions."""
-
-import httpx
-import tenacity
-from tenacity.wait import wait_random_exponential
-
-
-# We follow tenacity's convention here to use lower snake-case class names for the wait
-# strategies.
-class smart_wait(wait_random_exponential):  # noqa: N801
-    """
-    Wait strategy that smartly backs-off.
-
-    This strategy first checks the Retry-After response header for a wait time
-    indication and uses random exponential backoff if that is not the case.
-
-    """
-
-    def __call__(self, retry_state: tenacity.RetryCallState) -> float:  # noqa: D102
-        # When we have a successful outcome, i.e., we received an HTTP response without
-        # encountering a Python exception, check if the response headers indicate a wait
-        # time; otherwise use the random exponential strategy to determine the wait
-        # time.
-        if retry_state.outcome and not retry_state.outcome.failed:
-            response: httpx.Response = retry_state.outcome.result()
-            if (
-                response.status_code == httpx.codes.TOO_MANY_REQUESTS
-                and (wait := response.headers.get("Retry-After")) is not None
-            ):
-                return float(wait)
-
-        return super().__call__(retry_state=retry_state)