capo-lambda 0.5.0__py3-none-any.whl

capo_lambda/_operations/aws_gir_api_service/put_function_concurrency.py

@@ -0,0 +1,149 @@
+"""Generated from Smithy shape ``com.amazonaws.lambda#PutFunctionConcurrency``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_lambda._auth._signers
+import capo_lambda._auth._sigv4
+import capo_lambda.errors.invalid_parameter_value_exception
+import capo_lambda.errors.resource_conflict_exception
+import capo_lambda.errors.resource_not_found_exception
+import capo_lambda.errors.service_exception
+import capo_lambda.errors.too_many_requests_exception
+import capo_lambda.types.concurrency
+import capo_lambda.types.put_function_concurrency_request
+from capo_lambda._protocol.errors import parse_error_metadata_json
+from capo_lambda._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_lambda._services._pipeline import AsyncOperationOptions, OperationOptions
+from capo_lambda.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 "InvalidParameterValueException":
+            raise capo_lambda.errors.invalid_parameter_value_exception.InvalidParameterValueException.from_json(
+                data
+            )
+        case "ResourceConflictException":
+            raise capo_lambda.errors.resource_conflict_exception.ResourceConflictException.from_json(
+                data
+            )
+        case "ResourceNotFoundException":
+            raise capo_lambda.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data
+            )
+        case "ServiceException":
+            raise capo_lambda.errors.service_exception.ServiceException.from_json(data)
+        case "TooManyRequestsException":
+            raise capo_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,
+) -> capo_lambda.types.concurrency.Concurrency:
+    out: capo_lambda.types.concurrency.Concurrency = (
+        capo_lambda.types.concurrency.deserialize_json(json.loads(response.read()))
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_lambda.types.concurrency.Concurrency:
+    out: capo_lambda.types.concurrency.Concurrency = (
+        capo_lambda.types.concurrency.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._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:
+        sigv4_config = (
+            name_to_schema.get("sigv4")
+            or name_to_schema.get("sigv4a")
+            or name_to_schema.get("sigv4-s3express")
+            or capo_lambda._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region
+            )
+        )
+        if sigv4_config is not None:
+            return capo_lambda._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.types.put_function_concurrency_request.PutFunctionConcurrencyRequest,
+) -> 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("/") + "/2017-10-31/functions/{FunctionName}/concurrency"
+    url = url.replace("{FunctionName}", quote(str(input_["function_name"]), safe=""))
+    params: dict[str, str] = {}
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    body: bytes | None = json.dumps(
+        capo_lambda.types.put_function_concurrency_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, "PUT", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def put_function_concurrency(
+    options: OperationOptions,
+    input_: capo_lambda.types.put_function_concurrency_request.PutFunctionConcurrencyRequest,
+) -> tuple[capo_lambda.types.concurrency.Concurrency, 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), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_put_function_concurrency(
+    options: AsyncOperationOptions,
+    input_: capo_lambda.types.put_function_concurrency_request.PutFunctionConcurrencyRequest,
+) -> tuple[capo_lambda.types.concurrency.Concurrency, zapros.Response]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 400:
+            await response.aread()
+            handle_error(response)
+        return await async_handle_response(response), response
+    except BaseException:
+        await response.aclose()
+        raise

capo_lambda/_operations/aws_gir_api_service/put_function_event_invoke_config.py

@@ -0,0 +1,166 @@
+"""Generated from Smithy shape ``com.amazonaws.lambda#PutFunctionEventInvokeConfig``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_lambda._auth._signers
+import capo_lambda._auth._sigv4
+import capo_lambda.errors.invalid_parameter_value_exception
+import capo_lambda.errors.resource_conflict_exception
+import capo_lambda.errors.resource_not_found_exception
+import capo_lambda.errors.service_exception
+import capo_lambda.errors.too_many_requests_exception
+import capo_lambda.types.date
+import capo_lambda.types.destination_config
+import capo_lambda.types.function_event_invoke_config
+import capo_lambda.types.put_function_event_invoke_config_request
+from capo_lambda._protocol.errors import parse_error_metadata_json
+from capo_lambda._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_lambda._services._pipeline import AsyncOperationOptions, OperationOptions
+from capo_lambda.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 "InvalidParameterValueException":
+            raise capo_lambda.errors.invalid_parameter_value_exception.InvalidParameterValueException.from_json(
+                data
+            )
+        case "ResourceConflictException":
+            raise capo_lambda.errors.resource_conflict_exception.ResourceConflictException.from_json(
+                data
+            )
+        case "ResourceNotFoundException":
+            raise capo_lambda.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data
+            )
+        case "ServiceException":
+            raise capo_lambda.errors.service_exception.ServiceException.from_json(data)
+        case "TooManyRequestsException":
+            raise capo_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,
+) -> capo_lambda.types.function_event_invoke_config.FunctionEventInvokeConfig:
+    out: capo_lambda.types.function_event_invoke_config.FunctionEventInvokeConfig = (
+        capo_lambda.types.function_event_invoke_config.deserialize_json(
+            json.loads(response.read())
+        )
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_lambda.types.function_event_invoke_config.FunctionEventInvokeConfig:
+    out: capo_lambda.types.function_event_invoke_config.FunctionEventInvokeConfig = (
+        capo_lambda.types.function_event_invoke_config.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._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:
+        sigv4_config = (
+            name_to_schema.get("sigv4")
+            or name_to_schema.get("sigv4a")
+            or name_to_schema.get("sigv4-s3express")
+            or capo_lambda._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region
+            )
+        )
+        if sigv4_config is not None:
+            return capo_lambda._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.types.put_function_event_invoke_config_request.PutFunctionEventInvokeConfigRequest,
+) -> 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("/")
+        + "/2019-09-25/functions/{FunctionName}/event-invoke-config"
+    )
+    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()}
+    body: bytes | None = json.dumps(
+        capo_lambda.types.put_function_event_invoke_config_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, "PUT", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def put_function_event_invoke_config(
+    options: OperationOptions,
+    input_: capo_lambda.types.put_function_event_invoke_config_request.PutFunctionEventInvokeConfigRequest,
+) -> tuple[
+    capo_lambda.types.function_event_invoke_config.FunctionEventInvokeConfig,
+    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), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_put_function_event_invoke_config(
+    options: AsyncOperationOptions,
+    input_: capo_lambda.types.put_function_event_invoke_config_request.PutFunctionEventInvokeConfigRequest,
+) -> tuple[
+    capo_lambda.types.function_event_invoke_config.FunctionEventInvokeConfig,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 400:
+            await response.aread()
+            handle_error(response)
+        return await async_handle_response(response), response
+    except BaseException:
+        await response.aclose()
+        raise

capo_lambda/_operations/aws_gir_api_service/put_function_recursion_config.py

@@ -0,0 +1,157 @@
+"""Generated from Smithy shape ``com.amazonaws.lambda#PutFunctionRecursionConfig``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_lambda._auth._signers
+import capo_lambda._auth._sigv4
+import capo_lambda.errors.invalid_parameter_value_exception
+import capo_lambda.errors.resource_conflict_exception
+import capo_lambda.errors.resource_not_found_exception
+import capo_lambda.errors.service_exception
+import capo_lambda.errors.too_many_requests_exception
+import capo_lambda.types.put_function_recursion_config_request
+import capo_lambda.types.put_function_recursion_config_response
+import capo_lambda.types.recursive_loop
+from capo_lambda._protocol.errors import parse_error_metadata_json
+from capo_lambda._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_lambda._services._pipeline import AsyncOperationOptions, OperationOptions
+from capo_lambda.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 "InvalidParameterValueException":
+            raise capo_lambda.errors.invalid_parameter_value_exception.InvalidParameterValueException.from_json(
+                data
+            )
+        case "ResourceConflictException":
+            raise capo_lambda.errors.resource_conflict_exception.ResourceConflictException.from_json(
+                data
+            )
+        case "ResourceNotFoundException":
+            raise capo_lambda.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data
+            )
+        case "ServiceException":
+            raise capo_lambda.errors.service_exception.ServiceException.from_json(data)
+        case "TooManyRequestsException":
+            raise capo_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,
+) -> capo_lambda.types.put_function_recursion_config_response.PutFunctionRecursionConfigResponse:
+    out: capo_lambda.types.put_function_recursion_config_response.PutFunctionRecursionConfigResponse = capo_lambda.types.put_function_recursion_config_response.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_lambda.types.put_function_recursion_config_response.PutFunctionRecursionConfigResponse:
+    out: capo_lambda.types.put_function_recursion_config_response.PutFunctionRecursionConfigResponse = capo_lambda.types.put_function_recursion_config_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._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:
+        sigv4_config = (
+            name_to_schema.get("sigv4")
+            or name_to_schema.get("sigv4a")
+            or name_to_schema.get("sigv4-s3express")
+            or capo_lambda._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region
+            )
+        )
+        if sigv4_config is not None:
+            return capo_lambda._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.types.put_function_recursion_config_request.PutFunctionRecursionConfigRequest,
+) -> 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("/")
+        + "/2024-08-31/functions/{FunctionName}/recursion-config"
+    )
+    url = url.replace("{FunctionName}", quote(str(input_["function_name"]), safe=""))
+    params: dict[str, str] = {}
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    body: bytes | None = json.dumps(
+        capo_lambda.types.put_function_recursion_config_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, "PUT", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def put_function_recursion_config(
+    options: OperationOptions,
+    input_: capo_lambda.types.put_function_recursion_config_request.PutFunctionRecursionConfigRequest,
+) -> tuple[
+    capo_lambda.types.put_function_recursion_config_response.PutFunctionRecursionConfigResponse,
+    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), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_put_function_recursion_config(
+    options: AsyncOperationOptions,
+    input_: capo_lambda.types.put_function_recursion_config_request.PutFunctionRecursionConfigRequest,
+) -> tuple[
+    capo_lambda.types.put_function_recursion_config_response.PutFunctionRecursionConfigResponse,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 400:
+            await response.aread()
+            handle_error(response)
+        return await async_handle_response(response), response
+    except BaseException:
+        await response.aclose()
+        raise

capo_lambda/_operations/aws_gir_api_service/put_function_scaling_config.py

@@ -0,0 +1,160 @@
+"""Generated from Smithy shape ``com.amazonaws.lambda#PutFunctionScalingConfig``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_lambda._auth._signers
+import capo_lambda._auth._sigv4
+import capo_lambda.errors.invalid_parameter_value_exception
+import capo_lambda.errors.resource_conflict_exception
+import capo_lambda.errors.resource_not_found_exception
+import capo_lambda.errors.service_exception
+import capo_lambda.errors.too_many_requests_exception
+import capo_lambda.types.function_scaling_config
+import capo_lambda.types.put_function_scaling_config_request
+import capo_lambda.types.put_function_scaling_config_response
+import capo_lambda.types.state
+from capo_lambda._protocol.errors import parse_error_metadata_json
+from capo_lambda._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_lambda._services._pipeline import AsyncOperationOptions, OperationOptions
+from capo_lambda.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 "InvalidParameterValueException":
+            raise capo_lambda.errors.invalid_parameter_value_exception.InvalidParameterValueException.from_json(
+                data
+            )
+        case "ResourceConflictException":
+            raise capo_lambda.errors.resource_conflict_exception.ResourceConflictException.from_json(
+                data
+            )
+        case "ResourceNotFoundException":
+            raise capo_lambda.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
+                data
+            )
+        case "ServiceException":
+            raise capo_lambda.errors.service_exception.ServiceException.from_json(data)
+        case "TooManyRequestsException":
+            raise capo_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,
+) -> capo_lambda.types.put_function_scaling_config_response.PutFunctionScalingConfigResponse:
+    out: capo_lambda.types.put_function_scaling_config_response.PutFunctionScalingConfigResponse = capo_lambda.types.put_function_scaling_config_response.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_lambda.types.put_function_scaling_config_response.PutFunctionScalingConfigResponse:
+    out: capo_lambda.types.put_function_scaling_config_response.PutFunctionScalingConfigResponse = capo_lambda.types.put_function_scaling_config_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._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:
+        sigv4_config = (
+            name_to_schema.get("sigv4")
+            or name_to_schema.get("sigv4a")
+            or name_to_schema.get("sigv4-s3express")
+            or capo_lambda._auth._sigv4.build_sigv4_auth_scheme(
+                "lambda", options.region
+            )
+        )
+        if sigv4_config is not None:
+            return capo_lambda._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.types.put_function_scaling_config_request.PutFunctionScalingConfigRequest,
+) -> 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-11-30/functions/{FunctionName}/function-scaling-config"
+    )
+    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()}
+    body: bytes | None = json.dumps(
+        capo_lambda.types.put_function_scaling_config_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, "PUT", headers=headers, body=body, context={"signer": signer}
+    )
+
+
+def put_function_scaling_config(
+    options: OperationOptions,
+    input_: capo_lambda.types.put_function_scaling_config_request.PutFunctionScalingConfigRequest,
+) -> tuple[
+    capo_lambda.types.put_function_scaling_config_response.PutFunctionScalingConfigResponse,
+    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), response
+    except BaseException:
+        response.close()
+        raise
+
+
+async def async_put_function_scaling_config(
+    options: AsyncOperationOptions,
+    input_: capo_lambda.types.put_function_scaling_config_request.PutFunctionScalingConfigRequest,
+) -> tuple[
+    capo_lambda.types.put_function_scaling_config_response.PutFunctionScalingConfigResponse,
+    zapros.Response,
+]:
+    response = await options.client.handler.ahandle(build_request(options, input_))
+    try:
+        if response.status >= 400:
+            await response.aread()
+            handle_error(response)
+        return await async_handle_response(response), response
+    except BaseException:
+        await response.aclose()
+        raise