capo-keyspaces 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (174) hide show
  1. capo_keyspaces/__init__.py +70 -0
  2. capo_keyspaces/_async.py +25 -0
  3. capo_keyspaces/_auth/_identity.py +16 -0
  4. capo_keyspaces/_auth/_providers.py +886 -0
  5. capo_keyspaces/_auth/_signers.py +88 -0
  6. capo_keyspaces/_auth/_sigv4.py +434 -0
  7. capo_keyspaces/_auth/_zapros_handler.py +80 -0
  8. capo_keyspaces/_iter.py +113 -0
  9. capo_keyspaces/_operations/keyspaces_service/create_keyspace.py +170 -0
  10. capo_keyspaces/_operations/keyspaces_service/create_table.py +183 -0
  11. capo_keyspaces/_operations/keyspaces_service/create_type.py +172 -0
  12. capo_keyspaces/_operations/keyspaces_service/delete_keyspace.py +165 -0
  13. capo_keyspaces/_operations/keyspaces_service/delete_table.py +163 -0
  14. capo_keyspaces/_operations/keyspaces_service/delete_type.py +171 -0
  15. capo_keyspaces/_operations/keyspaces_service/get_keyspace.py +168 -0
  16. capo_keyspaces/_operations/keyspaces_service/get_table.py +173 -0
  17. capo_keyspaces/_operations/keyspaces_service/get_table_auto_scaling_settings.py +168 -0
  18. capo_keyspaces/_operations/keyspaces_service/get_type.py +166 -0
  19. capo_keyspaces/_operations/keyspaces_service/list_keyspaces.py +167 -0
  20. capo_keyspaces/_operations/keyspaces_service/list_tables.py +167 -0
  21. capo_keyspaces/_operations/keyspaces_service/list_tags_for_resource.py +167 -0
  22. capo_keyspaces/_operations/keyspaces_service/list_types.py +163 -0
  23. capo_keyspaces/_operations/keyspaces_service/restore_table.py +178 -0
  24. capo_keyspaces/_operations/keyspaces_service/tag_resource.py +164 -0
  25. capo_keyspaces/_operations/keyspaces_service/untag_resource.py +164 -0
  26. capo_keyspaces/_operations/keyspaces_service/update_keyspace.py +175 -0
  27. capo_keyspaces/_operations/keyspaces_service/update_table.py +181 -0
  28. capo_keyspaces/_pagination.py +21 -0
  29. capo_keyspaces/_protocol/__init__.py +1 -0
  30. capo_keyspaces/_protocol/errors.py +93 -0
  31. capo_keyspaces/_protocol/eventstream.py +238 -0
  32. capo_keyspaces/_protocol/serialize.py +47 -0
  33. capo_keyspaces/_protocol/xml.py +33 -0
  34. capo_keyspaces/_rule_engine/__init__.py +0 -0
  35. capo_keyspaces/_rule_engine/_aws_partition.py +160 -0
  36. capo_keyspaces/_rule_engine/_endpoint_rule_set.py +150 -0
  37. capo_keyspaces/_rule_engine/_endpoint_runtime.py +389 -0
  38. capo_keyspaces/_services/_aws_config.py +160 -0
  39. capo_keyspaces/_services/_pipeline.py +198 -0
  40. capo_keyspaces/_services/async_keyspaces.py +1467 -0
  41. capo_keyspaces/_services/keyspaces.py +1448 -0
  42. capo_keyspaces/errors/__init__.py +35 -0
  43. capo_keyspaces/errors/_base.py +94 -0
  44. capo_keyspaces/errors/access_denied_exception.py +47 -0
  45. capo_keyspaces/errors/conflict_exception.py +47 -0
  46. capo_keyspaces/errors/internal_server_exception.py +47 -0
  47. capo_keyspaces/errors/resource_not_found_exception.py +58 -0
  48. capo_keyspaces/errors/service_quota_exceeded_exception.py +49 -0
  49. capo_keyspaces/errors/validation_exception.py +47 -0
  50. capo_keyspaces/py.typed +0 -0
  51. capo_keyspaces/types/_prelude/blob.py +12 -0
  52. capo_keyspaces/types/_prelude/timestamp.py +17 -0
  53. capo_keyspaces/types/arn.py +5 -0
  54. capo_keyspaces/types/auto_scaling_policy.py +42 -0
  55. capo_keyspaces/types/auto_scaling_settings.py +63 -0
  56. capo_keyspaces/types/auto_scaling_specification.py +62 -0
  57. capo_keyspaces/types/boolean_object.py +5 -0
  58. capo_keyspaces/types/capacity_specification.py +48 -0
  59. capo_keyspaces/types/capacity_specification_summary.py +71 -0
  60. capo_keyspaces/types/capacity_units.py +5 -0
  61. capo_keyspaces/types/cdc_propagate_tags.py +5 -0
  62. capo_keyspaces/types/cdc_specification.py +62 -0
  63. capo_keyspaces/types/cdc_specification_summary.py +38 -0
  64. capo_keyspaces/types/cdc_status.py +5 -0
  65. capo_keyspaces/types/client_side_timestamps.py +33 -0
  66. capo_keyspaces/types/client_side_timestamps_status.py +5 -0
  67. capo_keyspaces/types/clustering_key.py +39 -0
  68. capo_keyspaces/types/clustering_key_list.py +29 -0
  69. capo_keyspaces/types/column_definition.py +38 -0
  70. capo_keyspaces/types/column_definition_list.py +33 -0
  71. capo_keyspaces/types/comment.py +26 -0
  72. capo_keyspaces/types/create_keyspace_request.py +67 -0
  73. capo_keyspaces/types/create_keyspace_response.py +31 -0
  74. capo_keyspaces/types/create_table_request.py +281 -0
  75. capo_keyspaces/types/create_table_response.py +31 -0
  76. capo_keyspaces/types/create_type_request.py +57 -0
  77. capo_keyspaces/types/create_type_response.py +39 -0
  78. capo_keyspaces/types/default_time_to_live.py +5 -0
  79. capo_keyspaces/types/delete_keyspace_request.py +31 -0
  80. capo_keyspaces/types/delete_keyspace_response.py +18 -0
  81. capo_keyspaces/types/delete_table_request.py +39 -0
  82. capo_keyspaces/types/delete_table_response.py +18 -0
  83. capo_keyspaces/types/delete_type_request.py +39 -0
  84. capo_keyspaces/types/delete_type_response.py +39 -0
  85. capo_keyspaces/types/depth.py +5 -0
  86. capo_keyspaces/types/double_object.py +5 -0
  87. capo_keyspaces/types/encryption_specification.py +38 -0
  88. capo_keyspaces/types/encryption_type.py +5 -0
  89. capo_keyspaces/types/field_definition.py +38 -0
  90. capo_keyspaces/types/field_list.py +29 -0
  91. capo_keyspaces/types/generic_string.py +5 -0
  92. capo_keyspaces/types/get_keyspace_request.py +31 -0
  93. capo_keyspaces/types/get_keyspace_response.py +87 -0
  94. capo_keyspaces/types/get_table_auto_scaling_settings_request.py +43 -0
  95. capo_keyspaces/types/get_table_auto_scaling_settings_response.py +95 -0
  96. capo_keyspaces/types/get_table_request.py +39 -0
  97. capo_keyspaces/types/get_table_response.py +287 -0
  98. capo_keyspaces/types/get_type_request.py +39 -0
  99. capo_keyspaces/types/get_type_response.py +140 -0
  100. capo_keyspaces/types/integer_object.py +5 -0
  101. capo_keyspaces/types/keyspace_name.py +5 -0
  102. capo_keyspaces/types/keyspace_status.py +5 -0
  103. capo_keyspaces/types/keyspace_summary.py +66 -0
  104. capo_keyspaces/types/keyspace_summary_list.py +31 -0
  105. capo_keyspaces/types/kms_key_arn.py +5 -0
  106. capo_keyspaces/types/list_keyspaces_request.py +35 -0
  107. capo_keyspaces/types/list_keyspaces_response.py +50 -0
  108. capo_keyspaces/types/list_tables_request.py +45 -0
  109. capo_keyspaces/types/list_tables_response.py +45 -0
  110. capo_keyspaces/types/list_tags_for_resource_request.py +45 -0
  111. capo_keyspaces/types/list_tags_for_resource_response.py +43 -0
  112. capo_keyspaces/types/list_types_request.py +45 -0
  113. capo_keyspaces/types/list_types_response.py +46 -0
  114. capo_keyspaces/types/max_results.py +5 -0
  115. capo_keyspaces/types/next_token.py +5 -0
  116. capo_keyspaces/types/partition_key.py +31 -0
  117. capo_keyspaces/types/partition_key_list.py +29 -0
  118. capo_keyspaces/types/point_in_time_recovery.py +33 -0
  119. capo_keyspaces/types/point_in_time_recovery_status.py +5 -0
  120. capo_keyspaces/types/point_in_time_recovery_summary.py +54 -0
  121. capo_keyspaces/types/region.py +5 -0
  122. capo_keyspaces/types/region_list.py +17 -0
  123. capo_keyspaces/types/replica_auto_scaling_specification.py +49 -0
  124. capo_keyspaces/types/replica_auto_scaling_specification_list.py +39 -0
  125. capo_keyspaces/types/replica_specification.py +61 -0
  126. capo_keyspaces/types/replica_specification_list.py +35 -0
  127. capo_keyspaces/types/replica_specification_summary.py +76 -0
  128. capo_keyspaces/types/replica_specification_summary_list.py +39 -0
  129. capo_keyspaces/types/replication_group_status.py +48 -0
  130. capo_keyspaces/types/replication_group_status_list.py +35 -0
  131. capo_keyspaces/types/replication_specification.py +48 -0
  132. capo_keyspaces/types/restore_table_request.py +190 -0
  133. capo_keyspaces/types/restore_table_response.py +31 -0
  134. capo_keyspaces/types/rs.py +5 -0
  135. capo_keyspaces/types/schema_definition.py +105 -0
  136. capo_keyspaces/types/sort_order.py +5 -0
  137. capo_keyspaces/types/static_column.py +31 -0
  138. capo_keyspaces/types/static_column_list.py +29 -0
  139. capo_keyspaces/types/stream_arn.py +5 -0
  140. capo_keyspaces/types/table_name.py +5 -0
  141. capo_keyspaces/types/table_name_list.py +17 -0
  142. capo_keyspaces/types/table_status.py +5 -0
  143. capo_keyspaces/types/table_summary.py +47 -0
  144. capo_keyspaces/types/table_summary_list.py +29 -0
  145. capo_keyspaces/types/tables_replication_progress.py +5 -0
  146. capo_keyspaces/types/tag.py +39 -0
  147. capo_keyspaces/types/tag_key.py +5 -0
  148. capo_keyspaces/types/tag_list.py +29 -0
  149. capo_keyspaces/types/tag_resource_request.py +45 -0
  150. capo_keyspaces/types/tag_resource_response.py +18 -0
  151. capo_keyspaces/types/tag_value.py +5 -0
  152. capo_keyspaces/types/target_tracking_scaling_policy_configuration.py +60 -0
  153. capo_keyspaces/types/throughput_mode.py +5 -0
  154. capo_keyspaces/types/time_to_live.py +31 -0
  155. capo_keyspaces/types/time_to_live_status.py +5 -0
  156. capo_keyspaces/types/timestamp.py +20 -0
  157. capo_keyspaces/types/type_name.py +5 -0
  158. capo_keyspaces/types/type_name_list.py +17 -0
  159. capo_keyspaces/types/type_status.py +5 -0
  160. capo_keyspaces/types/untag_resource_request.py +45 -0
  161. capo_keyspaces/types/untag_resource_response.py +18 -0
  162. capo_keyspaces/types/update_keyspace_request.py +74 -0
  163. capo_keyspaces/types/update_keyspace_response.py +31 -0
  164. capo_keyspaces/types/update_table_request.py +252 -0
  165. capo_keyspaces/types/update_table_response.py +31 -0
  166. capo_keyspaces/types/view_type.py +5 -0
  167. capo_keyspaces/types/warm_throughput_specification.py +29 -0
  168. capo_keyspaces/types/warm_throughput_specification_summary.py +49 -0
  169. capo_keyspaces/types/warm_throughput_status.py +5 -0
  170. capo_keyspaces-0.1.0.dist-info/METADATA +101 -0
  171. capo_keyspaces-0.1.0.dist-info/RECORD +174 -0
  172. capo_keyspaces-0.1.0.dist-info/WHEEL +5 -0
  173. capo_keyspaces-0.1.0.dist-info/licenses/LICENSE +21 -0
  174. capo_keyspaces-0.1.0.dist-info/top_level.txt +1 -0
@@ -0,0 +1,164 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#TagResource``."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import json
6
+ from typing import Any
7
+
8
+ import zapros
9
+ from typing_extensions import Never
10
+
11
+ import capo_keyspaces._auth._signers
12
+ import capo_keyspaces._auth._sigv4
13
+ import capo_keyspaces._protocol.eventstream
14
+ import capo_keyspaces.errors.access_denied_exception
15
+ import capo_keyspaces.errors.conflict_exception
16
+ import capo_keyspaces.errors.internal_server_exception
17
+ import capo_keyspaces.errors.resource_not_found_exception
18
+ import capo_keyspaces.errors.service_quota_exceeded_exception
19
+ import capo_keyspaces.errors.validation_exception
20
+ import capo_keyspaces.types.tag_list
21
+ import capo_keyspaces.types.tag_resource_request
22
+ import capo_keyspaces.types.tag_resource_response
23
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
24
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
25
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
26
+ from capo_keyspaces.errors import UnknownServiceError
27
+
28
+
29
+ def handle_error(response: zapros.Response) -> Never:
30
+ data = json.loads(response.read())
31
+ code, message = parse_error_metadata_json(response, data)
32
+ match code:
33
+ case "AccessDeniedException":
34
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
35
+ data, message
36
+ )
37
+ case "ConflictException":
38
+ raise capo_keyspaces.errors.conflict_exception.ConflictException.from_aws_json_1_0(
39
+ data, message
40
+ )
41
+ case "InternalServerException":
42
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
43
+ data, message
44
+ )
45
+ case "ResourceNotFoundException":
46
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
47
+ data, message
48
+ )
49
+ case "ServiceQuotaExceededException":
50
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
51
+ data, message
52
+ )
53
+ case "ValidationException":
54
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
55
+ data, message
56
+ )
57
+ case _:
58
+ raise UnknownServiceError(code=code, message=message, response=response)
59
+
60
+
61
+ def handle_response(
62
+ response: zapros.Response,
63
+ ) -> capo_keyspaces.types.tag_resource_response.TagResourceResponse:
64
+ out: capo_keyspaces.types.tag_resource_response.TagResourceResponse = {} # type: ignore[typeddict-item]
65
+ return out
66
+
67
+
68
+ async def async_handle_response(
69
+ response: zapros.Response,
70
+ ) -> capo_keyspaces.types.tag_resource_response.TagResourceResponse:
71
+ out: capo_keyspaces.types.tag_resource_response.TagResourceResponse = {} # type: ignore[typeddict-item]
72
+ return out
73
+
74
+
75
+ def get_signer(
76
+ options: AsyncOperationOptions | OperationOptions,
77
+ auth_schemes: list[dict[str, Any]] | None = None,
78
+ ) -> capo_keyspaces._auth._signers.Signer | None:
79
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
80
+ if (
81
+ options.credentials_provider is not None
82
+ and name_to_schema
83
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
84
+ ):
85
+ raise RuntimeError(
86
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
87
+ )
88
+ if options.credentials_provider is not None:
89
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
90
+ "sigv4-s3express"
91
+ )
92
+ if endpoint_scheme is not None or not name_to_schema:
93
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
94
+ "cassandra", options.region, endpoint_scheme
95
+ )
96
+ if sigv4_config is not None:
97
+ return capo_keyspaces._auth._signers.SigV4Signer(
98
+ options.credentials_provider, auth_scheme=sigv4_config
99
+ )
100
+ raise RuntimeError("Auth was not resolved")
101
+
102
+
103
+ def build_request(
104
+ options: OperationOptions | AsyncOperationOptions,
105
+ input_: capo_keyspaces.types.tag_resource_request.TagResourceRequest,
106
+ ) -> zapros.Request:
107
+ endpoint = resolve(
108
+ EndpointParams(
109
+ Region=options.region,
110
+ UseDualStack=options.use_dual_stack,
111
+ UseFIPS=options.use_fips,
112
+ Endpoint=options.endpoint,
113
+ )
114
+ ) # noqa: F841
115
+ url = endpoint.url.rstrip("/") + ""
116
+ params: list[tuple[str, str]] = []
117
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
118
+ headers["X-Amz-Target"] = "KeyspacesService.TagResource"
119
+ body: bytes | None = json.dumps(
120
+ capo_keyspaces.types.tag_resource_request.serialize_aws_json_1_0(input_),
121
+ allow_nan=False,
122
+ ).encode()
123
+ headers["content-type"] = "application/x-amz-json-1.0"
124
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
125
+ normalized_url = zapros.URL(url)
126
+ for k, v in params:
127
+ normalized_url.search_params.append(k, v)
128
+ return zapros.Request(
129
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
130
+ )
131
+
132
+
133
+ def tag_resource(
134
+ options: OperationOptions,
135
+ input_: capo_keyspaces.types.tag_resource_request.TagResourceRequest,
136
+ ) -> tuple[
137
+ capo_keyspaces.types.tag_resource_response.TagResourceResponse, zapros.Response
138
+ ]:
139
+ response = options.client.handler.handle(build_request(options, input_))
140
+ try:
141
+ if response.status >= 300:
142
+ response.read()
143
+ handle_error(response)
144
+ return handle_response(response), response
145
+ except BaseException:
146
+ response.close()
147
+ raise
148
+
149
+
150
+ async def async_tag_resource(
151
+ options: AsyncOperationOptions,
152
+ input_: capo_keyspaces.types.tag_resource_request.TagResourceRequest,
153
+ ) -> tuple[
154
+ capo_keyspaces.types.tag_resource_response.TagResourceResponse, zapros.Response
155
+ ]:
156
+ response = await options.client.handler.ahandle(build_request(options, input_))
157
+ try:
158
+ if response.status >= 300:
159
+ await response.aread()
160
+ handle_error(response)
161
+ return await async_handle_response(response), response
162
+ except BaseException:
163
+ await response.aclose()
164
+ raise
@@ -0,0 +1,164 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#UntagResource``."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import json
6
+ from typing import Any
7
+
8
+ import zapros
9
+ from typing_extensions import Never
10
+
11
+ import capo_keyspaces._auth._signers
12
+ import capo_keyspaces._auth._sigv4
13
+ import capo_keyspaces._protocol.eventstream
14
+ import capo_keyspaces.errors.access_denied_exception
15
+ import capo_keyspaces.errors.conflict_exception
16
+ import capo_keyspaces.errors.internal_server_exception
17
+ import capo_keyspaces.errors.resource_not_found_exception
18
+ import capo_keyspaces.errors.service_quota_exceeded_exception
19
+ import capo_keyspaces.errors.validation_exception
20
+ import capo_keyspaces.types.tag_list
21
+ import capo_keyspaces.types.untag_resource_request
22
+ import capo_keyspaces.types.untag_resource_response
23
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
24
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
25
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
26
+ from capo_keyspaces.errors import UnknownServiceError
27
+
28
+
29
+ def handle_error(response: zapros.Response) -> Never:
30
+ data = json.loads(response.read())
31
+ code, message = parse_error_metadata_json(response, data)
32
+ match code:
33
+ case "AccessDeniedException":
34
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
35
+ data, message
36
+ )
37
+ case "ConflictException":
38
+ raise capo_keyspaces.errors.conflict_exception.ConflictException.from_aws_json_1_0(
39
+ data, message
40
+ )
41
+ case "InternalServerException":
42
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
43
+ data, message
44
+ )
45
+ case "ResourceNotFoundException":
46
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
47
+ data, message
48
+ )
49
+ case "ServiceQuotaExceededException":
50
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
51
+ data, message
52
+ )
53
+ case "ValidationException":
54
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
55
+ data, message
56
+ )
57
+ case _:
58
+ raise UnknownServiceError(code=code, message=message, response=response)
59
+
60
+
61
+ def handle_response(
62
+ response: zapros.Response,
63
+ ) -> capo_keyspaces.types.untag_resource_response.UntagResourceResponse:
64
+ out: capo_keyspaces.types.untag_resource_response.UntagResourceResponse = {} # type: ignore[typeddict-item]
65
+ return out
66
+
67
+
68
+ async def async_handle_response(
69
+ response: zapros.Response,
70
+ ) -> capo_keyspaces.types.untag_resource_response.UntagResourceResponse:
71
+ out: capo_keyspaces.types.untag_resource_response.UntagResourceResponse = {} # type: ignore[typeddict-item]
72
+ return out
73
+
74
+
75
+ def get_signer(
76
+ options: AsyncOperationOptions | OperationOptions,
77
+ auth_schemes: list[dict[str, Any]] | None = None,
78
+ ) -> capo_keyspaces._auth._signers.Signer | None:
79
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
80
+ if (
81
+ options.credentials_provider is not None
82
+ and name_to_schema
83
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
84
+ ):
85
+ raise RuntimeError(
86
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
87
+ )
88
+ if options.credentials_provider is not None:
89
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
90
+ "sigv4-s3express"
91
+ )
92
+ if endpoint_scheme is not None or not name_to_schema:
93
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
94
+ "cassandra", options.region, endpoint_scheme
95
+ )
96
+ if sigv4_config is not None:
97
+ return capo_keyspaces._auth._signers.SigV4Signer(
98
+ options.credentials_provider, auth_scheme=sigv4_config
99
+ )
100
+ raise RuntimeError("Auth was not resolved")
101
+
102
+
103
+ def build_request(
104
+ options: OperationOptions | AsyncOperationOptions,
105
+ input_: capo_keyspaces.types.untag_resource_request.UntagResourceRequest,
106
+ ) -> zapros.Request:
107
+ endpoint = resolve(
108
+ EndpointParams(
109
+ Region=options.region,
110
+ UseDualStack=options.use_dual_stack,
111
+ UseFIPS=options.use_fips,
112
+ Endpoint=options.endpoint,
113
+ )
114
+ ) # noqa: F841
115
+ url = endpoint.url.rstrip("/") + ""
116
+ params: list[tuple[str, str]] = []
117
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
118
+ headers["X-Amz-Target"] = "KeyspacesService.UntagResource"
119
+ body: bytes | None = json.dumps(
120
+ capo_keyspaces.types.untag_resource_request.serialize_aws_json_1_0(input_),
121
+ allow_nan=False,
122
+ ).encode()
123
+ headers["content-type"] = "application/x-amz-json-1.0"
124
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
125
+ normalized_url = zapros.URL(url)
126
+ for k, v in params:
127
+ normalized_url.search_params.append(k, v)
128
+ return zapros.Request(
129
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
130
+ )
131
+
132
+
133
+ def untag_resource(
134
+ options: OperationOptions,
135
+ input_: capo_keyspaces.types.untag_resource_request.UntagResourceRequest,
136
+ ) -> tuple[
137
+ capo_keyspaces.types.untag_resource_response.UntagResourceResponse, zapros.Response
138
+ ]:
139
+ response = options.client.handler.handle(build_request(options, input_))
140
+ try:
141
+ if response.status >= 300:
142
+ response.read()
143
+ handle_error(response)
144
+ return handle_response(response), response
145
+ except BaseException:
146
+ response.close()
147
+ raise
148
+
149
+
150
+ async def async_untag_resource(
151
+ options: AsyncOperationOptions,
152
+ input_: capo_keyspaces.types.untag_resource_request.UntagResourceRequest,
153
+ ) -> tuple[
154
+ capo_keyspaces.types.untag_resource_response.UntagResourceResponse, zapros.Response
155
+ ]:
156
+ response = await options.client.handler.ahandle(build_request(options, input_))
157
+ try:
158
+ if response.status >= 300:
159
+ await response.aread()
160
+ handle_error(response)
161
+ return await async_handle_response(response), response
162
+ except BaseException:
163
+ await response.aclose()
164
+ raise
@@ -0,0 +1,175 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#UpdateKeyspace``."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import json
6
+ from typing import Any
7
+
8
+ import zapros
9
+ from typing_extensions import Never
10
+
11
+ import capo_keyspaces._auth._signers
12
+ import capo_keyspaces._auth._sigv4
13
+ import capo_keyspaces._protocol.eventstream
14
+ import capo_keyspaces.errors.access_denied_exception
15
+ import capo_keyspaces.errors.conflict_exception
16
+ import capo_keyspaces.errors.internal_server_exception
17
+ import capo_keyspaces.errors.resource_not_found_exception
18
+ import capo_keyspaces.errors.service_quota_exceeded_exception
19
+ import capo_keyspaces.errors.validation_exception
20
+ import capo_keyspaces.types.client_side_timestamps
21
+ import capo_keyspaces.types.replication_specification
22
+ import capo_keyspaces.types.update_keyspace_request
23
+ import capo_keyspaces.types.update_keyspace_response
24
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
25
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
26
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
27
+ from capo_keyspaces.errors import UnknownServiceError
28
+
29
+
30
+ def handle_error(response: zapros.Response) -> Never:
31
+ data = json.loads(response.read())
32
+ code, message = parse_error_metadata_json(response, data)
33
+ match code:
34
+ case "AccessDeniedException":
35
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
36
+ data, message
37
+ )
38
+ case "ConflictException":
39
+ raise capo_keyspaces.errors.conflict_exception.ConflictException.from_aws_json_1_0(
40
+ data, message
41
+ )
42
+ case "InternalServerException":
43
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
44
+ data, message
45
+ )
46
+ case "ResourceNotFoundException":
47
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
48
+ data, message
49
+ )
50
+ case "ServiceQuotaExceededException":
51
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
52
+ data, message
53
+ )
54
+ case "ValidationException":
55
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
56
+ data, message
57
+ )
58
+ case _:
59
+ raise UnknownServiceError(code=code, message=message, response=response)
60
+
61
+
62
+ def handle_response(
63
+ response: zapros.Response,
64
+ ) -> capo_keyspaces.types.update_keyspace_response.UpdateKeyspaceResponse:
65
+ out: capo_keyspaces.types.update_keyspace_response.UpdateKeyspaceResponse = (
66
+ capo_keyspaces.types.update_keyspace_response.deserialize_aws_json_1_0(
67
+ json.loads(response.read())
68
+ )
69
+ )
70
+ return out
71
+
72
+
73
+ async def async_handle_response(
74
+ response: zapros.Response,
75
+ ) -> capo_keyspaces.types.update_keyspace_response.UpdateKeyspaceResponse:
76
+ out: capo_keyspaces.types.update_keyspace_response.UpdateKeyspaceResponse = (
77
+ capo_keyspaces.types.update_keyspace_response.deserialize_aws_json_1_0(
78
+ json.loads(await response.aread())
79
+ )
80
+ )
81
+ return out
82
+
83
+
84
+ def get_signer(
85
+ options: AsyncOperationOptions | OperationOptions,
86
+ auth_schemes: list[dict[str, Any]] | None = None,
87
+ ) -> capo_keyspaces._auth._signers.Signer | None:
88
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
89
+ if (
90
+ options.credentials_provider is not None
91
+ and name_to_schema
92
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
93
+ ):
94
+ raise RuntimeError(
95
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
96
+ )
97
+ if options.credentials_provider is not None:
98
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
99
+ "sigv4-s3express"
100
+ )
101
+ if endpoint_scheme is not None or not name_to_schema:
102
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
103
+ "cassandra", options.region, endpoint_scheme
104
+ )
105
+ if sigv4_config is not None:
106
+ return capo_keyspaces._auth._signers.SigV4Signer(
107
+ options.credentials_provider, auth_scheme=sigv4_config
108
+ )
109
+ raise RuntimeError("Auth was not resolved")
110
+
111
+
112
+ def build_request(
113
+ options: OperationOptions | AsyncOperationOptions,
114
+ input_: capo_keyspaces.types.update_keyspace_request.UpdateKeyspaceRequest,
115
+ ) -> zapros.Request:
116
+ endpoint = resolve(
117
+ EndpointParams(
118
+ Region=options.region,
119
+ UseDualStack=options.use_dual_stack,
120
+ UseFIPS=options.use_fips,
121
+ Endpoint=options.endpoint,
122
+ )
123
+ ) # noqa: F841
124
+ url = endpoint.url.rstrip("/") + ""
125
+ params: list[tuple[str, str]] = []
126
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
127
+ headers["X-Amz-Target"] = "KeyspacesService.UpdateKeyspace"
128
+ body: bytes | None = json.dumps(
129
+ capo_keyspaces.types.update_keyspace_request.serialize_aws_json_1_0(input_),
130
+ allow_nan=False,
131
+ ).encode()
132
+ headers["content-type"] = "application/x-amz-json-1.0"
133
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
134
+ normalized_url = zapros.URL(url)
135
+ for k, v in params:
136
+ normalized_url.search_params.append(k, v)
137
+ return zapros.Request(
138
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
139
+ )
140
+
141
+
142
+ def update_keyspace(
143
+ options: OperationOptions,
144
+ input_: capo_keyspaces.types.update_keyspace_request.UpdateKeyspaceRequest,
145
+ ) -> tuple[
146
+ capo_keyspaces.types.update_keyspace_response.UpdateKeyspaceResponse,
147
+ zapros.Response,
148
+ ]:
149
+ response = options.client.handler.handle(build_request(options, input_))
150
+ try:
151
+ if response.status >= 300:
152
+ response.read()
153
+ handle_error(response)
154
+ return handle_response(response), response
155
+ except BaseException:
156
+ response.close()
157
+ raise
158
+
159
+
160
+ async def async_update_keyspace(
161
+ options: AsyncOperationOptions,
162
+ input_: capo_keyspaces.types.update_keyspace_request.UpdateKeyspaceRequest,
163
+ ) -> tuple[
164
+ capo_keyspaces.types.update_keyspace_response.UpdateKeyspaceResponse,
165
+ zapros.Response,
166
+ ]:
167
+ response = await options.client.handler.ahandle(build_request(options, input_))
168
+ try:
169
+ if response.status >= 300:
170
+ await response.aread()
171
+ handle_error(response)
172
+ return await async_handle_response(response), response
173
+ except BaseException:
174
+ await response.aclose()
175
+ raise
@@ -0,0 +1,181 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#UpdateTable``."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import json
6
+ from typing import Any
7
+
8
+ import zapros
9
+ from typing_extensions import Never
10
+
11
+ import capo_keyspaces._auth._signers
12
+ import capo_keyspaces._auth._sigv4
13
+ import capo_keyspaces._protocol.eventstream
14
+ import capo_keyspaces.errors.access_denied_exception
15
+ import capo_keyspaces.errors.conflict_exception
16
+ import capo_keyspaces.errors.internal_server_exception
17
+ import capo_keyspaces.errors.resource_not_found_exception
18
+ import capo_keyspaces.errors.service_quota_exceeded_exception
19
+ import capo_keyspaces.errors.validation_exception
20
+ import capo_keyspaces.types.auto_scaling_specification
21
+ import capo_keyspaces.types.capacity_specification
22
+ import capo_keyspaces.types.cdc_specification
23
+ import capo_keyspaces.types.client_side_timestamps
24
+ import capo_keyspaces.types.column_definition_list
25
+ import capo_keyspaces.types.encryption_specification
26
+ import capo_keyspaces.types.point_in_time_recovery
27
+ import capo_keyspaces.types.replica_specification_list
28
+ import capo_keyspaces.types.time_to_live
29
+ import capo_keyspaces.types.update_table_request
30
+ import capo_keyspaces.types.update_table_response
31
+ import capo_keyspaces.types.warm_throughput_specification
32
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
33
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
34
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
35
+ from capo_keyspaces.errors import UnknownServiceError
36
+
37
+
38
+ def handle_error(response: zapros.Response) -> Never:
39
+ data = json.loads(response.read())
40
+ code, message = parse_error_metadata_json(response, data)
41
+ match code:
42
+ case "AccessDeniedException":
43
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
44
+ data, message
45
+ )
46
+ case "ConflictException":
47
+ raise capo_keyspaces.errors.conflict_exception.ConflictException.from_aws_json_1_0(
48
+ data, message
49
+ )
50
+ case "InternalServerException":
51
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
52
+ data, message
53
+ )
54
+ case "ResourceNotFoundException":
55
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
56
+ data, message
57
+ )
58
+ case "ServiceQuotaExceededException":
59
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
60
+ data, message
61
+ )
62
+ case "ValidationException":
63
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
64
+ data, message
65
+ )
66
+ case _:
67
+ raise UnknownServiceError(code=code, message=message, response=response)
68
+
69
+
70
+ def handle_response(
71
+ response: zapros.Response,
72
+ ) -> capo_keyspaces.types.update_table_response.UpdateTableResponse:
73
+ out: capo_keyspaces.types.update_table_response.UpdateTableResponse = (
74
+ capo_keyspaces.types.update_table_response.deserialize_aws_json_1_0(
75
+ json.loads(response.read())
76
+ )
77
+ )
78
+ return out
79
+
80
+
81
+ async def async_handle_response(
82
+ response: zapros.Response,
83
+ ) -> capo_keyspaces.types.update_table_response.UpdateTableResponse:
84
+ out: capo_keyspaces.types.update_table_response.UpdateTableResponse = (
85
+ capo_keyspaces.types.update_table_response.deserialize_aws_json_1_0(
86
+ json.loads(await response.aread())
87
+ )
88
+ )
89
+ return out
90
+
91
+
92
+ def get_signer(
93
+ options: AsyncOperationOptions | OperationOptions,
94
+ auth_schemes: list[dict[str, Any]] | None = None,
95
+ ) -> capo_keyspaces._auth._signers.Signer | None:
96
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
97
+ if (
98
+ options.credentials_provider is not None
99
+ and name_to_schema
100
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
101
+ ):
102
+ raise RuntimeError(
103
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
104
+ )
105
+ if options.credentials_provider is not None:
106
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
107
+ "sigv4-s3express"
108
+ )
109
+ if endpoint_scheme is not None or not name_to_schema:
110
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
111
+ "cassandra", options.region, endpoint_scheme
112
+ )
113
+ if sigv4_config is not None:
114
+ return capo_keyspaces._auth._signers.SigV4Signer(
115
+ options.credentials_provider, auth_scheme=sigv4_config
116
+ )
117
+ raise RuntimeError("Auth was not resolved")
118
+
119
+
120
+ def build_request(
121
+ options: OperationOptions | AsyncOperationOptions,
122
+ input_: capo_keyspaces.types.update_table_request.UpdateTableRequest,
123
+ ) -> zapros.Request:
124
+ endpoint = resolve(
125
+ EndpointParams(
126
+ Region=options.region,
127
+ UseDualStack=options.use_dual_stack,
128
+ UseFIPS=options.use_fips,
129
+ Endpoint=options.endpoint,
130
+ )
131
+ ) # noqa: F841
132
+ url = endpoint.url.rstrip("/") + ""
133
+ params: list[tuple[str, str]] = []
134
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
135
+ headers["X-Amz-Target"] = "KeyspacesService.UpdateTable"
136
+ body: bytes | None = json.dumps(
137
+ capo_keyspaces.types.update_table_request.serialize_aws_json_1_0(input_),
138
+ allow_nan=False,
139
+ ).encode()
140
+ headers["content-type"] = "application/x-amz-json-1.0"
141
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
142
+ normalized_url = zapros.URL(url)
143
+ for k, v in params:
144
+ normalized_url.search_params.append(k, v)
145
+ return zapros.Request(
146
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
147
+ )
148
+
149
+
150
+ def update_table(
151
+ options: OperationOptions,
152
+ input_: capo_keyspaces.types.update_table_request.UpdateTableRequest,
153
+ ) -> tuple[
154
+ capo_keyspaces.types.update_table_response.UpdateTableResponse, zapros.Response
155
+ ]:
156
+ response = options.client.handler.handle(build_request(options, input_))
157
+ try:
158
+ if response.status >= 300:
159
+ response.read()
160
+ handle_error(response)
161
+ return handle_response(response), response
162
+ except BaseException:
163
+ response.close()
164
+ raise
165
+
166
+
167
+ async def async_update_table(
168
+ options: AsyncOperationOptions,
169
+ input_: capo_keyspaces.types.update_table_request.UpdateTableRequest,
170
+ ) -> tuple[
171
+ capo_keyspaces.types.update_table_response.UpdateTableResponse, zapros.Response
172
+ ]:
173
+ response = await options.client.handler.ahandle(build_request(options, input_))
174
+ try:
175
+ if response.status >= 300:
176
+ await response.aread()
177
+ handle_error(response)
178
+ return await async_handle_response(response), response
179
+ except BaseException:
180
+ await response.aclose()
181
+ raise