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,165 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#DeleteKeyspace``."""
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.delete_keyspace_request
21
+ import capo_keyspaces.types.delete_keyspace_response
22
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
23
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
24
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
25
+ from capo_keyspaces.errors import UnknownServiceError
26
+
27
+
28
+ def handle_error(response: zapros.Response) -> Never:
29
+ data = json.loads(response.read())
30
+ code, message = parse_error_metadata_json(response, data)
31
+ match code:
32
+ case "AccessDeniedException":
33
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
34
+ data, message
35
+ )
36
+ case "ConflictException":
37
+ raise capo_keyspaces.errors.conflict_exception.ConflictException.from_aws_json_1_0(
38
+ data, message
39
+ )
40
+ case "InternalServerException":
41
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
42
+ data, message
43
+ )
44
+ case "ResourceNotFoundException":
45
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
46
+ data, message
47
+ )
48
+ case "ServiceQuotaExceededException":
49
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
50
+ data, message
51
+ )
52
+ case "ValidationException":
53
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
54
+ data, message
55
+ )
56
+ case _:
57
+ raise UnknownServiceError(code=code, message=message, response=response)
58
+
59
+
60
+ def handle_response(
61
+ response: zapros.Response,
62
+ ) -> capo_keyspaces.types.delete_keyspace_response.DeleteKeyspaceResponse:
63
+ out: capo_keyspaces.types.delete_keyspace_response.DeleteKeyspaceResponse = {} # type: ignore[typeddict-item]
64
+ return out
65
+
66
+
67
+ async def async_handle_response(
68
+ response: zapros.Response,
69
+ ) -> capo_keyspaces.types.delete_keyspace_response.DeleteKeyspaceResponse:
70
+ out: capo_keyspaces.types.delete_keyspace_response.DeleteKeyspaceResponse = {} # type: ignore[typeddict-item]
71
+ return out
72
+
73
+
74
+ def get_signer(
75
+ options: AsyncOperationOptions | OperationOptions,
76
+ auth_schemes: list[dict[str, Any]] | None = None,
77
+ ) -> capo_keyspaces._auth._signers.Signer | None:
78
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
79
+ if (
80
+ options.credentials_provider is not None
81
+ and name_to_schema
82
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
83
+ ):
84
+ raise RuntimeError(
85
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
86
+ )
87
+ if options.credentials_provider is not None:
88
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
89
+ "sigv4-s3express"
90
+ )
91
+ if endpoint_scheme is not None or not name_to_schema:
92
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
93
+ "cassandra", options.region, endpoint_scheme
94
+ )
95
+ if sigv4_config is not None:
96
+ return capo_keyspaces._auth._signers.SigV4Signer(
97
+ options.credentials_provider, auth_scheme=sigv4_config
98
+ )
99
+ raise RuntimeError("Auth was not resolved")
100
+
101
+
102
+ def build_request(
103
+ options: OperationOptions | AsyncOperationOptions,
104
+ input_: capo_keyspaces.types.delete_keyspace_request.DeleteKeyspaceRequest,
105
+ ) -> zapros.Request:
106
+ endpoint = resolve(
107
+ EndpointParams(
108
+ Region=options.region,
109
+ UseDualStack=options.use_dual_stack,
110
+ UseFIPS=options.use_fips,
111
+ Endpoint=options.endpoint,
112
+ )
113
+ ) # noqa: F841
114
+ url = endpoint.url.rstrip("/") + ""
115
+ params: list[tuple[str, str]] = []
116
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
117
+ headers["X-Amz-Target"] = "KeyspacesService.DeleteKeyspace"
118
+ body: bytes | None = json.dumps(
119
+ capo_keyspaces.types.delete_keyspace_request.serialize_aws_json_1_0(input_),
120
+ allow_nan=False,
121
+ ).encode()
122
+ headers["content-type"] = "application/x-amz-json-1.0"
123
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
124
+ normalized_url = zapros.URL(url)
125
+ for k, v in params:
126
+ normalized_url.search_params.append(k, v)
127
+ return zapros.Request(
128
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
129
+ )
130
+
131
+
132
+ def delete_keyspace(
133
+ options: OperationOptions,
134
+ input_: capo_keyspaces.types.delete_keyspace_request.DeleteKeyspaceRequest,
135
+ ) -> tuple[
136
+ capo_keyspaces.types.delete_keyspace_response.DeleteKeyspaceResponse,
137
+ 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_delete_keyspace(
151
+ options: AsyncOperationOptions,
152
+ input_: capo_keyspaces.types.delete_keyspace_request.DeleteKeyspaceRequest,
153
+ ) -> tuple[
154
+ capo_keyspaces.types.delete_keyspace_response.DeleteKeyspaceResponse,
155
+ zapros.Response,
156
+ ]:
157
+ response = await options.client.handler.ahandle(build_request(options, input_))
158
+ try:
159
+ if response.status >= 300:
160
+ await response.aread()
161
+ handle_error(response)
162
+ return await async_handle_response(response), response
163
+ except BaseException:
164
+ await response.aclose()
165
+ raise
@@ -0,0 +1,163 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#DeleteTable``."""
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.delete_table_request
21
+ import capo_keyspaces.types.delete_table_response
22
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
23
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
24
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
25
+ from capo_keyspaces.errors import UnknownServiceError
26
+
27
+
28
+ def handle_error(response: zapros.Response) -> Never:
29
+ data = json.loads(response.read())
30
+ code, message = parse_error_metadata_json(response, data)
31
+ match code:
32
+ case "AccessDeniedException":
33
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
34
+ data, message
35
+ )
36
+ case "ConflictException":
37
+ raise capo_keyspaces.errors.conflict_exception.ConflictException.from_aws_json_1_0(
38
+ data, message
39
+ )
40
+ case "InternalServerException":
41
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
42
+ data, message
43
+ )
44
+ case "ResourceNotFoundException":
45
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
46
+ data, message
47
+ )
48
+ case "ServiceQuotaExceededException":
49
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
50
+ data, message
51
+ )
52
+ case "ValidationException":
53
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
54
+ data, message
55
+ )
56
+ case _:
57
+ raise UnknownServiceError(code=code, message=message, response=response)
58
+
59
+
60
+ def handle_response(
61
+ response: zapros.Response,
62
+ ) -> capo_keyspaces.types.delete_table_response.DeleteTableResponse:
63
+ out: capo_keyspaces.types.delete_table_response.DeleteTableResponse = {} # type: ignore[typeddict-item]
64
+ return out
65
+
66
+
67
+ async def async_handle_response(
68
+ response: zapros.Response,
69
+ ) -> capo_keyspaces.types.delete_table_response.DeleteTableResponse:
70
+ out: capo_keyspaces.types.delete_table_response.DeleteTableResponse = {} # type: ignore[typeddict-item]
71
+ return out
72
+
73
+
74
+ def get_signer(
75
+ options: AsyncOperationOptions | OperationOptions,
76
+ auth_schemes: list[dict[str, Any]] | None = None,
77
+ ) -> capo_keyspaces._auth._signers.Signer | None:
78
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
79
+ if (
80
+ options.credentials_provider is not None
81
+ and name_to_schema
82
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
83
+ ):
84
+ raise RuntimeError(
85
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
86
+ )
87
+ if options.credentials_provider is not None:
88
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
89
+ "sigv4-s3express"
90
+ )
91
+ if endpoint_scheme is not None or not name_to_schema:
92
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
93
+ "cassandra", options.region, endpoint_scheme
94
+ )
95
+ if sigv4_config is not None:
96
+ return capo_keyspaces._auth._signers.SigV4Signer(
97
+ options.credentials_provider, auth_scheme=sigv4_config
98
+ )
99
+ raise RuntimeError("Auth was not resolved")
100
+
101
+
102
+ def build_request(
103
+ options: OperationOptions | AsyncOperationOptions,
104
+ input_: capo_keyspaces.types.delete_table_request.DeleteTableRequest,
105
+ ) -> zapros.Request:
106
+ endpoint = resolve(
107
+ EndpointParams(
108
+ Region=options.region,
109
+ UseDualStack=options.use_dual_stack,
110
+ UseFIPS=options.use_fips,
111
+ Endpoint=options.endpoint,
112
+ )
113
+ ) # noqa: F841
114
+ url = endpoint.url.rstrip("/") + ""
115
+ params: list[tuple[str, str]] = []
116
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
117
+ headers["X-Amz-Target"] = "KeyspacesService.DeleteTable"
118
+ body: bytes | None = json.dumps(
119
+ capo_keyspaces.types.delete_table_request.serialize_aws_json_1_0(input_),
120
+ allow_nan=False,
121
+ ).encode()
122
+ headers["content-type"] = "application/x-amz-json-1.0"
123
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
124
+ normalized_url = zapros.URL(url)
125
+ for k, v in params:
126
+ normalized_url.search_params.append(k, v)
127
+ return zapros.Request(
128
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
129
+ )
130
+
131
+
132
+ def delete_table(
133
+ options: OperationOptions,
134
+ input_: capo_keyspaces.types.delete_table_request.DeleteTableRequest,
135
+ ) -> tuple[
136
+ capo_keyspaces.types.delete_table_response.DeleteTableResponse, zapros.Response
137
+ ]:
138
+ response = options.client.handler.handle(build_request(options, input_))
139
+ try:
140
+ if response.status >= 300:
141
+ response.read()
142
+ handle_error(response)
143
+ return handle_response(response), response
144
+ except BaseException:
145
+ response.close()
146
+ raise
147
+
148
+
149
+ async def async_delete_table(
150
+ options: AsyncOperationOptions,
151
+ input_: capo_keyspaces.types.delete_table_request.DeleteTableRequest,
152
+ ) -> tuple[
153
+ capo_keyspaces.types.delete_table_response.DeleteTableResponse, zapros.Response
154
+ ]:
155
+ response = await options.client.handler.ahandle(build_request(options, input_))
156
+ try:
157
+ if response.status >= 300:
158
+ await response.aread()
159
+ handle_error(response)
160
+ return await async_handle_response(response), response
161
+ except BaseException:
162
+ await response.aclose()
163
+ raise
@@ -0,0 +1,171 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#DeleteType``."""
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.delete_type_request
21
+ import capo_keyspaces.types.delete_type_response
22
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
23
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
24
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
25
+ from capo_keyspaces.errors import UnknownServiceError
26
+
27
+
28
+ def handle_error(response: zapros.Response) -> Never:
29
+ data = json.loads(response.read())
30
+ code, message = parse_error_metadata_json(response, data)
31
+ match code:
32
+ case "AccessDeniedException":
33
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
34
+ data, message
35
+ )
36
+ case "ConflictException":
37
+ raise capo_keyspaces.errors.conflict_exception.ConflictException.from_aws_json_1_0(
38
+ data, message
39
+ )
40
+ case "InternalServerException":
41
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
42
+ data, message
43
+ )
44
+ case "ResourceNotFoundException":
45
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
46
+ data, message
47
+ )
48
+ case "ServiceQuotaExceededException":
49
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
50
+ data, message
51
+ )
52
+ case "ValidationException":
53
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
54
+ data, message
55
+ )
56
+ case _:
57
+ raise UnknownServiceError(code=code, message=message, response=response)
58
+
59
+
60
+ def handle_response(
61
+ response: zapros.Response,
62
+ ) -> capo_keyspaces.types.delete_type_response.DeleteTypeResponse:
63
+ out: capo_keyspaces.types.delete_type_response.DeleteTypeResponse = (
64
+ capo_keyspaces.types.delete_type_response.deserialize_aws_json_1_0(
65
+ json.loads(response.read())
66
+ )
67
+ )
68
+ return out
69
+
70
+
71
+ async def async_handle_response(
72
+ response: zapros.Response,
73
+ ) -> capo_keyspaces.types.delete_type_response.DeleteTypeResponse:
74
+ out: capo_keyspaces.types.delete_type_response.DeleteTypeResponse = (
75
+ capo_keyspaces.types.delete_type_response.deserialize_aws_json_1_0(
76
+ json.loads(await response.aread())
77
+ )
78
+ )
79
+ return out
80
+
81
+
82
+ def get_signer(
83
+ options: AsyncOperationOptions | OperationOptions,
84
+ auth_schemes: list[dict[str, Any]] | None = None,
85
+ ) -> capo_keyspaces._auth._signers.Signer | None:
86
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
87
+ if (
88
+ options.credentials_provider is not None
89
+ and name_to_schema
90
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
91
+ ):
92
+ raise RuntimeError(
93
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
94
+ )
95
+ if options.credentials_provider is not None:
96
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
97
+ "sigv4-s3express"
98
+ )
99
+ if endpoint_scheme is not None or not name_to_schema:
100
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
101
+ "cassandra", options.region, endpoint_scheme
102
+ )
103
+ if sigv4_config is not None:
104
+ return capo_keyspaces._auth._signers.SigV4Signer(
105
+ options.credentials_provider, auth_scheme=sigv4_config
106
+ )
107
+ raise RuntimeError("Auth was not resolved")
108
+
109
+
110
+ def build_request(
111
+ options: OperationOptions | AsyncOperationOptions,
112
+ input_: capo_keyspaces.types.delete_type_request.DeleteTypeRequest,
113
+ ) -> zapros.Request:
114
+ endpoint = resolve(
115
+ EndpointParams(
116
+ Region=options.region,
117
+ UseDualStack=options.use_dual_stack,
118
+ UseFIPS=options.use_fips,
119
+ Endpoint=options.endpoint,
120
+ )
121
+ ) # noqa: F841
122
+ url = endpoint.url.rstrip("/") + ""
123
+ params: list[tuple[str, str]] = []
124
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
125
+ headers["X-Amz-Target"] = "KeyspacesService.DeleteType"
126
+ body: bytes | None = json.dumps(
127
+ capo_keyspaces.types.delete_type_request.serialize_aws_json_1_0(input_),
128
+ allow_nan=False,
129
+ ).encode()
130
+ headers["content-type"] = "application/x-amz-json-1.0"
131
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
132
+ normalized_url = zapros.URL(url)
133
+ for k, v in params:
134
+ normalized_url.search_params.append(k, v)
135
+ return zapros.Request(
136
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
137
+ )
138
+
139
+
140
+ def delete_type(
141
+ options: OperationOptions,
142
+ input_: capo_keyspaces.types.delete_type_request.DeleteTypeRequest,
143
+ ) -> tuple[
144
+ capo_keyspaces.types.delete_type_response.DeleteTypeResponse, zapros.Response
145
+ ]:
146
+ response = options.client.handler.handle(build_request(options, input_))
147
+ try:
148
+ if response.status >= 300:
149
+ response.read()
150
+ handle_error(response)
151
+ return handle_response(response), response
152
+ except BaseException:
153
+ response.close()
154
+ raise
155
+
156
+
157
+ async def async_delete_type(
158
+ options: AsyncOperationOptions,
159
+ input_: capo_keyspaces.types.delete_type_request.DeleteTypeRequest,
160
+ ) -> tuple[
161
+ capo_keyspaces.types.delete_type_response.DeleteTypeResponse, zapros.Response
162
+ ]:
163
+ response = await options.client.handler.ahandle(build_request(options, input_))
164
+ try:
165
+ if response.status >= 300:
166
+ await response.aread()
167
+ handle_error(response)
168
+ return await async_handle_response(response), response
169
+ except BaseException:
170
+ await response.aclose()
171
+ raise
@@ -0,0 +1,168 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#GetKeyspace``."""
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.internal_server_exception
16
+ import capo_keyspaces.errors.resource_not_found_exception
17
+ import capo_keyspaces.errors.service_quota_exceeded_exception
18
+ import capo_keyspaces.errors.validation_exception
19
+ import capo_keyspaces.types.get_keyspace_request
20
+ import capo_keyspaces.types.get_keyspace_response
21
+ import capo_keyspaces.types.region_list
22
+ import capo_keyspaces.types.replication_group_status_list
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 "InternalServerException":
38
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
39
+ data, message
40
+ )
41
+ case "ResourceNotFoundException":
42
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
43
+ data, message
44
+ )
45
+ case "ServiceQuotaExceededException":
46
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
47
+ data, message
48
+ )
49
+ case "ValidationException":
50
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
51
+ data, message
52
+ )
53
+ case _:
54
+ raise UnknownServiceError(code=code, message=message, response=response)
55
+
56
+
57
+ def handle_response(
58
+ response: zapros.Response,
59
+ ) -> capo_keyspaces.types.get_keyspace_response.GetKeyspaceResponse:
60
+ out: capo_keyspaces.types.get_keyspace_response.GetKeyspaceResponse = (
61
+ capo_keyspaces.types.get_keyspace_response.deserialize_aws_json_1_0(
62
+ json.loads(response.read())
63
+ )
64
+ )
65
+ return out
66
+
67
+
68
+ async def async_handle_response(
69
+ response: zapros.Response,
70
+ ) -> capo_keyspaces.types.get_keyspace_response.GetKeyspaceResponse:
71
+ out: capo_keyspaces.types.get_keyspace_response.GetKeyspaceResponse = (
72
+ capo_keyspaces.types.get_keyspace_response.deserialize_aws_json_1_0(
73
+ json.loads(await response.aread())
74
+ )
75
+ )
76
+ return out
77
+
78
+
79
+ def get_signer(
80
+ options: AsyncOperationOptions | OperationOptions,
81
+ auth_schemes: list[dict[str, Any]] | None = None,
82
+ ) -> capo_keyspaces._auth._signers.Signer | None:
83
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
84
+ if (
85
+ options.credentials_provider is not None
86
+ and name_to_schema
87
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
88
+ ):
89
+ raise RuntimeError(
90
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
91
+ )
92
+ if options.credentials_provider is not None:
93
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
94
+ "sigv4-s3express"
95
+ )
96
+ if endpoint_scheme is not None or not name_to_schema:
97
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
98
+ "cassandra", options.region, endpoint_scheme
99
+ )
100
+ if sigv4_config is not None:
101
+ return capo_keyspaces._auth._signers.SigV4Signer(
102
+ options.credentials_provider, auth_scheme=sigv4_config
103
+ )
104
+ raise RuntimeError("Auth was not resolved")
105
+
106
+
107
+ def build_request(
108
+ options: OperationOptions | AsyncOperationOptions,
109
+ input_: capo_keyspaces.types.get_keyspace_request.GetKeyspaceRequest,
110
+ ) -> zapros.Request:
111
+ endpoint = resolve(
112
+ EndpointParams(
113
+ Region=options.region,
114
+ UseDualStack=options.use_dual_stack,
115
+ UseFIPS=options.use_fips,
116
+ Endpoint=options.endpoint,
117
+ )
118
+ ) # noqa: F841
119
+ url = endpoint.url.rstrip("/") + ""
120
+ params: list[tuple[str, str]] = []
121
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
122
+ headers["X-Amz-Target"] = "KeyspacesService.GetKeyspace"
123
+ body: bytes | None = json.dumps(
124
+ capo_keyspaces.types.get_keyspace_request.serialize_aws_json_1_0(input_),
125
+ allow_nan=False,
126
+ ).encode()
127
+ headers["content-type"] = "application/x-amz-json-1.0"
128
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
129
+ normalized_url = zapros.URL(url)
130
+ for k, v in params:
131
+ normalized_url.search_params.append(k, v)
132
+ return zapros.Request(
133
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
134
+ )
135
+
136
+
137
+ def get_keyspace(
138
+ options: OperationOptions,
139
+ input_: capo_keyspaces.types.get_keyspace_request.GetKeyspaceRequest,
140
+ ) -> tuple[
141
+ capo_keyspaces.types.get_keyspace_response.GetKeyspaceResponse, zapros.Response
142
+ ]:
143
+ response = options.client.handler.handle(build_request(options, input_))
144
+ try:
145
+ if response.status >= 300:
146
+ response.read()
147
+ handle_error(response)
148
+ return handle_response(response), response
149
+ except BaseException:
150
+ response.close()
151
+ raise
152
+
153
+
154
+ async def async_get_keyspace(
155
+ options: AsyncOperationOptions,
156
+ input_: capo_keyspaces.types.get_keyspace_request.GetKeyspaceRequest,
157
+ ) -> tuple[
158
+ capo_keyspaces.types.get_keyspace_response.GetKeyspaceResponse, zapros.Response
159
+ ]:
160
+ response = await options.client.handler.ahandle(build_request(options, input_))
161
+ try:
162
+ if response.status >= 300:
163
+ await response.aread()
164
+ handle_error(response)
165
+ return await async_handle_response(response), response
166
+ except BaseException:
167
+ await response.aclose()
168
+ raise