rboto 0.1.0__py3-none-any.whl

rboto/__init__.py

@@ -0,0 +1,297 @@
+# Generated by rboto-codegen. DO NOT EDIT.
+from typing import TYPE_CHECKING
+
+from ._imports import import_service as _import_service
+from .config import ClientConfig
+from .interceptors import (
+    Attempt,
+    Interceptor,
+    Next,
+    OperationContext,
+    OperationMetadata,
+    OperationRequest,
+    OperationResponse,
+    operation_context,
+)
+from .logging import LogLevel, LogTraceContext, configure_logging, get_logger
+from .services import (
+    api_gateway,
+    bedrock,
+    bedrock_agent,
+    bedrock_agent_runtime,
+    bedrock_agentcore,
+    bedrock_agentcore_control,
+    bedrock_data_automation,
+    bedrock_data_automation_runtime,
+    bedrock_runtime,
+    cloudtrail,
+    cloudwatch_logs,
+    cognito_identity,
+    connecthealth,
+    dynamodb,
+    eventbridge,
+    guardduty,
+    kms,
+    lambda_,
+    lex_runtime_v2,
+    pipes,
+    polly,
+    qbusiness,
+    s3,
+    sagemaker_runtime_http2,
+    scheduler,
+    secrets_manager,
+    sesv2,
+    sfn,
+    signin,
+    sns,
+    sqs,
+    ssm,
+    sts,
+    transcribe_streaming,
+)
+from .session import Session
+from .types import UnknownEnumValue
+
+if TYPE_CHECKING:
+    from rboto_api_gateway import APIGatewayClient
+    from rboto_bedrock import BedrockClient
+    from rboto_bedrock_agent import BedrockAgentClient
+    from rboto_bedrock_agent_runtime import BedrockAgentRuntimeClient
+    from rboto_bedrock_agentcore import BedrockAgentCoreClient
+    from rboto_bedrock_agentcore_control import BedrockAgentCoreControlClient
+    from rboto_bedrock_data_automation import BedrockDataAutomationClient
+    from rboto_bedrock_data_automation_runtime import BedrockDataAutomationRuntimeClient
+    from rboto_bedrock_runtime import BedrockRuntimeClient
+    from rboto_cloudtrail import CloudTrailClient
+    from rboto_cloudwatch_logs import CloudWatchLogsClient
+    from rboto_cognito_identity import CognitoIdentityClient
+    from rboto_connecthealth import ConnectHealthClient
+    from rboto_dynamodb import DynamoDBClient
+    from rboto_eventbridge import EventBridgeClient
+    from rboto_guardduty import GuardDutyClient
+    from rboto_kms import KMSClient
+    from rboto_lambda import LambdaClient
+    from rboto_lex_runtime_v2 import LexRuntimeV2Client
+    from rboto_pipes import EventBridgePipesClient
+    from rboto_polly import PollyClient
+    from rboto_qbusiness import QBusinessClient
+    from rboto_s3 import S3Client
+    from rboto_sagemaker_runtime_http2 import SageMakerRuntimeHTTP2Client
+    from rboto_scheduler import EventBridgeSchedulerClient
+    from rboto_secrets_manager import SecretsManagerClient
+    from rboto_sesv2 import SESV2Client
+    from rboto_sfn import StepFunctionsClient
+    from rboto_signin import SignInClient
+    from rboto_sns import SNSClient
+    from rboto_sqs import SQSClient
+    from rboto_ssm import SSMClient
+    from rboto_sts import STSClient
+    from rboto_transcribe_streaming import TranscribeStreamingClient
+
+__version__ = "0.1.0"
+__all__ = [
+    "Attempt",
+    "ClientConfig",
+    "Interceptor",
+    "LogLevel",
+    "LogTraceContext",
+    "configure_logging",
+    "get_logger",
+    "OperationContext",
+    "Next",
+    "OperationMetadata",
+    "OperationRequest",
+    "OperationResponse",
+    "operation_context",
+    "Session",
+    "UnknownEnumValue",
+    "APIGatewayClient",
+    "BedrockClient",
+    "BedrockAgentClient",
+    "BedrockAgentRuntimeClient",
+    "BedrockAgentCoreClient",
+    "BedrockAgentCoreControlClient",
+    "BedrockDataAutomationClient",
+    "BedrockDataAutomationRuntimeClient",
+    "BedrockRuntimeClient",
+    "CloudTrailClient",
+    "CloudWatchLogsClient",
+    "CognitoIdentityClient",
+    "ConnectHealthClient",
+    "DynamoDBClient",
+    "EventBridgeClient",
+    "GuardDutyClient",
+    "KMSClient",
+    "LambdaClient",
+    "LexRuntimeV2Client",
+    "EventBridgePipesClient",
+    "PollyClient",
+    "QBusinessClient",
+    "S3Client",
+    "SageMakerRuntimeHTTP2Client",
+    "EventBridgeSchedulerClient",
+    "SecretsManagerClient",
+    "SESV2Client",
+    "StepFunctionsClient",
+    "SignInClient",
+    "SNSClient",
+    "SQSClient",
+    "SSMClient",
+    "STSClient",
+    "TranscribeStreamingClient",
+
+    "api_gateway",
+    "bedrock",
+    "bedrock_agent",
+    "bedrock_agent_runtime",
+    "bedrock_agentcore",
+    "bedrock_agentcore_control",
+    "bedrock_data_automation",
+    "bedrock_data_automation_runtime",
+    "bedrock_runtime",
+    "cloudtrail",
+    "cloudwatch_logs",
+    "cognito_identity",
+    "connecthealth",
+    "dynamodb",
+    "eventbridge",
+    "guardduty",
+    "kms",
+    "lambda_",
+    "lex_runtime_v2",
+    "pipes",
+    "polly",
+    "qbusiness",
+    "s3",
+    "sagemaker_runtime_http2",
+    "scheduler",
+    "secrets_manager",
+    "sesv2",
+    "sfn",
+    "signin",
+    "sns",
+    "sqs",
+    "ssm",
+    "sts",
+    "transcribe_streaming",
+]
+
+_CLIENT_MODULES = {
+    "APIGatewayClient": (
+        "rboto_api_gateway", "api-gateway",
+    ),
+    "BedrockClient": (
+        "rboto_bedrock", "bedrock",
+    ),
+    "BedrockAgentClient": (
+        "rboto_bedrock_agent", "bedrock-agent",
+    ),
+    "BedrockAgentRuntimeClient": (
+        "rboto_bedrock_agent_runtime", "bedrock-agent-runtime",
+    ),
+    "BedrockAgentCoreClient": (
+        "rboto_bedrock_agentcore", "bedrock-agentcore",
+    ),
+    "BedrockAgentCoreControlClient": (
+        "rboto_bedrock_agentcore_control", "bedrock-agentcore-control",
+    ),
+    "BedrockDataAutomationClient": (
+        "rboto_bedrock_data_automation", "bedrock-data-automation",
+    ),
+    "BedrockDataAutomationRuntimeClient": (
+        "rboto_bedrock_data_automation_runtime", "bedrock-data-automation-runtime",
+    ),
+    "BedrockRuntimeClient": (
+        "rboto_bedrock_runtime", "bedrock-runtime",
+    ),
+    "CloudTrailClient": (
+        "rboto_cloudtrail", "cloudtrail",
+    ),
+    "CloudWatchLogsClient": (
+        "rboto_cloudwatch_logs", "cloudwatch-logs",
+    ),
+    "CognitoIdentityClient": (
+        "rboto_cognito_identity", "cognito-identity",
+    ),
+    "ConnectHealthClient": (
+        "rboto_connecthealth", "connecthealth",
+    ),
+    "DynamoDBClient": (
+        "rboto_dynamodb", "dynamodb",
+    ),
+    "EventBridgeClient": (
+        "rboto_eventbridge", "eventbridge",
+    ),
+    "GuardDutyClient": (
+        "rboto_guardduty", "guardduty",
+    ),
+    "KMSClient": (
+        "rboto_kms", "kms",
+    ),
+    "LambdaClient": (
+        "rboto_lambda", "lambda",
+    ),
+    "LexRuntimeV2Client": (
+        "rboto_lex_runtime_v2", "lex-runtime-v2",
+    ),
+    "EventBridgePipesClient": (
+        "rboto_pipes", "pipes",
+    ),
+    "PollyClient": (
+        "rboto_polly", "polly",
+    ),
+    "QBusinessClient": (
+        "rboto_qbusiness", "qbusiness",
+    ),
+    "S3Client": (
+        "rboto_s3", "s3",
+    ),
+    "SageMakerRuntimeHTTP2Client": (
+        "rboto_sagemaker_runtime_http2", "sagemaker-runtime-http2",
+    ),
+    "EventBridgeSchedulerClient": (
+        "rboto_scheduler", "scheduler",
+    ),
+    "SecretsManagerClient": (
+        "rboto_secrets_manager", "secrets-manager",
+    ),
+    "SESV2Client": (
+        "rboto_sesv2", "sesv2",
+    ),
+    "StepFunctionsClient": (
+        "rboto_sfn", "sfn",
+    ),
+    "SignInClient": (
+        "rboto_signin", "signin",
+    ),
+    "SNSClient": (
+        "rboto_sns", "sns",
+    ),
+    "SQSClient": (
+        "rboto_sqs", "sqs",
+    ),
+    "SSMClient": (
+        "rboto_ssm", "ssm",
+    ),
+    "STSClient": (
+        "rboto_sts", "sts",
+    ),
+    "TranscribeStreamingClient": (
+        "rboto_transcribe_streaming", "transcribe-streaming",
+    ),
+}
+
+
+def __getattr__(name: str) -> object:
+    target = _CLIENT_MODULES.get(name)
+    if target is None:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+    package, extra = target
+    client_class = getattr(_import_service(package, extra), name)
+    globals()[name] = client_class
+    return client_class
+
+
+def __dir__() -> list[str]:
+    return sorted(set(globals()) | _CLIENT_MODULES.keys())

rboto/__init__.pyi

@@ -0,0 +1,92 @@
+# Generated by rboto-codegen. DO NOT EDIT.
+from rboto_api_gateway import APIGatewayClient as APIGatewayClient
+from rboto_bedrock import BedrockClient as BedrockClient
+from rboto_bedrock_agent import BedrockAgentClient as BedrockAgentClient
+from rboto_bedrock_agent_runtime import BedrockAgentRuntimeClient as BedrockAgentRuntimeClient
+from rboto_bedrock_agentcore import BedrockAgentCoreClient as BedrockAgentCoreClient
+from rboto_bedrock_agentcore_control import (
+    BedrockAgentCoreControlClient as BedrockAgentCoreControlClient,
+)
+from rboto_bedrock_data_automation import BedrockDataAutomationClient as BedrockDataAutomationClient
+from rboto_bedrock_data_automation_runtime import (
+    BedrockDataAutomationRuntimeClient as BedrockDataAutomationRuntimeClient,
+)
+from rboto_bedrock_runtime import BedrockRuntimeClient as BedrockRuntimeClient
+from rboto_cloudtrail import CloudTrailClient as CloudTrailClient
+from rboto_cloudwatch_logs import CloudWatchLogsClient as CloudWatchLogsClient
+from rboto_cognito_identity import CognitoIdentityClient as CognitoIdentityClient
+from rboto_connecthealth import ConnectHealthClient as ConnectHealthClient
+from rboto_dynamodb import DynamoDBClient as DynamoDBClient
+from rboto_eventbridge import EventBridgeClient as EventBridgeClient
+from rboto_guardduty import GuardDutyClient as GuardDutyClient
+from rboto_kms import KMSClient as KMSClient
+from rboto_lambda import LambdaClient as LambdaClient
+from rboto_lex_runtime_v2 import LexRuntimeV2Client as LexRuntimeV2Client
+from rboto_pipes import EventBridgePipesClient as EventBridgePipesClient
+from rboto_polly import PollyClient as PollyClient
+from rboto_qbusiness import QBusinessClient as QBusinessClient
+from rboto_s3 import S3Client as S3Client
+from rboto_sagemaker_runtime_http2 import SageMakerRuntimeHTTP2Client as SageMakerRuntimeHTTP2Client
+from rboto_scheduler import EventBridgeSchedulerClient as EventBridgeSchedulerClient
+from rboto_secrets_manager import SecretsManagerClient as SecretsManagerClient
+from rboto_sesv2 import SESV2Client as SESV2Client
+from rboto_sfn import StepFunctionsClient as StepFunctionsClient
+from rboto_signin import SignInClient as SignInClient
+from rboto_sns import SNSClient as SNSClient
+from rboto_sqs import SQSClient as SQSClient
+from rboto_ssm import SSMClient as SSMClient
+from rboto_sts import STSClient as STSClient
+from rboto_transcribe_streaming import TranscribeStreamingClient as TranscribeStreamingClient
+
+from .config import ClientConfig as ClientConfig
+from .interceptors import Attempt as Attempt
+from .interceptors import Interceptor as Interceptor
+from .interceptors import Next as Next
+from .interceptors import OperationContext as OperationContext
+from .interceptors import OperationMetadata as OperationMetadata
+from .interceptors import OperationRequest as OperationRequest
+from .interceptors import OperationResponse as OperationResponse
+from .interceptors import operation_context as operation_context
+from .logging import LogLevel as LogLevel
+from .logging import LogTraceContext as LogTraceContext
+from .logging import configure_logging as configure_logging
+from .logging import get_logger as get_logger
+from .services import api_gateway as api_gateway
+from .services import bedrock as bedrock
+from .services import bedrock_agent as bedrock_agent
+from .services import bedrock_agent_runtime as bedrock_agent_runtime
+from .services import bedrock_agentcore as bedrock_agentcore
+from .services import bedrock_agentcore_control as bedrock_agentcore_control
+from .services import bedrock_data_automation as bedrock_data_automation
+from .services import bedrock_data_automation_runtime as bedrock_data_automation_runtime
+from .services import bedrock_runtime as bedrock_runtime
+from .services import cloudtrail as cloudtrail
+from .services import cloudwatch_logs as cloudwatch_logs
+from .services import cognito_identity as cognito_identity
+from .services import connecthealth as connecthealth
+from .services import dynamodb as dynamodb
+from .services import eventbridge as eventbridge
+from .services import guardduty as guardduty
+from .services import kms as kms
+from .services import lambda_ as lambda_
+from .services import lex_runtime_v2 as lex_runtime_v2
+from .services import pipes as pipes
+from .services import polly as polly
+from .services import qbusiness as qbusiness
+from .services import s3 as s3
+from .services import sagemaker_runtime_http2 as sagemaker_runtime_http2
+from .services import scheduler as scheduler
+from .services import secrets_manager as secrets_manager
+from .services import sesv2 as sesv2
+from .services import sfn as sfn
+from .services import signin as signin
+from .services import sns as sns
+from .services import sqs as sqs
+from .services import ssm as ssm
+from .services import sts as sts
+from .services import transcribe_streaming as transcribe_streaming
+from .session import Session as Session
+from .types import UnknownEnumValue as UnknownEnumValue
+
+__version__: str
+__all__: list[str]

rboto/_imports.py

@@ -0,0 +1,15 @@
+"""Import optional service packages without hiding broken installations."""
+
+from importlib import import_module
+from types import ModuleType
+
+
+def import_service(package: str, extra: str) -> ModuleType:
+    try:
+        return import_module(package)
+    except ModuleNotFoundError as error:
+        if error.name != package:
+            raise
+        raise ImportError(
+            f'{extra} support is not installed. Install it with: pip install "rboto[{extra}]"'
+        ) from error

rboto/_pagination.py

@@ -0,0 +1,93 @@
+"""Internal iteration shared by generated, typed service paginators."""
+
+from __future__ import annotations
+
+from collections.abc import AsyncIterator, Awaitable, Callable, Mapping, Sequence, Sized
+from copy import deepcopy
+from dataclasses import dataclass
+from typing import Protocol, TypeVar, cast, runtime_checkable
+
+_T = TypeVar("_T")
+Path = tuple[str | int, ...]
+
+
+@dataclass(frozen=True, slots=True)
+class PaginationSpec:
+    input_tokens: tuple[str, ...]
+    output_tokens: tuple[tuple[Path, ...], ...]
+    more_results: Path | None = None
+
+
+@runtime_checkable
+class _NativeCursor(Protocol):
+    def to_dict(self) -> dict[str, object]: ...
+
+
+def _snapshot(value: object) -> object:
+    # Copy only continuation state. Yielded pages retain their native ownership
+    # and can be modified by the consumer without changing the next request.
+    if isinstance(value, _NativeCursor):
+        return _snapshot(value.to_dict())
+    if isinstance(value, Mapping):
+        return {key: _snapshot(item) for key, item in cast(Mapping[object, object], value).items()}
+    if isinstance(value, (list, tuple)):
+        return [_snapshot(item) for item in cast(Sequence[object], value)]
+    return deepcopy(value)
+
+
+def _present(value: object) -> bool:
+    return value is not None and not (
+        isinstance(value, (str, bytes, Mapping, Sequence)) and len(cast(Sized, value)) == 0
+    )
+
+
+def _read(value: object, path: Path) -> object:
+    for part in path:
+        if value is None:
+            return None
+        if isinstance(part, int):
+            if not isinstance(value, Sequence) or isinstance(value, (str, bytes)):
+                return None
+            try:
+                value = cast(Sequence[object], value)[part]
+            except IndexError:
+                return None
+        elif isinstance(value, Mapping):
+            value = cast(Mapping[str, object], value).get(part)
+        else:
+            value = getattr(value, part, None)
+    return value
+
+
+def _token(page: object, alternatives: tuple[Path, ...]) -> object:
+    for path in alternatives:
+        value = _read(page, path)
+        if _present(value):
+            return value
+    return None
+
+
+async def paginate(
+    call: Callable[..., Awaitable[_T]],
+    params: dict[str, object],
+    spec: PaginationSpec,
+) -> AsyncIterator[_T]:
+    # Generated signatures validate the public arguments. Model-checked token
+    # paths are the only arguments this loop updates across requests.
+    request = params.copy()
+    previous = tuple(_snapshot(request.get(name)) for name in spec.input_tokens)
+    while True:
+        page = await call(**request)
+        following = tuple(_snapshot(_token(page, paths)) for paths in spec.output_tokens)
+        has_more = spec.more_results is None or bool(_read(page, spec.more_results))
+        finished = not has_more or not any(_present(token) for token in following)
+        repeated = following == previous
+        yield page
+        if finished or repeated:
+            return
+        for name, token in zip(spec.input_tokens, following, strict=True):
+            if _present(token):
+                request[name] = token
+            else:
+                request.pop(name, None)
+        previous = tuple(_snapshot(token) for token in following)

rboto/config.py

@@ -0,0 +1,94 @@
+from dataclasses import dataclass, field
+from math import isfinite
+from typing import Literal
+from urllib.parse import urlsplit
+
+from .exceptions import ConfigurationError
+
+
+def _valid_attempts(value: object) -> bool:
+    return isinstance(value, int) and not isinstance(value, bool) and 1 <= value <= 2**32 - 1
+
+
+@dataclass(frozen=True, slots=True, kw_only=True)
+class ClientConfig:
+    """Credential and request settings shared when constructing clients.
+
+    Timeouts are in seconds. ``max_attempts`` includes the initial request.
+    Explicit credentials or a profile can supply the source credentials for
+    ``role_arn``. Credential values are excluded from the representation.
+    """
+
+    access_key_id: str | None = field(default=None, repr=False)
+    secret_access_key: str | None = field(default=None, repr=False)
+    session_token: str | None = field(default=None, repr=False)
+    role_arn: str | None = None
+    role_session_name: str | None = None
+    external_id: str | None = field(default=None, repr=False)
+    connect_timeout: float | None = None
+    read_timeout: float | None = None
+    operation_timeout: float | None = None
+    attempt_timeout: float | None = None
+    max_attempts: int | None = None
+    retry_mode: Literal["standard", "adaptive"] | None = None
+    proxy_url: str | None = field(default=None, repr=False)
+    no_proxy: str | None = None
+    ca_bundle: str | None = None
+
+    def __post_init__(self) -> None:
+        if (self.access_key_id is None) != (self.secret_access_key is None):
+            raise ConfigurationError(
+                "access_key_id and secret_access_key must be supplied together"
+            )
+        if self.session_token is not None and self.access_key_id is None:
+            raise ConfigurationError("session_token requires explicit credentials")
+        if self.role_arn is None and (
+            self.role_session_name is not None or self.external_id is not None
+        ):
+            raise ConfigurationError("role_session_name and external_id require role_arn")
+        for name in (
+            "access_key_id",
+            "secret_access_key",
+            "session_token",
+            "role_arn",
+            "role_session_name",
+            "external_id",
+            "proxy_url",
+            "no_proxy",
+            "ca_bundle",
+        ):
+            value = getattr(self, name)
+            if value is not None and (not isinstance(value, str) or not value.strip()):
+                raise ConfigurationError(f"{name} must be a nonempty string")
+        for name in ("connect_timeout", "read_timeout", "operation_timeout", "attempt_timeout"):
+            value = getattr(self, name)
+            if value is not None and (
+                isinstance(value, bool)
+                or not isinstance(value, (int, float))
+                or value <= 0
+                or value > 2**32 - 1
+                or not isfinite(value)
+            ):
+                raise ConfigurationError(f"{name} must be finite and positive (at most 2**32 - 1)")
+        if self.max_attempts is not None and not _valid_attempts(self.max_attempts):
+            raise ConfigurationError("max_attempts must be an integer between 1 and 2**32 - 1")
+        if self.retry_mode not in (None, "standard", "adaptive"):
+            raise ConfigurationError("retry_mode must be 'standard' or 'adaptive'")
+        if self.no_proxy is not None and self.proxy_url is None:
+            raise ConfigurationError("no_proxy requires proxy_url")
+        if self.proxy_url is not None:
+            try:
+                url = urlsplit(self.proxy_url)
+                valid = (
+                    url.scheme in ("http", "https")
+                    and bool(url.hostname)
+                    and url.port != 0
+                    and url.path in ("", "/")
+                    and not url.query
+                    and not url.fragment
+                    and all(32 < ord(char) < 127 for char in self.proxy_url)
+                )
+            except ValueError:
+                valid = False
+            if not valid:
+                raise ConfigurationError("proxy_url must be an HTTP(S) URL with a host")

rboto/exceptions.py

@@ -0,0 +1,73 @@
+from typing import NotRequired, TypedDict
+
+
+class ErrorDetails(TypedDict):
+    Code: str
+    Message: str
+
+
+class ErrorResponse(TypedDict):
+    Error: ErrorDetails
+    RequestId: NotRequired[str]
+
+
+class RbotoError(Exception):
+    """Base exception for all rboto errors."""
+
+
+class ClientError(RbotoError):
+    """Final error returned by an AWS client operation."""
+
+    message: str
+    error_code: str
+    request_id: str | None
+    operation_name: str
+    response: ErrorResponse
+
+    def __init__(
+        self,
+        message: str,
+        error_code: str = "UnknownError",
+        request_id: str | None = None,
+        operation_name: str = "UnknownOperation",
+    ) -> None:
+        super().__init__(message)
+        self.message = message
+        self.error_code = error_code
+        self.request_id = request_id
+        self.operation_name = operation_name
+        self.response = {
+            "Error": {"Code": error_code, "Message": message},
+        }
+        if request_id is not None:
+            self.response["RequestId"] = request_id
+
+    def __str__(self) -> str:
+        return (
+            f"An error occurred ({self.error_code}) when calling "
+            f"{self.operation_name}: {self.message}"
+        )
+
+
+class ServiceError(ClientError):
+    """Base exception for service-specific AWS errors."""
+
+
+class ConfigurationError(RbotoError):
+    """Invalid rboto configuration."""
+
+
+class CredentialsError(RbotoError):
+    """AWS credentials could not be loaded or refreshed."""
+
+
+class ConnectionError(ClientError):
+    """The request could not reach the service."""
+
+
+class TimeoutError(ConnectionError):
+    """A connection, request attempt, or operation exceeded its timeout."""
+
+
+class ResponseError(ClientError):
+    """The service response could not be decoded."""