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,173 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#GetTable``."""
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.capacity_specification_summary
20
+ import capo_keyspaces.types.cdc_specification_summary
21
+ import capo_keyspaces.types.client_side_timestamps
22
+ import capo_keyspaces.types.comment
23
+ import capo_keyspaces.types.encryption_specification
24
+ import capo_keyspaces.types.get_table_request
25
+ import capo_keyspaces.types.get_table_response
26
+ import capo_keyspaces.types.point_in_time_recovery_summary
27
+ import capo_keyspaces.types.replica_specification_summary_list
28
+ import capo_keyspaces.types.schema_definition
29
+ import capo_keyspaces.types.time_to_live
30
+ import capo_keyspaces.types.timestamp
31
+ import capo_keyspaces.types.warm_throughput_specification_summary
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 "InternalServerException":
47
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
48
+ data, message
49
+ )
50
+ case "ResourceNotFoundException":
51
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
52
+ data, message
53
+ )
54
+ case "ServiceQuotaExceededException":
55
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
56
+ data, message
57
+ )
58
+ case "ValidationException":
59
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
60
+ data, message
61
+ )
62
+ case _:
63
+ raise UnknownServiceError(code=code, message=message, response=response)
64
+
65
+
66
+ def handle_response(
67
+ response: zapros.Response,
68
+ ) -> capo_keyspaces.types.get_table_response.GetTableResponse:
69
+ out: capo_keyspaces.types.get_table_response.GetTableResponse = (
70
+ capo_keyspaces.types.get_table_response.deserialize_aws_json_1_0(
71
+ json.loads(response.read())
72
+ )
73
+ )
74
+ return out
75
+
76
+
77
+ async def async_handle_response(
78
+ response: zapros.Response,
79
+ ) -> capo_keyspaces.types.get_table_response.GetTableResponse:
80
+ out: capo_keyspaces.types.get_table_response.GetTableResponse = (
81
+ capo_keyspaces.types.get_table_response.deserialize_aws_json_1_0(
82
+ json.loads(await response.aread())
83
+ )
84
+ )
85
+ return out
86
+
87
+
88
+ def get_signer(
89
+ options: AsyncOperationOptions | OperationOptions,
90
+ auth_schemes: list[dict[str, Any]] | None = None,
91
+ ) -> capo_keyspaces._auth._signers.Signer | None:
92
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
93
+ if (
94
+ options.credentials_provider is not None
95
+ and name_to_schema
96
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
97
+ ):
98
+ raise RuntimeError(
99
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
100
+ )
101
+ if options.credentials_provider is not None:
102
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
103
+ "sigv4-s3express"
104
+ )
105
+ if endpoint_scheme is not None or not name_to_schema:
106
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
107
+ "cassandra", options.region, endpoint_scheme
108
+ )
109
+ if sigv4_config is not None:
110
+ return capo_keyspaces._auth._signers.SigV4Signer(
111
+ options.credentials_provider, auth_scheme=sigv4_config
112
+ )
113
+ raise RuntimeError("Auth was not resolved")
114
+
115
+
116
+ def build_request(
117
+ options: OperationOptions | AsyncOperationOptions,
118
+ input_: capo_keyspaces.types.get_table_request.GetTableRequest,
119
+ ) -> zapros.Request:
120
+ endpoint = resolve(
121
+ EndpointParams(
122
+ Region=options.region,
123
+ UseDualStack=options.use_dual_stack,
124
+ UseFIPS=options.use_fips,
125
+ Endpoint=options.endpoint,
126
+ )
127
+ ) # noqa: F841
128
+ url = endpoint.url.rstrip("/") + ""
129
+ params: list[tuple[str, str]] = []
130
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
131
+ headers["X-Amz-Target"] = "KeyspacesService.GetTable"
132
+ body: bytes | None = json.dumps(
133
+ capo_keyspaces.types.get_table_request.serialize_aws_json_1_0(input_),
134
+ allow_nan=False,
135
+ ).encode()
136
+ headers["content-type"] = "application/x-amz-json-1.0"
137
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
138
+ normalized_url = zapros.URL(url)
139
+ for k, v in params:
140
+ normalized_url.search_params.append(k, v)
141
+ return zapros.Request(
142
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
143
+ )
144
+
145
+
146
+ def get_table(
147
+ options: OperationOptions,
148
+ input_: capo_keyspaces.types.get_table_request.GetTableRequest,
149
+ ) -> tuple[capo_keyspaces.types.get_table_response.GetTableResponse, zapros.Response]:
150
+ response = options.client.handler.handle(build_request(options, input_))
151
+ try:
152
+ if response.status >= 300:
153
+ response.read()
154
+ handle_error(response)
155
+ return handle_response(response), response
156
+ except BaseException:
157
+ response.close()
158
+ raise
159
+
160
+
161
+ async def async_get_table(
162
+ options: AsyncOperationOptions,
163
+ input_: capo_keyspaces.types.get_table_request.GetTableRequest,
164
+ ) -> tuple[capo_keyspaces.types.get_table_response.GetTableResponse, zapros.Response]:
165
+ response = await options.client.handler.ahandle(build_request(options, input_))
166
+ try:
167
+ if response.status >= 300:
168
+ await response.aread()
169
+ handle_error(response)
170
+ return await async_handle_response(response), response
171
+ except BaseException:
172
+ await response.aclose()
173
+ raise
@@ -0,0 +1,168 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#GetTableAutoScalingSettings``."""
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.auto_scaling_specification
20
+ import capo_keyspaces.types.get_table_auto_scaling_settings_request
21
+ import capo_keyspaces.types.get_table_auto_scaling_settings_response
22
+ import capo_keyspaces.types.replica_auto_scaling_specification_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_table_auto_scaling_settings_response.GetTableAutoScalingSettingsResponse:
60
+ out: capo_keyspaces.types.get_table_auto_scaling_settings_response.GetTableAutoScalingSettingsResponse = capo_keyspaces.types.get_table_auto_scaling_settings_response.deserialize_aws_json_1_0(
61
+ json.loads(response.read())
62
+ )
63
+ return out
64
+
65
+
66
+ async def async_handle_response(
67
+ response: zapros.Response,
68
+ ) -> capo_keyspaces.types.get_table_auto_scaling_settings_response.GetTableAutoScalingSettingsResponse:
69
+ out: capo_keyspaces.types.get_table_auto_scaling_settings_response.GetTableAutoScalingSettingsResponse = capo_keyspaces.types.get_table_auto_scaling_settings_response.deserialize_aws_json_1_0(
70
+ json.loads(await response.aread())
71
+ )
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.get_table_auto_scaling_settings_request.GetTableAutoScalingSettingsRequest,
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.GetTableAutoScalingSettings"
119
+ body: bytes | None = json.dumps(
120
+ capo_keyspaces.types.get_table_auto_scaling_settings_request.serialize_aws_json_1_0(
121
+ input_
122
+ ),
123
+ allow_nan=False,
124
+ ).encode()
125
+ headers["content-type"] = "application/x-amz-json-1.0"
126
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
127
+ normalized_url = zapros.URL(url)
128
+ for k, v in params:
129
+ normalized_url.search_params.append(k, v)
130
+ return zapros.Request(
131
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
132
+ )
133
+
134
+
135
+ def get_table_auto_scaling_settings(
136
+ options: OperationOptions,
137
+ input_: capo_keyspaces.types.get_table_auto_scaling_settings_request.GetTableAutoScalingSettingsRequest,
138
+ ) -> tuple[
139
+ capo_keyspaces.types.get_table_auto_scaling_settings_response.GetTableAutoScalingSettingsResponse,
140
+ zapros.Response,
141
+ ]:
142
+ response = options.client.handler.handle(build_request(options, input_))
143
+ try:
144
+ if response.status >= 300:
145
+ response.read()
146
+ handle_error(response)
147
+ return handle_response(response), response
148
+ except BaseException:
149
+ response.close()
150
+ raise
151
+
152
+
153
+ async def async_get_table_auto_scaling_settings(
154
+ options: AsyncOperationOptions,
155
+ input_: capo_keyspaces.types.get_table_auto_scaling_settings_request.GetTableAutoScalingSettingsRequest,
156
+ ) -> tuple[
157
+ capo_keyspaces.types.get_table_auto_scaling_settings_response.GetTableAutoScalingSettingsResponse,
158
+ 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
@@ -0,0 +1,166 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#GetType``."""
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.field_list
20
+ import capo_keyspaces.types.get_type_request
21
+ import capo_keyspaces.types.get_type_response
22
+ import capo_keyspaces.types.table_name_list
23
+ import capo_keyspaces.types.timestamp
24
+ import capo_keyspaces.types.type_name_list
25
+ from capo_keyspaces._protocol.errors import parse_error_metadata_json
26
+ from capo_keyspaces._rule_engine._endpoint_rule_set import EndpointParams, resolve
27
+ from capo_keyspaces._services._pipeline import AsyncOperationOptions, OperationOptions
28
+ from capo_keyspaces.errors import UnknownServiceError
29
+
30
+
31
+ def handle_error(response: zapros.Response) -> Never:
32
+ data = json.loads(response.read())
33
+ code, message = parse_error_metadata_json(response, data)
34
+ match code:
35
+ case "AccessDeniedException":
36
+ raise capo_keyspaces.errors.access_denied_exception.AccessDeniedException.from_aws_json_1_0(
37
+ data, message
38
+ )
39
+ case "InternalServerException":
40
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
41
+ data, message
42
+ )
43
+ case "ResourceNotFoundException":
44
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
45
+ data, message
46
+ )
47
+ case "ServiceQuotaExceededException":
48
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
49
+ data, message
50
+ )
51
+ case "ValidationException":
52
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
53
+ data, message
54
+ )
55
+ case _:
56
+ raise UnknownServiceError(code=code, message=message, response=response)
57
+
58
+
59
+ def handle_response(
60
+ response: zapros.Response,
61
+ ) -> capo_keyspaces.types.get_type_response.GetTypeResponse:
62
+ out: capo_keyspaces.types.get_type_response.GetTypeResponse = (
63
+ capo_keyspaces.types.get_type_response.deserialize_aws_json_1_0(
64
+ json.loads(response.read())
65
+ )
66
+ )
67
+ return out
68
+
69
+
70
+ async def async_handle_response(
71
+ response: zapros.Response,
72
+ ) -> capo_keyspaces.types.get_type_response.GetTypeResponse:
73
+ out: capo_keyspaces.types.get_type_response.GetTypeResponse = (
74
+ capo_keyspaces.types.get_type_response.deserialize_aws_json_1_0(
75
+ json.loads(await response.aread())
76
+ )
77
+ )
78
+ return out
79
+
80
+
81
+ def get_signer(
82
+ options: AsyncOperationOptions | OperationOptions,
83
+ auth_schemes: list[dict[str, Any]] | None = None,
84
+ ) -> capo_keyspaces._auth._signers.Signer | None:
85
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
86
+ if (
87
+ options.credentials_provider is not None
88
+ and name_to_schema
89
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
90
+ ):
91
+ raise RuntimeError(
92
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
93
+ )
94
+ if options.credentials_provider is not None:
95
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
96
+ "sigv4-s3express"
97
+ )
98
+ if endpoint_scheme is not None or not name_to_schema:
99
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
100
+ "cassandra", options.region, endpoint_scheme
101
+ )
102
+ if sigv4_config is not None:
103
+ return capo_keyspaces._auth._signers.SigV4Signer(
104
+ options.credentials_provider, auth_scheme=sigv4_config
105
+ )
106
+ raise RuntimeError("Auth was not resolved")
107
+
108
+
109
+ def build_request(
110
+ options: OperationOptions | AsyncOperationOptions,
111
+ input_: capo_keyspaces.types.get_type_request.GetTypeRequest,
112
+ ) -> zapros.Request:
113
+ endpoint = resolve(
114
+ EndpointParams(
115
+ Region=options.region,
116
+ UseDualStack=options.use_dual_stack,
117
+ UseFIPS=options.use_fips,
118
+ Endpoint=options.endpoint,
119
+ )
120
+ ) # noqa: F841
121
+ url = endpoint.url.rstrip("/") + ""
122
+ params: list[tuple[str, str]] = []
123
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
124
+ headers["X-Amz-Target"] = "KeyspacesService.GetType"
125
+ body: bytes | None = json.dumps(
126
+ capo_keyspaces.types.get_type_request.serialize_aws_json_1_0(input_),
127
+ allow_nan=False,
128
+ ).encode()
129
+ headers["content-type"] = "application/x-amz-json-1.0"
130
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
131
+ normalized_url = zapros.URL(url)
132
+ for k, v in params:
133
+ normalized_url.search_params.append(k, v)
134
+ return zapros.Request(
135
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
136
+ )
137
+
138
+
139
+ def get_type(
140
+ options: OperationOptions,
141
+ input_: capo_keyspaces.types.get_type_request.GetTypeRequest,
142
+ ) -> tuple[capo_keyspaces.types.get_type_response.GetTypeResponse, zapros.Response]:
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_type(
155
+ options: AsyncOperationOptions,
156
+ input_: capo_keyspaces.types.get_type_request.GetTypeRequest,
157
+ ) -> tuple[capo_keyspaces.types.get_type_response.GetTypeResponse, zapros.Response]:
158
+ response = await options.client.handler.ahandle(build_request(options, input_))
159
+ try:
160
+ if response.status >= 300:
161
+ await response.aread()
162
+ handle_error(response)
163
+ return await async_handle_response(response), response
164
+ except BaseException:
165
+ await response.aclose()
166
+ raise
@@ -0,0 +1,167 @@
1
+ """Generated from Smithy shape ``com.amazonaws.keyspaces#ListKeyspaces``."""
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.keyspace_summary_list
20
+ import capo_keyspaces.types.list_keyspaces_request
21
+ import capo_keyspaces.types.list_keyspaces_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 "InternalServerException":
37
+ raise capo_keyspaces.errors.internal_server_exception.InternalServerException.from_aws_json_1_0(
38
+ data, message
39
+ )
40
+ case "ResourceNotFoundException":
41
+ raise capo_keyspaces.errors.resource_not_found_exception.ResourceNotFoundException.from_aws_json_1_0(
42
+ data, message
43
+ )
44
+ case "ServiceQuotaExceededException":
45
+ raise capo_keyspaces.errors.service_quota_exceeded_exception.ServiceQuotaExceededException.from_aws_json_1_0(
46
+ data, message
47
+ )
48
+ case "ValidationException":
49
+ raise capo_keyspaces.errors.validation_exception.ValidationException.from_aws_json_1_0(
50
+ data, message
51
+ )
52
+ case _:
53
+ raise UnknownServiceError(code=code, message=message, response=response)
54
+
55
+
56
+ def handle_response(
57
+ response: zapros.Response,
58
+ ) -> capo_keyspaces.types.list_keyspaces_response.ListKeyspacesResponse:
59
+ out: capo_keyspaces.types.list_keyspaces_response.ListKeyspacesResponse = (
60
+ capo_keyspaces.types.list_keyspaces_response.deserialize_aws_json_1_0(
61
+ json.loads(response.read())
62
+ )
63
+ )
64
+ return out
65
+
66
+
67
+ async def async_handle_response(
68
+ response: zapros.Response,
69
+ ) -> capo_keyspaces.types.list_keyspaces_response.ListKeyspacesResponse:
70
+ out: capo_keyspaces.types.list_keyspaces_response.ListKeyspacesResponse = (
71
+ capo_keyspaces.types.list_keyspaces_response.deserialize_aws_json_1_0(
72
+ json.loads(await response.aread())
73
+ )
74
+ )
75
+ return out
76
+
77
+
78
+ def get_signer(
79
+ options: AsyncOperationOptions | OperationOptions,
80
+ auth_schemes: list[dict[str, Any]] | None = None,
81
+ ) -> capo_keyspaces._auth._signers.Signer | None:
82
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
83
+ if (
84
+ options.credentials_provider is not None
85
+ and name_to_schema
86
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
87
+ ):
88
+ raise RuntimeError(
89
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
90
+ )
91
+ if options.credentials_provider is not None:
92
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
93
+ "sigv4-s3express"
94
+ )
95
+ if endpoint_scheme is not None or not name_to_schema:
96
+ sigv4_config = capo_keyspaces._auth._sigv4.build_sigv4_auth_scheme(
97
+ "cassandra", options.region, endpoint_scheme
98
+ )
99
+ if sigv4_config is not None:
100
+ return capo_keyspaces._auth._signers.SigV4Signer(
101
+ options.credentials_provider, auth_scheme=sigv4_config
102
+ )
103
+ raise RuntimeError("Auth was not resolved")
104
+
105
+
106
+ def build_request(
107
+ options: OperationOptions | AsyncOperationOptions,
108
+ input_: capo_keyspaces.types.list_keyspaces_request.ListKeyspacesRequest,
109
+ ) -> zapros.Request:
110
+ endpoint = resolve(
111
+ EndpointParams(
112
+ Region=options.region,
113
+ UseDualStack=options.use_dual_stack,
114
+ UseFIPS=options.use_fips,
115
+ Endpoint=options.endpoint,
116
+ )
117
+ ) # noqa: F841
118
+ url = endpoint.url.rstrip("/") + ""
119
+ params: list[tuple[str, str]] = []
120
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
121
+ headers["X-Amz-Target"] = "KeyspacesService.ListKeyspaces"
122
+ body: bytes | None = json.dumps(
123
+ capo_keyspaces.types.list_keyspaces_request.serialize_aws_json_1_0(input_),
124
+ allow_nan=False,
125
+ ).encode()
126
+ headers["content-type"] = "application/x-amz-json-1.0"
127
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
128
+ normalized_url = zapros.URL(url)
129
+ for k, v in params:
130
+ normalized_url.search_params.append(k, v)
131
+ return zapros.Request(
132
+ normalized_url, "POST", headers=headers, body=body, context={"signer": signer}
133
+ )
134
+
135
+
136
+ def list_keyspaces(
137
+ options: OperationOptions,
138
+ input_: capo_keyspaces.types.list_keyspaces_request.ListKeyspacesRequest,
139
+ ) -> tuple[
140
+ capo_keyspaces.types.list_keyspaces_response.ListKeyspacesResponse, zapros.Response
141
+ ]:
142
+ response = options.client.handler.handle(build_request(options, input_))
143
+ try:
144
+ if response.status >= 300:
145
+ response.read()
146
+ handle_error(response)
147
+ return handle_response(response), response
148
+ except BaseException:
149
+ response.close()
150
+ raise
151
+
152
+
153
+ async def async_list_keyspaces(
154
+ options: AsyncOperationOptions,
155
+ input_: capo_keyspaces.types.list_keyspaces_request.ListKeyspacesRequest,
156
+ ) -> tuple[
157
+ capo_keyspaces.types.list_keyspaces_response.ListKeyspacesResponse, zapros.Response
158
+ ]:
159
+ response = await options.client.handler.ahandle(build_request(options, input_))
160
+ try:
161
+ if response.status >= 300:
162
+ await response.aread()
163
+ handle_error(response)
164
+ return await async_handle_response(response), response
165
+ except BaseException:
166
+ await response.aclose()
167
+ raise