capo-codeartifact 0.1.0__py3-none-any.whl

capo_codeartifact/_operations/code_artifact_control_plane_service/update_package_versions_status.py

@@ -0,0 +1,198 @@
+"""Generated from Smithy shape ``com.amazonaws.codeartifact#UpdatePackageVersionsStatus``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+
+import zapros
+from typing_extensions import Never
+
+import capo_codeartifact._auth._signers
+import capo_codeartifact._auth._sigv4
+import capo_codeartifact._protocol.eventstream
+import capo_codeartifact.errors.access_denied_exception
+import capo_codeartifact.errors.conflict_exception
+import capo_codeartifact.errors.internal_server_exception
+import capo_codeartifact.errors.resource_not_found_exception
+import capo_codeartifact.errors.throttling_exception
+import capo_codeartifact.errors.validation_exception
+import capo_codeartifact.types.package_format
+import capo_codeartifact.types.package_version_error_map
+import capo_codeartifact.types.package_version_list
+import capo_codeartifact.types.package_version_revision_map
+import capo_codeartifact.types.package_version_status
+import capo_codeartifact.types.successful_package_version_info_map
+import capo_codeartifact.types.update_package_versions_status_request
+import capo_codeartifact.types.update_package_versions_status_result
+from capo_codeartifact._protocol.errors import parse_error_metadata_json
+from capo_codeartifact._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_codeartifact._services._pipeline import (
+    AsyncOperationOptions,
+    OperationOptions,
+)
+from capo_codeartifact.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_codeartifact.errors.access_denied_exception.AccessDeniedException.from_json(
+                data, message
+            )
+        case "ConflictException":
+            raise capo_codeartifact.errors.conflict_exception.ConflictException.from_json(
+                data, message
+            )
+        case "InternalServerException":
+            raise capo_codeartifact.errors.internal_server_exception.InternalServerException.from_json(
+                data, message
+            )
+        case "ResourceNotFoundException":
+            raise capo_codeartifact.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data, message
+            )
+        case "ThrottlingException":
+            raise capo_codeartifact.errors.throttling_exception.ThrottlingException.from_json(
+                data, message
+            )
+        case "ValidationException":
+            raise capo_codeartifact.errors.validation_exception.ValidationException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_codeartifact.types.update_package_versions_status_result.UpdatePackageVersionsStatusResult:
+    out: capo_codeartifact.types.update_package_versions_status_result.UpdatePackageVersionsStatusResult = capo_codeartifact.types.update_package_versions_status_result.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_codeartifact.types.update_package_versions_status_result.UpdatePackageVersionsStatusResult:
+    out: capo_codeartifact.types.update_package_versions_status_result.UpdatePackageVersionsStatusResult = capo_codeartifact.types.update_package_versions_status_result.deserialize_json(
+        json.loads(await response.aread())
+    )
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> capo_codeartifact._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_codeartifact._auth._sigv4.build_sigv4_auth_scheme(
+                "codeartifact", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_codeartifact._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_codeartifact.types.update_package_versions_status_request.UpdatePackageVersionsStatusRequest,
+) -> zapros.Request:
+    endpoint = resolve(
+        EndpointParams(
+            Region=options.region,
+            UseDualStack=options.use_dual_stack,
+            UseFIPS=options.use_fips,
+            Endpoint=options.endpoint,
+        )
+    )  # noqa: F841
+    import capo_codeartifact.types.package_format
+
+    url = endpoint.url.rstrip("/") + "/v1/package/versions/update_status"
+    params: list[tuple[str, str]] = []
+    if "domain" in input_:
+        params.append(("domain", input_["domain"]))
+    if "domain_owner" in input_:
+        params.append(("domain-owner", input_["domain_owner"]))
+    if "repository" in input_:
+        params.append(("repository", input_["repository"]))
+    if "format" in input_:
+        params.append(
+            (
+                "format",
+                capo_codeartifact.types.package_format.serialize_json(input_["format"]),
+            )
+        )
+    if "namespace" in input_:
+        params.append(("namespace", input_["namespace"]))
+    if "package" in input_:
+        params.append(("package", input_["package"]))
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    body: bytes | None = json.dumps(
+        capo_codeartifact.types.update_package_versions_status_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, "POST", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def update_package_versions_status(
+    options: OperationOptions,
+    input_: capo_codeartifact.types.update_package_versions_status_request.UpdatePackageVersionsStatusRequest,
+) -> tuple[
+    capo_codeartifact.types.update_package_versions_status_result.UpdatePackageVersionsStatusResult,
+    zapros.Response,
+]:
+    response = options.client.handler.handle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            response.read()
+            handle_error(response)
+        return handle_response(response), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_update_package_versions_status(
+    options: AsyncOperationOptions,
+    input_: capo_codeartifact.types.update_package_versions_status_request.UpdatePackageVersionsStatusRequest,
+) -> tuple[
+    capo_codeartifact.types.update_package_versions_status_result.UpdatePackageVersionsStatusResult,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            await response.aread()
+            handle_error(response)
+        return await async_handle_response(response), response
+    except BaseException:
+        await response.aclose()
+        raise

capo_codeartifact/_operations/code_artifact_control_plane_service/update_repository.py

@@ -0,0 +1,188 @@
+"""Generated from Smithy shape ``com.amazonaws.codeartifact#UpdateRepository``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+
+import zapros
+from typing_extensions import Never
+
+import capo_codeartifact._auth._signers
+import capo_codeartifact._auth._sigv4
+import capo_codeartifact._protocol.eventstream
+import capo_codeartifact.errors.access_denied_exception
+import capo_codeartifact.errors.conflict_exception
+import capo_codeartifact.errors.internal_server_exception
+import capo_codeartifact.errors.resource_not_found_exception
+import capo_codeartifact.errors.service_quota_exceeded_exception
+import capo_codeartifact.errors.throttling_exception
+import capo_codeartifact.errors.validation_exception
+import capo_codeartifact.types.repository_description
+import capo_codeartifact.types.update_repository_request
+import capo_codeartifact.types.update_repository_result
+import capo_codeartifact.types.upstream_repository_list
+from capo_codeartifact._protocol.errors import parse_error_metadata_json
+from capo_codeartifact._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_codeartifact._services._pipeline import (
+    AsyncOperationOptions,
+    OperationOptions,
+)
+from capo_codeartifact.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_codeartifact.errors.access_denied_exception.AccessDeniedException.from_json(
+                data, message
+            )
+        case "ConflictException":
+            raise capo_codeartifact.errors.conflict_exception.ConflictException.from_json(
+                data, message
+            )
+        case "InternalServerException":
+            raise capo_codeartifact.errors.internal_server_exception.InternalServerException.from_json(
+                data, message
+            )
+        case "ResourceNotFoundException":
+            raise capo_codeartifact.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data, message
+            )
+        case "ServiceQuotaExceededException":
+            raise capo_codeartifact.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_json(
+                data, message
+            )
+        case "ThrottlingException":
+            raise capo_codeartifact.errors.throttling_exception.ThrottlingException.from_json(
+                data, message
+            )
+        case "ValidationException":
+            raise capo_codeartifact.errors.validation_exception.ValidationException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_codeartifact.types.update_repository_result.UpdateRepositoryResult:
+    out: capo_codeartifact.types.update_repository_result.UpdateRepositoryResult = (
+        capo_codeartifact.types.update_repository_result.deserialize_json(
+            json.loads(response.read())
+        )
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_codeartifact.types.update_repository_result.UpdateRepositoryResult:
+    out: capo_codeartifact.types.update_repository_result.UpdateRepositoryResult = (
+        capo_codeartifact.types.update_repository_result.deserialize_json(
+            json.loads(await response.aread())
+        )
+    )
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> capo_codeartifact._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_codeartifact._auth._sigv4.build_sigv4_auth_scheme(
+                "codeartifact", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_codeartifact._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_codeartifact.types.update_repository_request.UpdateRepositoryRequest,
+) -> 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("/") + "/v1/repository"
+    params: list[tuple[str, str]] = []
+    if "domain" in input_:
+        params.append(("domain", input_["domain"]))
+    if "domain_owner" in input_:
+        params.append(("domain-owner", input_["domain_owner"]))
+    if "repository" in input_:
+        params.append(("repository", input_["repository"]))
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    body: bytes | None = json.dumps(
+        capo_codeartifact.types.update_repository_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 update_repository(
+    options: OperationOptions,
+    input_: capo_codeartifact.types.update_repository_request.UpdateRepositoryRequest,
+) -> tuple[
+    capo_codeartifact.types.update_repository_result.UpdateRepositoryResult,
+    zapros.Response,
+]:
+    response = options.client.handler.handle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            response.read()
+            handle_error(response)
+        return handle_response(response), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_update_repository(
+    options: AsyncOperationOptions,
+    input_: capo_codeartifact.types.update_repository_request.UpdateRepositoryRequest,
+) -> tuple[
+    capo_codeartifact.types.update_repository_result.UpdateRepositoryResult,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            await response.aread()
+            handle_error(response)
+        return await async_handle_response(response), response
+    except BaseException:
+        await response.aclose()
+        raise

capo_codeartifact/_pagination.py

@@ -0,0 +1,21 @@
+"""Runtime helpers for generated paginated ``iter_*`` operation methods."""
+
+from __future__ import annotations
+
+from typing import Any
+
+
+def resolve_path(obj: Any, path: tuple[str, ...]) -> Any:
+    """Walk ``path`` segments on a nested ``dict`` (TypedDict at runtime).
+
+    Returns ``None`` as soon as any intermediate value is missing or non-dict —
+    the generated iter loop uses a falsy token as the terminate signal, so this
+    short-circuit gives the same semantics for dotted paths as for top-level ones.
+    """
+    for key in path:
+        if not isinstance(obj, dict):
+            return None
+        obj = obj.get(key)
+        if obj is None:
+            return None
+    return obj

capo_codeartifact/_protocol/__init__.py

@@ -0,0 +1 @@
+"""Protocol runtime modules. Hand-written; not generated."""

capo_codeartifact/_protocol/errors.py

@@ -0,0 +1,93 @@
+"""Shared service-error runtime.
+
+Hand-written, not regenerated. Helpers for extracting error metadata
+from HTTP error responses.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from zapros import Response
+
+from .xml import fromstring
+
+if TYPE_CHECKING:
+    from xml.etree.ElementTree import Element
+
+
+def find_error_element(root: Element) -> Element:
+    """Return the element holding the error structure's members.
+
+    Handles the three XML error envelopes: a bare ``<Error>`` root
+    (restXml with ``noErrorWrapping``), an ``<ErrorResponse><Error>``
+    wrapper (restXml/awsQuery), and the ec2Query
+    ``<Response><Errors><Error>`` envelope. Falls back to ``root``
+    when no ``<Error>`` element is found.
+    """
+    if root.tag.endswith("Error"):
+        return root
+    for path in ("Error", "Errors/Error"):
+        err = root.find(path)
+        if err is not None:
+            return err
+    return root
+
+
+def parse_error_metadata(root: Element) -> tuple[str | None, str | None]:
+    """Return ``(code, message)`` from an XML error envelope.
+
+    Accepts any envelope understood by :func:`find_error_element`.
+    Missing children yield ``None``.
+    """
+    err = find_error_element(root)
+    code_el = err.find("Code")
+    msg_el = err.find("Message")
+    code = code_el.text if code_el is not None else None
+    message = msg_el.text if msg_el is not None else None
+    return code, message
+
+
+def parse_error_metadata_json(
+    response: Response, data: dict
+) -> tuple[str | None, str | None]:
+    """Return ``(code, message)`` from a restJson1 error response.
+
+    Code precedence: the ``X-Amzn-Errortype`` response header, then the
+    ``__type`` body field, then ``code``. The raw value is normalized by
+    dropping a trailing ``:uri`` suffix first, then a ``prefix#``
+    namespace — in that order, so a ``#`` inside the uri suffix cannot
+    hijack the code. Message comes from ``message`` or ``Message``.
+    Missing values yield ``None``.
+    """
+    code = (
+        response.headers.get("X-Amzn-Errortype")
+        or data.get("__type")
+        or data.get("code")
+    )
+    if code is not None:
+        code = code.split(":", 1)[0].rsplit("#", 1)[-1]
+    message = data.get("message") or data.get("Message")
+    return code, message
+
+
+def is_xml_error_body(body: bytes) -> bool:
+    """Whether a 2xx response body is really an XML error document.
+
+    S3 answers CopyObject, UploadPartCopy and CompleteMultipartUpload
+    with ``200 OK`` before the operation finishes; a failure after that
+    point is reported as an ``<Error>`` body (or an empty body) on the
+    200 response. Official SDKs check the body for exactly these
+    operations.
+    """
+    if not body:
+        return True
+    return fromstring(body).tag == "Error"
+
+
+__all__ = [
+    "find_error_element",
+    "is_xml_error_body",
+    "parse_error_metadata",
+    "parse_error_metadata_json",
+]