capo-keyspacesstreams 0.2.0__tar.gz → 0.3.0__tar.gz

{capo_keyspacesstreams-0.2.0/src/capo_keyspacesstreams.egg-info → capo_keyspacesstreams-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: capo-keyspacesstreams
-Version: 0.2.0
+Version: 0.3.0
 Summary: Python SDK for AWS Keyspacesstreams.
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers

{capo_keyspacesstreams-0.2.0 → capo_keyspacesstreams-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "capo-keyspacesstreams"
-version = "0.2.0"
+version = "0.3.0"
 description = "Python SDK for AWS Keyspacesstreams."
 readme = "README.md"
 requires-python = ">=3.10"

{capo_keyspacesstreams-0.2.0 → capo_keyspacesstreams-0.3.0}/src/capo_keyspacesstreams/_operations/keyspaces_streams/get_records.py

@@ -28,6 +28,7 @@ from capo_keyspacesstreams._rule_engine._endpoint_rule_set import (
 from capo_keyspacesstreams._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_keyspacesstreams.errors import UnknownServiceError
 
@@ -147,7 +148,7 @@ def get_records(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -164,7 +165,7 @@ async def async_get_records(
     try:
         if response.status >= 300:
             await response.aread()
-            handle_error(response)
+            raise_error(response, handle_error)
         return await async_handle_response(response), response
     except BaseException:
         await response.aclose()

{capo_keyspacesstreams-0.2.0 → capo_keyspacesstreams-0.3.0}/src/capo_keyspacesstreams/_operations/keyspaces_streams/get_shard_iterator.py

@@ -27,6 +27,7 @@ from capo_keyspacesstreams._rule_engine._endpoint_rule_set import (
 from capo_keyspacesstreams._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_keyspacesstreams.errors import UnknownServiceError
 
@@ -145,7 +146,7 @@ def get_shard_iterator(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -163,7 +164,7 @@ async def async_get_shard_iterator(
     try:
         if response.status >= 300:
             await response.aread()
-            handle_error(response)
+            raise_error(response, handle_error)
         return await async_handle_response(response), response
     except BaseException:
         await response.aclose()

{capo_keyspacesstreams-0.2.0 → capo_keyspacesstreams-0.3.0}/src/capo_keyspacesstreams/_operations/keyspaces_streams/get_stream.py

@@ -31,6 +31,7 @@ from capo_keyspacesstreams._rule_engine._endpoint_rule_set import (
 from capo_keyspacesstreams._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_keyspacesstreams.errors import UnknownServiceError
 
@@ -150,7 +151,7 @@ def get_stream(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -167,7 +168,7 @@ async def async_get_stream(
     try:
         if response.status >= 300:
             await response.aread()
-            handle_error(response)
+            raise_error(response, handle_error)
         return await async_handle_response(response), response
     except BaseException:
         await response.aclose()

{capo_keyspacesstreams-0.2.0 → capo_keyspacesstreams-0.3.0}/src/capo_keyspacesstreams/_operations/keyspaces_streams/list_streams.py

@@ -27,6 +27,7 @@ from capo_keyspacesstreams._rule_engine._endpoint_rule_set import (
 from capo_keyspacesstreams._services._pipeline import (
     AsyncOperationOptions,
     OperationOptions,
+    raise_error,
 )
 from capo_keyspacesstreams.errors import UnknownServiceError
 
@@ -146,7 +147,7 @@ def list_streams(
     try:
         if response.status >= 300:
             response.read()
-            handle_error(response)
+            raise_error(response, handle_error)
         return handle_response(response), response
     except BaseException:
         response.close()
@@ -163,7 +164,7 @@ async def async_list_streams(
     try:
         if response.status >= 300:
             await response.aread()
-            handle_error(response)
+            raise_error(response, handle_error)
         return await async_handle_response(response), response
     except BaseException:
         await response.aclose()

{capo_keyspacesstreams-0.2.0 → capo_keyspacesstreams-0.3.0}/src/capo_keyspacesstreams/_services/_pipeline.py

@@ -6,6 +6,7 @@ from collections.abc import AsyncIterator, Iterator, Sequence
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Awaitable, Callable, Generic, TypeVar, cast
 
+from typing_extensions import Never
 from zapros import (
     AsyncClient,
     Client,
@@ -133,6 +134,22 @@ async def aexecute_pipeline(
     return await make_chain(0)(request)
 
 
+def raise_error(response: Response, handle_error: Callable[[Response], Never]) -> Never:
+    try:
+        handle_error(response)
+    except ServiceError as exc:
+        # The response itself can say a retry is safe, even when the error is
+        # not marked @retryable: a 429, a transient 5xx or a Retry-After header.
+        if response.status == 429:
+            exc.is_throttling_error = True
+        if (
+            response.status in (429, 500, 502, 503, 504)
+            or "retry-after" in response.headers
+        ):
+            exc.is_retryable = True
+        raise
+
+
 def _is_retryable(exc: Exception) -> bool:
     if isinstance(exc, ServiceError):
         return exc.is_retryable

{capo_keyspacesstreams-0.2.0 → capo_keyspacesstreams-0.3.0/src/capo_keyspacesstreams.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: capo-keyspacesstreams
-Version: 0.2.0
+Version: 0.3.0
 Summary: Python SDK for AWS Keyspacesstreams.
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Developers