aws-sdk-lambda 0.4.0__py3-none-any.whl

aws_sdk_lambda/_auth/_zapros_handler.py

@@ -0,0 +1,62 @@
+from typing import Optional, cast
+
+from zapros import (
+    AsyncBaseHandler,
+    AsyncBaseMiddleware,
+    AsyncSyncMismatchError,
+    BaseHandler,
+    BaseMiddleware,
+    Request,
+)
+from zapros._models import Response
+
+from ._signers import Signer
+
+
+def ensure_async_handler(
+    handler: AsyncBaseHandler | BaseHandler,
+) -> AsyncBaseHandler:
+    if isinstance(handler, AsyncBaseHandler):
+        return handler
+    raise AsyncSyncMismatchError(
+        "Handler was expected to be an AsyncBaseHandler, but it is not."
+    )
+
+
+def ensure_sync_handler(
+    handler: AsyncBaseHandler | BaseHandler,
+) -> BaseHandler:
+    if isinstance(handler, BaseHandler):
+        return handler
+    raise AsyncSyncMismatchError(
+        "Handler was expected to be a BaseHandler, but it is not."
+    )
+
+
+class AuthMiddleware(BaseMiddleware, AsyncBaseMiddleware):
+    """Sign the outgoing request using the Signer placed in ``context["signer"]``.
+
+    The generated ``get_signer`` in each operation module resolves the
+    effective auth scheme to a concrete :class:`Signer` and threads it
+    through ``request.context`` before dispatch. This middleware does the
+    actual signing; ``None`` means the operation opted out via
+    ``@optionalAuth`` and the request passes through untouched.
+    """
+
+    def __init__(self, next_handler: BaseHandler | AsyncBaseHandler) -> None:
+        self.next = cast(BaseHandler, next_handler)
+        self.async_next = cast(AsyncBaseHandler, next_handler)
+
+    def handle(self, request: Request) -> Response:
+        next_handler = ensure_sync_handler(self.next)
+        signer = cast(Optional[Signer], request.context.get("signer"))
+        if signer is not None:
+            request = signer.sign(request)
+        return next_handler.handle(request)
+
+    async def ahandle(self, request: Request) -> Response:
+        next_handler = ensure_async_handler(self.async_next)
+        signer = cast(Optional[Signer], request.context.get("signer"))
+        if signer is not None:
+            request = await signer.asign(request)
+        return await next_handler.ahandle(request)

aws_sdk_lambda/_operations/aws_gir_api_service/add_layer_version_permission.py

@@ -0,0 +1,178 @@
+"""Generated from Smithy shape ``com.amazonaws.lambda#AddLayerVersionPermission``."""
+
+from __future__ import annotations
+
+import json
+from typing import TYPE_CHECKING, Any, Never
+from urllib.parse import quote
+
+import zapros
+
+import aws_sdk_lambda._auth._signers
+import aws_sdk_lambda._auth._sigv4
+from aws_sdk_lambda._protocol.errors import parse_error_metadata_json
+from aws_sdk_lambda._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from aws_sdk_lambda._services._pipeline import AsyncOperationOptions, OperationOptions
+from aws_sdk_lambda.errors import UnknownServiceError
+
+if TYPE_CHECKING:
+    import aws_sdk_lambda.types.add_layer_version_permission_request
+    import aws_sdk_lambda.types.add_layer_version_permission_response
+
+
+def handle_error(response: zapros.Response) -> Never:
+    data = json.loads(response.read())
+    code, message = parse_error_metadata_json(response, data)
+    match code:
+        case "InvalidParameterValueException":
+            import aws_sdk_lambda.errors.invalid_parameter_value_exception
+
+            raise aws_sdk_lambda.errors.invalid_parameter_value_exception.InvalidParameterValueException.from_json(
+                data
+            )
+        case "PolicyLengthExceededException":
+            import aws_sdk_lambda.errors.policy_length_exceeded_exception
+
+            raise aws_sdk_lambda.errors.policy_length_exceeded_exception.PolicyLengthExceededException.from_json(
+                data
+            )
+        case "PreconditionFailedException":
+            import aws_sdk_lambda.errors.precondition_failed_exception
+
+            raise aws_sdk_lambda.errors.precondition_failed_exception.PreconditionFailedException.from_json(
+                data
+            )
+        case "ResourceConflictException":
+            import aws_sdk_lambda.errors.resource_conflict_exception
+
+            raise aws_sdk_lambda.errors.resource_conflict_exception.ResourceConflictException.from_json(
+                data
+            )
+        case "ResourceNotFoundException":
+            import aws_sdk_lambda.errors.resource_not_found_exception
+
+            raise aws_sdk_lambda.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data
+            )
+        case "ServiceException":
+            import aws_sdk_lambda.errors.service_exception
+
+            raise aws_sdk_lambda.errors.service_exception.ServiceException.from_json(
+                data
+            )
+        case "TooManyRequestsException":
+            import aws_sdk_lambda.errors.too_many_requests_exception
+
+            raise aws_sdk_lambda.errors.too_many_requests_exception.TooManyRequestsException.from_json(
+                data
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response, is_async: bool
+) -> aws_sdk_lambda.types.add_layer_version_permission_response.AddLayerVersionPermissionResponse:
+    import aws_sdk_lambda.types.add_layer_version_permission_response
+
+    out: aws_sdk_lambda.types.add_layer_version_permission_response.AddLayerVersionPermissionResponse = aws_sdk_lambda.types.add_layer_version_permission_response.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> aws_sdk_lambda._auth._signers.Signer | None:
+    name_to_schema = {s["name"]: s for s in (auth_schemes or [])}
+    if options.credentials_provider is not None:
+        sigv4_config = (
+            name_to_schema.get("sigv4")
+            or name_to_schema.get("sigv4a")
+            or name_to_schema.get("sigv4-s3express")
+            or aws_sdk_lambda._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region
+            )
+        )
+        if sigv4_config is not None:
+            return aws_sdk_lambda._auth._signers.SigV4Signer(
+                options.credentials_provider, auth_scheme=sigv4_config
+            )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input: aws_sdk_lambda.types.add_layer_version_permission_request.AddLayerVersionPermissionRequest,
+) -> zapros.Request:
+    endpoint = resolve(  # noqa: F841
+        EndpointParams(
+            Region=options.region,
+            UseDualStack=options.use_dual_stack,
+            UseFIPS=options.use_fips,
+            Endpoint=options.endpoint,
+        )
+    )
+    url = (
+        endpoint.url.rstrip("/")
+        + "/2018-10-31/layers/{LayerName}/versions/{VersionNumber}/policy"
+    )
+    url = url.replace("{LayerName}", quote(str(input["layer_name"]), safe=""))
+    url = url.replace("{VersionNumber}", quote(str(input["version_number"]), safe=""))
+    params: dict[str, str] = {}
+    if "revision_id" in input:
+        params["RevisionId"] = str(input["revision_id"])
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    import aws_sdk_lambda.types.add_layer_version_permission_request
+
+    body: bytes | None = json.dumps(
+        aws_sdk_lambda.types.add_layer_version_permission_request.serialize_json(input)
+    ).encode()
+    headers["content-type"] = "application/json"
+    signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
+    normalized_url = zapros.URL(url)
+    normalized_url.search_params.update(params)
+    return zapros.Request(
+        normalized_url,
+        "POST",
+        headers=headers,
+        body=body,
+        context={"signer": signer},
+    )
+
+
+def add_layer_version_permission(
+    options: OperationOptions,
+    input: aws_sdk_lambda.types.add_layer_version_permission_request.AddLayerVersionPermissionRequest,
+) -> tuple[
+    aws_sdk_lambda.types.add_layer_version_permission_response.AddLayerVersionPermissionResponse,
+    zapros.Response,
+]:
+    response = options.client.handler.handle(build_request(options, input))
+    try:
+        if response.status >= 400:
+            response.read()
+            handle_error(response)
+        return handle_response(response, is_async=False), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_add_layer_version_permission(
+    options: AsyncOperationOptions,
+    input: aws_sdk_lambda.types.add_layer_version_permission_request.AddLayerVersionPermissionRequest,
+) -> tuple[
+    aws_sdk_lambda.types.add_layer_version_permission_response.AddLayerVersionPermissionResponse,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input))
+    try:
+        if response.status >= 400:
+            await response.aread()
+            handle_error(response)
+        return handle_response(response, is_async=True), response
+    except BaseException:
+        await response.aclose()
+        raise

aws_sdk_lambda/_operations/aws_gir_api_service/add_permission.py

@@ -0,0 +1,174 @@
+"""Generated from Smithy shape ``com.amazonaws.lambda#AddPermission``."""
+
+from __future__ import annotations
+
+import json
+from typing import TYPE_CHECKING, Any, Never
+from urllib.parse import quote
+
+import zapros
+
+import aws_sdk_lambda._auth._signers
+import aws_sdk_lambda._auth._sigv4
+from aws_sdk_lambda._protocol.errors import parse_error_metadata_json
+from aws_sdk_lambda._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from aws_sdk_lambda._services._pipeline import AsyncOperationOptions, OperationOptions
+from aws_sdk_lambda.errors import UnknownServiceError
+
+if TYPE_CHECKING:
+    import aws_sdk_lambda.types.add_permission_request
+    import aws_sdk_lambda.types.add_permission_response
+
+
+def handle_error(response: zapros.Response) -> Never:
+    data = json.loads(response.read())
+    code, message = parse_error_metadata_json(response, data)
+    match code:
+        case "InvalidParameterValueException":
+            import aws_sdk_lambda.errors.invalid_parameter_value_exception
+
+            raise aws_sdk_lambda.errors.invalid_parameter_value_exception.InvalidParameterValueException.from_json(
+                data
+            )
+        case "PolicyLengthExceededException":
+            import aws_sdk_lambda.errors.policy_length_exceeded_exception
+
+            raise aws_sdk_lambda.errors.policy_length_exceeded_exception.PolicyLengthExceededException.from_json(
+                data
+            )
+        case "PreconditionFailedException":
+            import aws_sdk_lambda.errors.precondition_failed_exception
+
+            raise aws_sdk_lambda.errors.precondition_failed_exception.PreconditionFailedException.from_json(
+                data
+            )
+        case "ResourceConflictException":
+            import aws_sdk_lambda.errors.resource_conflict_exception
+
+            raise aws_sdk_lambda.errors.resource_conflict_exception.ResourceConflictException.from_json(
+                data
+            )
+        case "ResourceNotFoundException":
+            import aws_sdk_lambda.errors.resource_not_found_exception
+
+            raise aws_sdk_lambda.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data
+            )
+        case "ServiceException":
+            import aws_sdk_lambda.errors.service_exception
+
+            raise aws_sdk_lambda.errors.service_exception.ServiceException.from_json(
+                data
+            )
+        case "TooManyRequestsException":
+            import aws_sdk_lambda.errors.too_many_requests_exception
+
+            raise aws_sdk_lambda.errors.too_many_requests_exception.TooManyRequestsException.from_json(
+                data
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response, is_async: bool
+) -> aws_sdk_lambda.types.add_permission_response.AddPermissionResponse:
+    import aws_sdk_lambda.types.add_permission_response
+
+    out: aws_sdk_lambda.types.add_permission_response.AddPermissionResponse = (
+        aws_sdk_lambda.types.add_permission_response.deserialize_json(
+            json.loads(response.read())
+        )
+    )
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> aws_sdk_lambda._auth._signers.Signer | None:
+    name_to_schema = {s["name"]: s for s in (auth_schemes or [])}
+    if options.credentials_provider is not None:
+        sigv4_config = (
+            name_to_schema.get("sigv4")
+            or name_to_schema.get("sigv4a")
+            or name_to_schema.get("sigv4-s3express")
+            or aws_sdk_lambda._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region
+            )
+        )
+        if sigv4_config is not None:
+            return aws_sdk_lambda._auth._signers.SigV4Signer(
+                options.credentials_provider, auth_scheme=sigv4_config
+            )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input: aws_sdk_lambda.types.add_permission_request.AddPermissionRequest,
+) -> zapros.Request:
+    endpoint = resolve(  # noqa: F841
+        EndpointParams(
+            Region=options.region,
+            UseDualStack=options.use_dual_stack,
+            UseFIPS=options.use_fips,
+            Endpoint=options.endpoint,
+        )
+    )
+    url = endpoint.url.rstrip("/") + "/2015-03-31/functions/{FunctionName}/policy"
+    url = url.replace("{FunctionName}", quote(str(input["function_name"]), safe=""))
+    params: dict[str, str] = {}
+    if "qualifier" in input:
+        params["Qualifier"] = str(input["qualifier"])
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    import aws_sdk_lambda.types.add_permission_request
+
+    body: bytes | None = json.dumps(
+        aws_sdk_lambda.types.add_permission_request.serialize_json(input)
+    ).encode()
+    headers["content-type"] = "application/json"
+    signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
+    normalized_url = zapros.URL(url)
+    normalized_url.search_params.update(params)
+    return zapros.Request(
+        normalized_url,
+        "POST",
+        headers=headers,
+        body=body,
+        context={"signer": signer},
+    )
+
+
+def add_permission(
+    options: OperationOptions,
+    input: aws_sdk_lambda.types.add_permission_request.AddPermissionRequest,
+) -> tuple[
+    aws_sdk_lambda.types.add_permission_response.AddPermissionResponse, zapros.Response
+]:
+    response = options.client.handler.handle(build_request(options, input))
+    try:
+        if response.status >= 400:
+            response.read()
+            handle_error(response)
+        return handle_response(response, is_async=False), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_add_permission(
+    options: AsyncOperationOptions,
+    input: aws_sdk_lambda.types.add_permission_request.AddPermissionRequest,
+) -> tuple[
+    aws_sdk_lambda.types.add_permission_response.AddPermissionResponse, zapros.Response
+]:
+    response = await options.client.handler.ahandle(build_request(options, input))
+    try:
+        if response.status >= 400:
+            await response.aread()
+            handle_error(response)
+        return handle_response(response, is_async=True), response
+    except BaseException:
+        await response.aclose()
+        raise

aws_sdk_lambda/_operations/aws_gir_api_service/checkpoint_durable_execution.py

@@ -0,0 +1,153 @@
+"""Generated from Smithy shape ``com.amazonaws.lambda#CheckpointDurableExecution``."""
+
+from __future__ import annotations
+
+import json
+from typing import TYPE_CHECKING, Any, Never
+from urllib.parse import quote
+
+import zapros
+
+import aws_sdk_lambda._auth._signers
+import aws_sdk_lambda._auth._sigv4
+from aws_sdk_lambda._protocol.errors import parse_error_metadata_json
+from aws_sdk_lambda._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from aws_sdk_lambda._services._pipeline import AsyncOperationOptions, OperationOptions
+from aws_sdk_lambda.errors import UnknownServiceError
+
+if TYPE_CHECKING:
+    import aws_sdk_lambda.types.checkpoint_durable_execution_request
+    import aws_sdk_lambda.types.checkpoint_durable_execution_response
+
+
+def handle_error(response: zapros.Response) -> Never:
+    data = json.loads(response.read())
+    code, message = parse_error_metadata_json(response, data)
+    match code:
+        case "InvalidParameterValueException":
+            import aws_sdk_lambda.errors.invalid_parameter_value_exception
+
+            raise aws_sdk_lambda.errors.invalid_parameter_value_exception.InvalidParameterValueException.from_json(
+                data
+            )
+        case "ServiceException":
+            import aws_sdk_lambda.errors.service_exception
+
+            raise aws_sdk_lambda.errors.service_exception.ServiceException.from_json(
+                data
+            )
+        case "TooManyRequestsException":
+            import aws_sdk_lambda.errors.too_many_requests_exception
+
+            raise aws_sdk_lambda.errors.too_many_requests_exception.TooManyRequestsException.from_json(
+                data
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response, is_async: bool
+) -> aws_sdk_lambda.types.checkpoint_durable_execution_response.CheckpointDurableExecutionResponse:
+    import aws_sdk_lambda.types.checkpoint_durable_execution_response
+
+    out: aws_sdk_lambda.types.checkpoint_durable_execution_response.CheckpointDurableExecutionResponse = aws_sdk_lambda.types.checkpoint_durable_execution_response.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> aws_sdk_lambda._auth._signers.Signer | None:
+    name_to_schema = {s["name"]: s for s in (auth_schemes or [])}
+    if options.credentials_provider is not None:
+        sigv4_config = (
+            name_to_schema.get("sigv4")
+            or name_to_schema.get("sigv4a")
+            or name_to_schema.get("sigv4-s3express")
+            or aws_sdk_lambda._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region
+            )
+        )
+        if sigv4_config is not None:
+            return aws_sdk_lambda._auth._signers.SigV4Signer(
+                options.credentials_provider, auth_scheme=sigv4_config
+            )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input: aws_sdk_lambda.types.checkpoint_durable_execution_request.CheckpointDurableExecutionRequest,
+) -> zapros.Request:
+    endpoint = resolve(  # noqa: F841
+        EndpointParams(
+            Region=options.region,
+            UseDualStack=options.use_dual_stack,
+            UseFIPS=options.use_fips,
+            Endpoint=options.endpoint,
+        )
+    )
+    url = (
+        endpoint.url.rstrip("/")
+        + "/2025-12-01/durable-executions/{DurableExecutionArn}/checkpoint"
+    )
+    url = url.replace(
+        "{DurableExecutionArn}", quote(str(input["durable_execution_arn"]), safe="")
+    )
+    params: dict[str, str] = {}
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    import aws_sdk_lambda.types.checkpoint_durable_execution_request
+
+    body: bytes | None = json.dumps(
+        aws_sdk_lambda.types.checkpoint_durable_execution_request.serialize_json(input)
+    ).encode()
+    headers["content-type"] = "application/json"
+    signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
+    normalized_url = zapros.URL(url)
+    normalized_url.search_params.update(params)
+    return zapros.Request(
+        normalized_url,
+        "POST",
+        headers=headers,
+        body=body,
+        context={"signer": signer},
+    )
+
+
+def checkpoint_durable_execution(
+    options: OperationOptions,
+    input: aws_sdk_lambda.types.checkpoint_durable_execution_request.CheckpointDurableExecutionRequest,
+) -> tuple[
+    aws_sdk_lambda.types.checkpoint_durable_execution_response.CheckpointDurableExecutionResponse,
+    zapros.Response,
+]:
+    response = options.client.handler.handle(build_request(options, input))
+    try:
+        if response.status >= 400:
+            response.read()
+            handle_error(response)
+        return handle_response(response, is_async=False), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_checkpoint_durable_execution(
+    options: AsyncOperationOptions,
+    input: aws_sdk_lambda.types.checkpoint_durable_execution_request.CheckpointDurableExecutionRequest,
+) -> tuple[
+    aws_sdk_lambda.types.checkpoint_durable_execution_response.CheckpointDurableExecutionResponse,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input))
+    try:
+        if response.status >= 400:
+            await response.aread()
+            handle_error(response)
+        return handle_response(response, is_async=True), response
+    except BaseException:
+        await response.aclose()
+        raise