capo-lambda-web 0.2.0__py3-none-any.whl

capo_lambda_web/_auth/_zapros_handler.py

@@ -0,0 +1,80 @@
+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."
+    )
+
+
+def strip_accept_encoding(request: Request) -> Request:
+    """Drop the ``Accept-Encoding`` zapros adds to every request.
+
+    Like the AWS SDKs, the client never asks for transport compression: a
+    response body must be the service's actual bytes so ``Content-Length``,
+    checksums and ``Range`` stay coherent and a ``Content-Encoding`` on it is
+    always stored metadata (an S3 object's), never something to undo. Removing
+    the header before signing also keeps it out of the canonical request, where
+    a proxy rewriting it in flight would break the signature.
+    """
+    if "accept-encoding" in request.headers:
+        del request.headers["Accept-Encoding"]
+    return request
+
+
+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 unsigned. Either way
+    ``Accept-Encoding`` is stripped first (see :func:`strip_accept_encoding`).
+    """
+
+    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)
+        request = strip_accept_encoding(request)
+        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)
+        request = strip_accept_encoding(request)
+        signer = cast(Optional[Signer], request.context.get("signer"))
+        if signer is not None:
+            request = await signer.asign(request)
+        return await next_handler.ahandle(request)

capo_lambda_web/_body.py

@@ -0,0 +1,221 @@
+"""Request body abstraction for streaming operations.
+
+Hand-written, not regenerated.
+"""
+
+from __future__ import annotations
+
+import os
+from collections.abc import AsyncIterator, Callable, Iterator
+from contextlib import (
+    AbstractAsyncContextManager,
+    AbstractContextManager,
+    asynccontextmanager,
+    contextmanager,
+)
+from typing import TYPE_CHECKING, Generic, TypeVar, Union, cast
+
+from ._iter import AnyIterator
+
+if TYPE_CHECKING:
+    import anyio
+else:
+    try:
+        import anyio
+    except ImportError:
+        anyio = None
+
+_CHUNK_SIZE = 64 * 1024
+
+TStream = TypeVar("TStream", Iterator[bytes], AsyncIterator[bytes])
+
+# The opener yields the fresh stream together with its length, or ``None`` when
+# no stream could be produced (an exhausted or non-rewindable source, or logic
+# that decides dynamically it cannot reopen). Pairing the length with each open
+# lets it change between rebuilds instead of being fixed for the whole body.
+_ContextManager = Union[
+    AbstractContextManager[Union[tuple[TStream, int], None]],
+    AbstractAsyncContextManager[Union[tuple[TStream, int], None]],
+]
+
+
+class Body(AnyIterator[bytes], Generic[TStream]):
+    """A replayable request body whose length travels with each open.
+
+    A plain ``Iterator[bytes]`` can be sent only once, so a failed request
+    cannot be retried. A ``Body`` instead holds an *opener* — a context manager
+    that opens a fresh stream over the same source — and rebuilds the stream
+    before every attempt, closing whatever was left of the previous one.
+
+    ``TStream`` is ``Iterator[bytes]`` for bodies built with :meth:`from_path`
+    and ``AsyncIterator[bytes]`` for :meth:`async_from_path`; the former use
+    ``rebuild()``/``close()``, the latter ``arebuild()``/``aclose()``.
+    ``build_request`` rebuilds before every attempt.
+
+    Each open produces ``(stream, length)`` together, so the length is not bound
+    to the whole body and may differ from one rebuild to the next. A rebuild is
+    also not guaranteed to succeed: the source may be exhausted or
+    non-rewindable, so the opener can yield ``None`` and :meth:`rebuild` then
+    returns ``None`` instead of a pair. Callers check the return value::
+
+        for attempt in range(max_attempts):
+            opened = body.rebuild()
+            if opened is None:
+                raise RuntimeError("body could not be rebuilt")
+            stream, length = opened
+            send(stream, length)
+
+    ``length`` also holds the current stream's length as an attribute (``None``
+    before the first rebuild). Iterating a ``Body`` reads its current stream.
+    """
+
+    def __init__(self, opener: Callable[[], _ContextManager[TStream]]) -> None:
+        self._opener = opener
+        self._cm: _ContextManager[TStream] | None = None
+        self._stream: TStream | None = None
+        self.length: int | None = None
+
+    @classmethod
+    def from_path(
+        cls, path: str | os.PathLike[str], chunk_size: int = _CHUNK_SIZE
+    ) -> Body[Iterator[bytes]]:
+        """A body that reads ``path`` in ``chunk_size`` chunks, reopening it (and re-measuring it) on every rebuild."""
+
+        @contextmanager
+        def open_() -> Iterator[tuple[Iterator[bytes], int]]:
+            with open(path, "rb") as file:
+                length = os.fstat(file.fileno()).st_size
+                yield iter(lambda: file.read(chunk_size), b""), length
+
+        return Body(open_)
+
+    @classmethod
+    def async_from_path(
+        cls, path: str | os.PathLike[str], chunk_size: int = _CHUNK_SIZE
+    ) -> Body[AsyncIterator[bytes]]:
+        """The async counterpart of :meth:`from_path`; needs ``anyio`` installed."""
+        if anyio is None:
+            dist = (__package__ or "capo").replace("_", "-")
+            raise RuntimeError(
+                "Body.async_from_path() requires the anyio package; "
+                f"reinstall with the anyio feature enabled: {dist}[anyio]"
+            )
+
+        @asynccontextmanager
+        async def open_() -> AsyncIterator[tuple[AsyncIterator[bytes], int]]:
+            async with await anyio.open_file(path, "rb") as file:
+                length = os.fstat(file.wrapped.fileno()).st_size
+
+                async def chunks() -> AsyncIterator[bytes]:
+                    while chunk := await file.read(chunk_size):
+                        yield chunk
+
+                yield chunks(), length
+
+        return Body(open_)
+
+    @property
+    def stream(self) -> TStream | None:
+        """The current open stream, or ``None`` if none is open (before the first
+        rebuild, or after a close).
+
+        ``build_request`` reuses a non-``None`` stream instead of rebuilding, so
+        the retry layer stays in control of when a fresh stream is created.
+        """
+        return self._stream
+
+    def rebuild(self: Body[Iterator[bytes]]) -> tuple[Iterator[bytes], int] | None:
+        """Reopen the stream, closing any previous one, and return ``(stream, length)``.
+
+        ``None`` if the source could not be reopened.
+        """
+        self.close()
+        cm = self._opener()
+        if not isinstance(cm, AbstractContextManager):
+            raise TypeError(
+                "this Body has an async opener; use it with the async client (arebuild/aclose)"
+            )
+        cm = cast(AbstractContextManager["tuple[Iterator[bytes], int] | None"], cm)
+        opened = cm.__enter__()
+        if opened is None:
+            cm.__exit__(None, None, None)
+            return None
+        stream, length = opened
+        self._stream, self.length, self._cm = stream, length, cm
+        return stream, length
+
+    async def arebuild(
+        self: Body[AsyncIterator[bytes]],
+    ) -> tuple[AsyncIterator[bytes], int] | None:
+        """Async :meth:`rebuild`, returning ``(stream, length)`` or ``None``."""
+        await self.aclose()
+        cm = self._opener()
+        if not isinstance(cm, AbstractAsyncContextManager):
+            raise TypeError(
+                "this Body has a sync opener; use it with the sync client (rebuild/close)"
+            )
+        cm = cast(
+            AbstractAsyncContextManager["tuple[AsyncIterator[bytes], int] | None"], cm
+        )
+        opened = await cm.__aenter__()
+        if opened is None:
+            await cm.__aexit__(None, None, None)
+            return None
+        stream, length = opened
+        self._stream, self.length, self._cm = stream, length, cm
+        return stream, length
+
+    def close(self: Body[Iterator[bytes]]) -> None:
+        cm, self._cm, self._stream, self.length = self._cm, None, None, None
+        if cm is not None:
+            cast(AbstractContextManager[object], cm).__exit__(None, None, None)
+
+    async def aclose(self: Body[AsyncIterator[bytes]]) -> None:
+        cm, self._cm, self._stream, self.length = self._cm, None, None, None
+        if cm is not None:
+            await cast(AbstractAsyncContextManager[object], cm).__aexit__(
+                None, None, None
+            )
+
+    def __next__(self) -> bytes:
+        if self._stream is None:
+            raise RuntimeError("body has no open stream; call rebuild() first")
+        return next(cast(Iterator[bytes], self._stream))
+
+    async def __anext__(self) -> bytes:
+        if self._stream is None:
+            raise RuntimeError("body has no open stream; call rebuild() first")
+        return await cast(AsyncIterator[bytes], self._stream).__anext__()
+
+
+@contextmanager
+def closing_bodies(inp: object) -> Iterator[None]:
+    """Close every :class:`Body` in a request input when the operation finishes.
+
+    The operation method wraps its whole run (request build across retries, plus
+    response handling) in this, so a ``Body``'s stream is released once — and only
+    once — the operation is done. Non-``Body`` values (bytes, plain iterators,
+    ``StaticAnyIterator``) are left untouched.
+    """
+    try:
+        yield
+    finally:
+        if isinstance(inp, dict):
+            for value in inp.values():
+                if isinstance(value, Body):
+                    cast("Body[Iterator[bytes]]", value).close()
+
+
+@asynccontextmanager
+async def aclosing_bodies(inp: object) -> AsyncIterator[None]:
+    """Async :func:`closing_bodies`, awaiting :meth:`Body.aclose`."""
+    try:
+        yield
+    finally:
+        if isinstance(inp, dict):
+            for value in inp.values():
+                if isinstance(value, Body):
+                    await cast("Body[AsyncIterator[bytes]]", value).aclose()
+
+
+__all__ = ["Body", "aclosing_bodies", "closing_bodies"]

capo_lambda_web/_iter.py

@@ -0,0 +1,113 @@
+"""Iterator helpers used by generated streaming operations.
+
+Hand-written, not regenerated.
+"""
+
+from __future__ import annotations
+
+from collections.abc import AsyncIterable, AsyncIterator, Callable, Iterable, Iterator
+from typing import Generic, TypeVar, cast
+
+T = TypeVar("T")
+U = TypeVar("U")
+
+
+class AnyIterator(AsyncIterator[T], Iterator[T], Generic[T]):
+    """A type that is both a synchronous and asynchronous iterator.
+
+    Streaming output fields use this so the same TypedDict type works for
+    both sync and async operation variants.
+    """
+
+    ...
+
+
+class StaticAnyIterator(AnyIterator[T], Generic[T]):
+    """An :class:`AnyIterator` backed by static, in-memory content.
+
+    Used when streaming output is produced from already-materialized bytes
+    rather than a live stream. The ``content`` is yielded once and remains
+    accessible as a public attribute.
+    """
+
+    def __init__(self, content: T) -> None:
+        self.content = content
+        self._consumed = False
+
+    def __next__(self) -> T:
+        if self._consumed:
+            raise StopIteration
+        self._consumed = True
+        return self.content
+
+    async def __anext__(self) -> T:
+        if self._consumed:
+            raise StopAsyncIteration
+        self._consumed = True
+        return self.content
+
+
+def map_sync_iterator(iterable: Iterable[T], fn: Callable[[T], U]) -> Iterator[U]:
+    """Map ``fn`` over a synchronous iterable, yielding results lazily."""
+    for item in iterable:
+        yield fn(item)
+
+
+async def map_async_iterator(
+    async_iterable: AsyncIterable[T], fn: Callable[[T], U]
+) -> AsyncIterator[U]:
+    """Map ``fn`` over an asynchronous iterable, yielding results lazily."""
+    async for item in async_iterable:
+        yield fn(item)
+
+
+def chain_sync_iterator(*iterables: Iterable[T]) -> Iterator[T]:
+    """Yield items from each iterable in order."""
+    for iterable in iterables:
+        yield from iterable
+
+
+async def chain_async_iterator(
+    prefix: Iterable[T], rest: AsyncIterable[T]
+) -> AsyncIterator[T]:
+    """Yield the synchronous ``prefix`` items, then the async ``rest`` items."""
+    for item in prefix:
+        yield item
+    async for item in rest:
+        yield item
+
+
+def ensure_async_iterator(value: AsyncIterator[T] | T) -> AnyIterator[T]:
+    """Return ``value`` as an :class:`AnyIterator`.
+
+    An async iterator is returned unchanged; a single value is wrapped in a
+    fresh single-item async iterator.
+    """
+    if isinstance(value, AsyncIterator):
+        return cast(AnyIterator[T], value)
+
+    return StaticAnyIterator(value)
+
+
+def ensure_sync_iterator(value: Iterator[T] | T) -> AnyIterator[T]:
+    """Return ``value`` as an :class:`AnyIterator`.
+
+    A sync iterator is returned unchanged; a single value is wrapped in a
+    fresh single-item sync iterator.
+    """
+    if isinstance(value, Iterator):
+        return cast(AnyIterator[T], value)
+
+    return StaticAnyIterator(value)
+
+
+__all__ = [
+    "AnyIterator",
+    "StaticAnyIterator",
+    "chain_async_iterator",
+    "chain_sync_iterator",
+    "ensure_async_iterator",
+    "ensure_sync_iterator",
+    "map_async_iterator",
+    "map_sync_iterator",
+]

capo_lambda_web/_operations/lambda_web/create_web_function.py

@@ -0,0 +1,184 @@
+"""Generated from Smithy shape ``com.amazonaws.lambdaweb#CreateWebFunction``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+
+import zapros
+from typing_extensions import Never
+
+import capo_lambda_web._auth._signers
+import capo_lambda_web._auth._sigv4
+import capo_lambda_web._protocol.eventstream
+import capo_lambda_web.errors.access_denied_exception
+import capo_lambda_web.errors.conflict_exception
+import capo_lambda_web.errors.internal_server_exception
+import capo_lambda_web.errors.resource_not_found_exception
+import capo_lambda_web.errors.service_quota_exceeded_exception
+import capo_lambda_web.errors.throttling_exception
+import capo_lambda_web.errors.validation_exception
+import capo_lambda_web.types.create_web_function_request
+import capo_lambda_web.types.create_web_function_response
+import capo_lambda_web.types.date_time
+import capo_lambda_web.types.endpoint_config
+import capo_lambda_web.types.function_endpoint_summary
+import capo_lambda_web.types.function_revision_summary
+import capo_lambda_web.types.function_state
+import capo_lambda_web.types.revision_config
+import capo_lambda_web.types.tags
+from capo_lambda_web._protocol.errors import parse_error_metadata_json
+from capo_lambda_web._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_lambda_web._services._pipeline import (
+    AsyncOperationOptions,
+    OperationOptions,
+    raise_error,
+)
+from capo_lambda_web.errors import UnknownServiceError
+
+
+def handle_error(response: zapros.Response) -> Never:
+    data = json.loads(response.read())
+    code, message = parse_error_metadata_json(response, data)
+    match code:
+        case "AccessDeniedException":
+            raise capo_lambda_web.errors.access_denied_exception.AccessDeniedException.from_json(
+                data, message
+            )
+        case "ConflictException":
+            raise capo_lambda_web.errors.conflict_exception.ConflictException.from_json(
+                data, message
+            )
+        case "InternalServerException":
+            raise capo_lambda_web.errors.internal_server_exception.InternalServerException.from_json(
+                data, message
+            )
+        case "ResourceNotFoundException":
+            raise capo_lambda_web.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data, message
+            )
+        case "ServiceQuotaExceededException":
+            raise capo_lambda_web.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_json(
+                data, message
+            )
+        case "ThrottlingException":
+            raise capo_lambda_web.errors.throttling_exception.ThrottlingException.from_json(
+                data, message
+            )
+        case "ValidationException":
+            raise capo_lambda_web.errors.validation_exception.ValidationException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_lambda_web.types.create_web_function_response.CreateWebFunctionResponse:
+    out: capo_lambda_web.types.create_web_function_response.CreateWebFunctionResponse = capo_lambda_web.types.create_web_function_response.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_lambda_web.types.create_web_function_response.CreateWebFunctionResponse:
+    out: capo_lambda_web.types.create_web_function_response.CreateWebFunctionResponse = capo_lambda_web.types.create_web_function_response.deserialize_json(
+        json.loads(await response.aread())
+    )
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> capo_lambda_web._auth._signers.Signer | None:
+    name_to_schema = {s["name"]: s for s in (auth_schemes or [])}  # noqa: F841
+    if (
+        options.credentials_provider is not None
+        and name_to_schema
+        and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
+    ):
+        raise RuntimeError(
+            "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
+        )
+    if options.credentials_provider is not None:
+        endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
+            "sigv4-s3express"
+        )
+        if endpoint_scheme is not None or not name_to_schema:
+            sigv4_config = capo_lambda_web._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_lambda_web._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_lambda_web.types.create_web_function_request.CreateWebFunctionRequest,
+) -> zapros.Request:
+    endpoint = resolve(
+        EndpointParams(
+            Region=options.region,
+            UseDualStack=options.use_dual_stack,
+            UseFIPS=options.use_fips,
+            Endpoint=options.endpoint,
+        )
+    )  # noqa: F841
+    url = endpoint.url.rstrip("/") + "/2025-03-07/web-functions"
+    params: list[tuple[str, str]] = []
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    body: bytes | None = json.dumps(
+        capo_lambda_web.types.create_web_function_request.serialize_json(input_),
+        allow_nan=False,
+    ).encode()
+    headers["content-type"] = "application/json"
+    signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
+    normalized_url = zapros.URL(url)
+    for k, v in params:
+        normalized_url.search_params.append(k, v)
+    return zapros.Request(
+        normalized_url, "PUT", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def create_web_function(
+    options: OperationOptions,
+    input_: capo_lambda_web.types.create_web_function_request.CreateWebFunctionRequest,
+) -> tuple[
+    capo_lambda_web.types.create_web_function_response.CreateWebFunctionResponse,
+    zapros.Response,
+]:
+    response = options.client.handler.handle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            response.read()
+            raise_error(response, handle_error)
+        return handle_response(response), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_create_web_function(
+    options: AsyncOperationOptions,
+    input_: capo_lambda_web.types.create_web_function_request.CreateWebFunctionRequest,
+) -> tuple[
+    capo_lambda_web.types.create_web_function_response.CreateWebFunctionResponse,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            await response.aread()
+            raise_error(response, handle_error)
+        return await async_handle_response(response), response
+    except BaseException:
+        await response.aclose()
+        raise