capo-workdocs 0.1.0__py3-none-any.whl

capo_workdocs/_operations/aws_gorilla_boy_service/get_folder_path.py

@@ -0,0 +1,172 @@
+"""Generated from Smithy shape ``com.amazonaws.workdocs#GetFolderPath``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+from urllib.parse import quote
+
+import zapros
+from typing_extensions import Never
+
+import capo_workdocs._auth._signers
+import capo_workdocs._auth._sigv4
+import capo_workdocs._protocol.eventstream
+import capo_workdocs.errors.entity_not_exists_exception
+import capo_workdocs.errors.failed_dependency_exception
+import capo_workdocs.errors.service_unavailable_exception
+import capo_workdocs.errors.unauthorized_operation_exception
+import capo_workdocs.errors.unauthorized_resource_access_exception
+import capo_workdocs.types.get_folder_path_request
+import capo_workdocs.types.get_folder_path_response
+import capo_workdocs.types.resource_path
+from capo_workdocs._protocol.errors import parse_error_metadata_json
+from capo_workdocs._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_workdocs._services._pipeline import AsyncOperationOptions, OperationOptions
+from capo_workdocs.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 "EntityNotExistsException":
+            raise capo_workdocs.errors.entity_not_exists_exception.EntityNotExistsException.from_json(
+                data, message
+            )
+        case "FailedDependencyException":
+            raise capo_workdocs.errors.failed_dependency_exception.FailedDependencyException.from_json(
+                data, message
+            )
+        case "ServiceUnavailableException":
+            raise capo_workdocs.errors.service_unavailable_exception.ServiceUnavailableException.from_json(
+                data, message
+            )
+        case "UnauthorizedOperationException":
+            raise capo_workdocs.errors.unauthorized_operation_exception.UnauthorizedOperationException.from_json(
+                data, message
+            )
+        case "UnauthorizedResourceAccessException":
+            raise capo_workdocs.errors.unauthorized_resource_access_exception.UnauthorizedResourceAccessException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_workdocs.types.get_folder_path_response.GetFolderPathResponse:
+    out: capo_workdocs.types.get_folder_path_response.GetFolderPathResponse = (
+        capo_workdocs.types.get_folder_path_response.deserialize_json(
+            json.loads(response.read())
+        )
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_workdocs.types.get_folder_path_response.GetFolderPathResponse:
+    out: capo_workdocs.types.get_folder_path_response.GetFolderPathResponse = (
+        capo_workdocs.types.get_folder_path_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_workdocs._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_workdocs._auth._sigv4.build_sigv4_auth_scheme(
+                "workdocs", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_workdocs._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_workdocs.types.get_folder_path_request.GetFolderPathRequest,
+) -> 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("/") + "/api/v1/folders/{FolderId}/path"
+    url = url.replace("{FolderId}", quote(input_["folder_id"], safe=""))
+    params: list[tuple[str, str]] = []
+    if "limit" in input_:
+        params.append(("limit", str(input_["limit"])))
+    if "fields" in input_:
+        params.append(("fields", input_["fields"]))
+    if "marker" in input_:
+        params.append(("marker", input_["marker"]))
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    if "authentication_token" in input_:
+        headers["Authentication"] = input_["authentication_token"]
+    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 get_folder_path(
+    options: OperationOptions,
+    input_: capo_workdocs.types.get_folder_path_request.GetFolderPathRequest,
+) -> tuple[
+    capo_workdocs.types.get_folder_path_response.GetFolderPathResponse, 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_get_folder_path(
+    options: AsyncOperationOptions,
+    input_: capo_workdocs.types.get_folder_path_request.GetFolderPathRequest,
+) -> tuple[
+    capo_workdocs.types.get_folder_path_response.GetFolderPathResponse, 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_workdocs/_operations/aws_gorilla_boy_service/get_resources.py

@@ -0,0 +1,183 @@
+"""Generated from Smithy shape ``com.amazonaws.workdocs#GetResources``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+
+import zapros
+from typing_extensions import Never
+
+import capo_workdocs._auth._signers
+import capo_workdocs._auth._sigv4
+import capo_workdocs._protocol.eventstream
+import capo_workdocs.errors.failed_dependency_exception
+import capo_workdocs.errors.invalid_argument_exception
+import capo_workdocs.errors.service_unavailable_exception
+import capo_workdocs.errors.unauthorized_operation_exception
+import capo_workdocs.errors.unauthorized_resource_access_exception
+import capo_workdocs.types.document_metadata_list
+import capo_workdocs.types.folder_metadata_list
+import capo_workdocs.types.get_resources_request
+import capo_workdocs.types.get_resources_response
+import capo_workdocs.types.resource_collection_type
+from capo_workdocs._protocol.errors import parse_error_metadata_json
+from capo_workdocs._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_workdocs._services._pipeline import AsyncOperationOptions, OperationOptions
+from capo_workdocs.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 "FailedDependencyException":
+            raise capo_workdocs.errors.failed_dependency_exception.FailedDependencyException.from_json(
+                data, message
+            )
+        case "InvalidArgumentException":
+            raise capo_workdocs.errors.invalid_argument_exception.InvalidArgumentException.from_json(
+                data, message
+            )
+        case "ServiceUnavailableException":
+            raise capo_workdocs.errors.service_unavailable_exception.ServiceUnavailableException.from_json(
+                data, message
+            )
+        case "UnauthorizedOperationException":
+            raise capo_workdocs.errors.unauthorized_operation_exception.UnauthorizedOperationException.from_json(
+                data, message
+            )
+        case "UnauthorizedResourceAccessException":
+            raise capo_workdocs.errors.unauthorized_resource_access_exception.UnauthorizedResourceAccessException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_workdocs.types.get_resources_response.GetResourcesResponse:
+    out: capo_workdocs.types.get_resources_response.GetResourcesResponse = (
+        capo_workdocs.types.get_resources_response.deserialize_json(
+            json.loads(response.read())
+        )
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_workdocs.types.get_resources_response.GetResourcesResponse:
+    out: capo_workdocs.types.get_resources_response.GetResourcesResponse = (
+        capo_workdocs.types.get_resources_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_workdocs._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_workdocs._auth._sigv4.build_sigv4_auth_scheme(
+                "workdocs", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_workdocs._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_workdocs.types.get_resources_request.GetResourcesRequest,
+) -> zapros.Request:
+    endpoint = resolve(
+        EndpointParams(
+            Region=options.region,
+            UseDualStack=options.use_dual_stack,
+            UseFIPS=options.use_fips,
+            Endpoint=options.endpoint,
+        )
+    )  # noqa: F841
+    import capo_workdocs.types.resource_collection_type
+
+    url = endpoint.url.rstrip("/") + "/api/v1/resources"
+    params: list[tuple[str, str]] = []
+    if "user_id" in input_:
+        params.append(("userId", input_["user_id"]))
+    if "collection_type" in input_:
+        params.append(
+            (
+                "collectionType",
+                capo_workdocs.types.resource_collection_type.serialize_json(
+                    input_["collection_type"]
+                ),
+            )
+        )
+    if "limit" in input_:
+        params.append(("limit", str(input_["limit"])))
+    if "marker" in input_:
+        params.append(("marker", input_["marker"]))
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    if "authentication_token" in input_:
+        headers["Authentication"] = input_["authentication_token"]
+    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 get_resources(
+    options: OperationOptions,
+    input_: capo_workdocs.types.get_resources_request.GetResourcesRequest,
+) -> tuple[
+    capo_workdocs.types.get_resources_response.GetResourcesResponse, 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_get_resources(
+    options: AsyncOperationOptions,
+    input_: capo_workdocs.types.get_resources_request.GetResourcesRequest,
+) -> tuple[
+    capo_workdocs.types.get_resources_response.GetResourcesResponse, 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_workdocs/_operations/aws_gorilla_boy_service/initiate_document_version_upload.py

@@ -0,0 +1,215 @@
+"""Generated from Smithy shape ``com.amazonaws.workdocs#InitiateDocumentVersionUpload``."""
+
+from __future__ import annotations
+
+import json
+from typing import Any
+
+import zapros
+from typing_extensions import Never
+
+import capo_workdocs._auth._signers
+import capo_workdocs._auth._sigv4
+import capo_workdocs._protocol.eventstream
+import capo_workdocs.errors.draft_upload_out_of_sync_exception
+import capo_workdocs.errors.entity_already_exists_exception
+import capo_workdocs.errors.entity_not_exists_exception
+import capo_workdocs.errors.failed_dependency_exception
+import capo_workdocs.errors.invalid_argument_exception
+import capo_workdocs.errors.invalid_password_exception
+import capo_workdocs.errors.limit_exceeded_exception
+import capo_workdocs.errors.prohibited_state_exception
+import capo_workdocs.errors.resource_already_checked_out_exception
+import capo_workdocs.errors.service_unavailable_exception
+import capo_workdocs.errors.storage_limit_exceeded_exception
+import capo_workdocs.errors.storage_limit_will_exceed_exception
+import capo_workdocs.errors.unauthorized_operation_exception
+import capo_workdocs.errors.unauthorized_resource_access_exception
+import capo_workdocs.types.document_metadata
+import capo_workdocs.types.initiate_document_version_upload_request
+import capo_workdocs.types.initiate_document_version_upload_response
+import capo_workdocs.types.timestamp_type
+import capo_workdocs.types.upload_metadata
+from capo_workdocs._protocol.errors import parse_error_metadata_json
+from capo_workdocs._rule_engine._endpoint_rule_set import EndpointParams, resolve
+from capo_workdocs._services._pipeline import AsyncOperationOptions, OperationOptions
+from capo_workdocs.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 "DraftUploadOutOfSyncException":
+            raise capo_workdocs.errors.draft_upload_out_of_sync_exception.DraftUploadOutOfSyncException.from_json(
+                data, message
+            )
+        case "EntityAlreadyExistsException":
+            raise capo_workdocs.errors.entity_already_exists_exception.EntityAlreadyExistsException.from_json(
+                data, message
+            )
+        case "EntityNotExistsException":
+            raise capo_workdocs.errors.entity_not_exists_exception.EntityNotExistsException.from_json(
+                data, message
+            )
+        case "FailedDependencyException":
+            raise capo_workdocs.errors.failed_dependency_exception.FailedDependencyException.from_json(
+                data, message
+            )
+        case "InvalidArgumentException":
+            raise capo_workdocs.errors.invalid_argument_exception.InvalidArgumentException.from_json(
+                data, message
+            )
+        case "InvalidPasswordException":
+            raise capo_workdocs.errors.invalid_password_exception.InvalidPasswordException.from_json(
+                data, message
+            )
+        case "LimitExceededException":
+            raise capo_workdocs.errors.limit_exceeded_exception.LimitExceededException.from_json(
+                data, message
+            )
+        case "ProhibitedStateException":
+            raise capo_workdocs.errors.prohibited_state_exception.ProhibitedStateException.from_json(
+                data, message
+            )
+        case "ResourceAlreadyCheckedOutException":
+            raise capo_workdocs.errors.resource_already_checked_out_exception.ResourceAlreadyCheckedOutException.from_json(
+                data, message
+            )
+        case "ServiceUnavailableException":
+            raise capo_workdocs.errors.service_unavailable_exception.ServiceUnavailableException.from_json(
+                data, message
+            )
+        case "StorageLimitExceededException":
+            raise capo_workdocs.errors.storage_limit_exceeded_exception.StorageLimitExceededException.from_json(
+                data, message
+            )
+        case "StorageLimitWillExceedException":
+            raise capo_workdocs.errors.storage_limit_will_exceed_exception.StorageLimitWillExceedException.from_json(
+                data, message
+            )
+        case "UnauthorizedOperationException":
+            raise capo_workdocs.errors.unauthorized_operation_exception.UnauthorizedOperationException.from_json(
+                data, message
+            )
+        case "UnauthorizedResourceAccessException":
+            raise capo_workdocs.errors.unauthorized_resource_access_exception.UnauthorizedResourceAccessException.from_json(
+                data, message
+            )
+        case _:
+            raise UnknownServiceError(code=code, message=message, response=response)
+
+
+def handle_response(
+    response: zapros.Response,
+) -> capo_workdocs.types.initiate_document_version_upload_response.InitiateDocumentVersionUploadResponse:
+    out: capo_workdocs.types.initiate_document_version_upload_response.InitiateDocumentVersionUploadResponse = capo_workdocs.types.initiate_document_version_upload_response.deserialize_json(
+        json.loads(response.read())
+    )
+    return out
+
+
+async def async_handle_response(
+    response: zapros.Response,
+) -> capo_workdocs.types.initiate_document_version_upload_response.InitiateDocumentVersionUploadResponse:
+    out: capo_workdocs.types.initiate_document_version_upload_response.InitiateDocumentVersionUploadResponse = capo_workdocs.types.initiate_document_version_upload_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_workdocs._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_workdocs._auth._sigv4.build_sigv4_auth_scheme(
+                "workdocs", options.region, endpoint_scheme
+            )
+            if sigv4_config is not None:
+                return capo_workdocs._auth._signers.SigV4Signer(
+                    options.credentials_provider, auth_scheme=sigv4_config
+                )
+    raise RuntimeError("Auth was not resolved")
+
+
+def build_request(
+    options: OperationOptions | AsyncOperationOptions,
+    input_: capo_workdocs.types.initiate_document_version_upload_request.InitiateDocumentVersionUploadRequest,
+) -> 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("/") + "/api/v1/documents"
+    params: list[tuple[str, str]] = []
+    headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
+    if "authentication_token" in input_:
+        headers["Authentication"] = input_["authentication_token"]
+    body: bytes | None = json.dumps(
+        capo_workdocs.types.initiate_document_version_upload_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 initiate_document_version_upload(
+    options: OperationOptions,
+    input_: capo_workdocs.types.initiate_document_version_upload_request.InitiateDocumentVersionUploadRequest,
+) -> tuple[
+    capo_workdocs.types.initiate_document_version_upload_response.InitiateDocumentVersionUploadResponse,
+    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_initiate_document_version_upload(
+    options: AsyncOperationOptions,
+    input_: capo_workdocs.types.initiate_document_version_upload_request.InitiateDocumentVersionUploadRequest,
+) -> tuple[
+    capo_workdocs.types.initiate_document_version_upload_response.InitiateDocumentVersionUploadResponse,
+    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