capo-devops-agent 0.1.0__py3-none-any.whl

capo_devops_agent/_operations/dev_ops_agent/describe_private_connection.py

@@ -0,0 +1,185 @@
+"""Generated from Smithy shape ``com.amazonaws.devopsagent#DescribePrivateConnection``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_devops_agent._auth._signers
+import capo_devops_agent._auth._sigv4
+import capo_devops_agent._protocol.eventstream
+import capo_devops_agent.errors.access_denied_exception
+import capo_devops_agent.errors.conflict_exception
+import capo_devops_agent.errors.content_size_exceeded_exception
+import capo_devops_agent.errors.internal_server_exception
+import capo_devops_agent.errors.invalid_parameter_exception
+import capo_devops_agent.errors.resource_not_found_exception
+import capo_devops_agent.errors.service_quota_exceeded_exception
+import capo_devops_agent.errors.throttling_exception
+import capo_devops_agent.errors.validation_exception
+import capo_devops_agent.types.describe_private_connection_input
+import capo_devops_agent.types.describe_private_connection_output
+import capo_devops_agent.types.private_connection_status
+import capo_devops_agent.types.private_connection_type
+import capo_devops_agent.types.resource_config_dns_resolution
+import capo_devops_agent.types.tags
+from capo_devops_agent._protocol.errors import parse_error_metadata_json
+from capo_devops_agent._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_devops_agent._services._pipeline import (
+    AsyncOperationOptions,
+    OperationOptions,
+)
+from capo_devops_agent.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_devops_agent.errors.access_denied_exception.AccessDeniedException.from_json(
+                data, message
+            )
+        case "ConflictException":
+            raise capo_devops_agent.errors.conflict_exception.ConflictException.from_json(
+                data, message
+            )
+        case "ContentSizeExceededException":
+            raise capo_devops_agent.errors.content_size_exceeded_exception.ContentSizeExceededException.from_json(
+                data, message
+            )
+        case "InternalServerException":
+            raise capo_devops_agent.errors.internal_server_exception.InternalServerException.from_json(
+                data, message
+            )
+        case "InvalidParameterException":
+            raise capo_devops_agent.errors.invalid_parameter_exception.InvalidParameterException.from_json(
+                data, message
+            )
+        case "ResourceNotFoundException":
+            raise capo_devops_agent.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data, message
+            )
+        case "ServiceQuotaExceededException":
+            raise capo_devops_agent.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_json(
+                data, message
+            )
+        case "ThrottlingException":
+            raise capo_devops_agent.errors.throttling_exception.ThrottlingException.from_json(
+                data, message
+            )
+        case "ValidationException":
+            raise capo_devops_agent.errors.validation_exception.ValidationException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_devops_agent.types.describe_private_connection_output.DescribePrivateConnectionOutput:
+    out: capo_devops_agent.types.describe_private_connection_output.DescribePrivateConnectionOutput = capo_devops_agent.types.describe_private_connection_output.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_devops_agent.types.describe_private_connection_output.DescribePrivateConnectionOutput:
+    out: capo_devops_agent.types.describe_private_connection_output.DescribePrivateConnectionOutput = capo_devops_agent.types.describe_private_connection_output.deserialize_json(
+        json.loads(await response.aread())
+    )
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> capo_devops_agent._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_devops_agent._auth._sigv4.build_sigv4_auth_scheme(
+                "aidevops", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_devops_agent._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_devops_agent.types.describe_private_connection_input.DescribePrivateConnectionInput,
+) -> zapros.Request:
+    endpoint = resolve(
+        EndpointParams(
+            UseFIPS=options.use_fips, Endpoint=options.endpoint, Region=options.region
+        )
+    )  # noqa: F841
+    url = endpoint.url.rstrip("/") + "/v1/private-connections/{name}"
+    url = url.replace("{name}", quote(input_["name"], safe=""))
+    params: list[tuple[str, str]] = []
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    body: bytes | None = b""
+    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, "GET", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def describe_private_connection(
+    options: OperationOptions,
+    input_: capo_devops_agent.types.describe_private_connection_input.DescribePrivateConnectionInput,
+) -> tuple[
+    capo_devops_agent.types.describe_private_connection_output.DescribePrivateConnectionOutput,
+    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_describe_private_connection(
+    options: AsyncOperationOptions,
+    input_: capo_devops_agent.types.describe_private_connection_input.DescribePrivateConnectionInput,
+) -> tuple[
+    capo_devops_agent.types.describe_private_connection_output.DescribePrivateConnectionOutput,
+    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_devops_agent/_operations/dev_ops_agent/disable_operator_app.py

@@ -0,0 +1,168 @@
+"""Generated from Smithy shape ``com.amazonaws.devopsagent#DisableOperatorApp``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_devops_agent._auth._signers
+import capo_devops_agent._auth._sigv4
+import capo_devops_agent._protocol.eventstream
+import capo_devops_agent.errors.access_denied_exception
+import capo_devops_agent.errors.conflict_exception
+import capo_devops_agent.errors.content_size_exceeded_exception
+import capo_devops_agent.errors.identity_center_service_exception
+import capo_devops_agent.errors.internal_server_exception
+import capo_devops_agent.errors.invalid_parameter_exception
+import capo_devops_agent.errors.resource_not_found_exception
+import capo_devops_agent.errors.service_quota_exceeded_exception
+import capo_devops_agent.errors.throttling_exception
+import capo_devops_agent.errors.validation_exception
+import capo_devops_agent.types.auth_flow
+import capo_devops_agent.types.disable_operator_app_input
+from capo_devops_agent._protocol.errors import parse_error_metadata_json
+from capo_devops_agent._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_devops_agent._services._pipeline import (
+    AsyncOperationOptions,
+    OperationOptions,
+)
+from capo_devops_agent.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_devops_agent.errors.access_denied_exception.AccessDeniedException.from_json(
+                data, message
+            )
+        case "ConflictException":
+            raise capo_devops_agent.errors.conflict_exception.ConflictException.from_json(
+                data, message
+            )
+        case "ContentSizeExceededException":
+            raise capo_devops_agent.errors.content_size_exceeded_exception.ContentSizeExceededException.from_json(
+                data, message
+            )
+        case "InternalServerException":
+            raise capo_devops_agent.errors.internal_server_exception.InternalServerException.from_json(
+                data, message
+            )
+        case "InvalidParameterException":
+            raise capo_devops_agent.errors.invalid_parameter_exception.InvalidParameterException.from_json(
+                data, message
+            )
+        case "ResourceNotFoundException":
+            raise capo_devops_agent.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data, message
+            )
+        case "ServiceQuotaExceededException":
+            raise capo_devops_agent.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_json(
+                data, message
+            )
+        case "ThrottlingException":
+            raise capo_devops_agent.errors.throttling_exception.ThrottlingException.from_json(
+                data, message
+            )
+        case "IdentityCenterServiceException":
+            raise capo_devops_agent.errors.identity_center_service_exception.IdentityCenterServiceException.from_json(
+                data, message
+            )
+        case "ValidationException":
+            raise capo_devops_agent.errors.validation_exception.ValidationException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> capo_devops_agent._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_devops_agent._auth._sigv4.build_sigv4_auth_scheme(
+                "aidevops", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_devops_agent._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_devops_agent.types.disable_operator_app_input.DisableOperatorAppInput,
+) -> zapros.Request:
+    endpoint = resolve(
+        EndpointParams(
+            UseFIPS=options.use_fips, Endpoint=options.endpoint, Region=options.region
+        )
+    )  # noqa: F841
+    import capo_devops_agent.types.auth_flow
+
+    url = endpoint.url.rstrip("/") + "/v1/agentspaces/{agentSpaceId}/operator"
+    url = url.replace("{agentSpaceId}", quote(input_["agent_space_id"], safe=""))
+    params: list[tuple[str, str]] = []
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    if "auth_flow" in input_:
+        headers["x-amzn-app-auth-flow"] = (
+            capo_devops_agent.types.auth_flow.serialize_json(input_["auth_flow"])
+        )
+    body: bytes | None = b""
+    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, "DELETE", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def disable_operator_app(
+    options: OperationOptions,
+    input_: capo_devops_agent.types.disable_operator_app_input.DisableOperatorAppInput,
+) -> tuple[None, zapros.Response]:
+    response = options.client.handler.handle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            response.read()
+            handle_error(response)
+        return None, response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_disable_operator_app(
+    options: AsyncOperationOptions,
+    input_: capo_devops_agent.types.disable_operator_app_input.DisableOperatorAppInput,
+) -> tuple[None, zapros.Response]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 300:
+            await response.aread()
+            handle_error(response)
+        return None, response
+    except BaseException:
+        await response.aclose()
+        raise

capo_devops_agent/_operations/dev_ops_agent/disassociate_service.py

@@ -0,0 +1,181 @@
+"""Generated from Smithy shape ``com.amazonaws.devopsagent#DisassociateService``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_devops_agent._auth._signers
+import capo_devops_agent._auth._sigv4
+import capo_devops_agent._protocol.eventstream
+import capo_devops_agent.errors.access_denied_exception
+import capo_devops_agent.errors.conflict_exception
+import capo_devops_agent.errors.content_size_exceeded_exception
+import capo_devops_agent.errors.internal_server_exception
+import capo_devops_agent.errors.invalid_parameter_exception
+import capo_devops_agent.errors.resource_not_found_exception
+import capo_devops_agent.errors.service_quota_exceeded_exception
+import capo_devops_agent.errors.throttling_exception
+import capo_devops_agent.errors.validation_exception
+import capo_devops_agent.types.disassociate_service_input
+import capo_devops_agent.types.disassociate_service_output
+from capo_devops_agent._protocol.errors import parse_error_metadata_json
+from capo_devops_agent._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_devops_agent._services._pipeline import (
+    AsyncOperationOptions,
+    OperationOptions,
+)
+from capo_devops_agent.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_devops_agent.errors.access_denied_exception.AccessDeniedException.from_json(
+                data, message
+            )
+        case "ConflictException":
+            raise capo_devops_agent.errors.conflict_exception.ConflictException.from_json(
+                data, message
+            )
+        case "ContentSizeExceededException":
+            raise capo_devops_agent.errors.content_size_exceeded_exception.ContentSizeExceededException.from_json(
+                data, message
+            )
+        case "InternalServerException":
+            raise capo_devops_agent.errors.internal_server_exception.InternalServerException.from_json(
+                data, message
+            )
+        case "InvalidParameterException":
+            raise capo_devops_agent.errors.invalid_parameter_exception.InvalidParameterException.from_json(
+                data, message
+            )
+        case "ResourceNotFoundException":
+            raise capo_devops_agent.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data, message
+            )
+        case "ServiceQuotaExceededException":
+            raise capo_devops_agent.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_json(
+                data, message
+            )
+        case "ThrottlingException":
+            raise capo_devops_agent.errors.throttling_exception.ThrottlingException.from_json(
+                data, message
+            )
+        case "ValidationException":
+            raise capo_devops_agent.errors.validation_exception.ValidationException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_devops_agent.types.disassociate_service_output.DisassociateServiceOutput:
+    out: capo_devops_agent.types.disassociate_service_output.DisassociateServiceOutput = {}  # type: ignore[typeddict-item]
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_devops_agent.types.disassociate_service_output.DisassociateServiceOutput:
+    out: capo_devops_agent.types.disassociate_service_output.DisassociateServiceOutput = {}  # type: ignore[typeddict-item]
+    return out
+
+
+def get_signer(
+    options: AsyncOperationOptions | OperationOptions,
+    auth_schemes: list[dict[str, Any]] | None = None,
+) -> capo_devops_agent._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_devops_agent._auth._sigv4.build_sigv4_auth_scheme(
+                "aidevops", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_devops_agent._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_devops_agent.types.disassociate_service_input.DisassociateServiceInput,
+) -> zapros.Request:
+    endpoint = resolve(
+        EndpointParams(
+            UseFIPS=options.use_fips, Endpoint=options.endpoint, Region=options.region
+        )
+    )  # noqa: F841
+    url = (
+        endpoint.url.rstrip("/")
+        + "/v1/agentspaces/{agentSpaceId}/associations/{associationId}"
+    )
+    url = url.replace("{agentSpaceId}", quote(input_["agent_space_id"], safe=""))
+    url = url.replace("{associationId}", quote(input_["association_id"], safe=""))
+    params: list[tuple[str, str]] = []
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    body: bytes | None = b""
+    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, "DELETE", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def disassociate_service(
+    options: OperationOptions,
+    input_: capo_devops_agent.types.disassociate_service_input.DisassociateServiceInput,
+) -> tuple[
+    capo_devops_agent.types.disassociate_service_output.DisassociateServiceOutput,
+    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_disassociate_service(
+    options: AsyncOperationOptions,
+    input_: capo_devops_agent.types.disassociate_service_input.DisassociateServiceInput,
+) -> tuple[
+    capo_devops_agent.types.disassociate_service_output.DisassociateServiceOutput,
+    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