capo-codeguru-security 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 (132) hide show
  1. capo_codeguru_security/__init__.py +74 -0
  2. capo_codeguru_security/_async.py +25 -0
  3. capo_codeguru_security/_auth/_identity.py +16 -0
  4. capo_codeguru_security/_auth/_providers.py +886 -0
  5. capo_codeguru_security/_auth/_signers.py +88 -0
  6. capo_codeguru_security/_auth/_sigv4.py +434 -0
  7. capo_codeguru_security/_auth/_zapros_handler.py +80 -0
  8. capo_codeguru_security/_iter.py +113 -0
  9. capo_codeguru_security/_operations/aws_code_guru_security/batch_get_findings.py +167 -0
  10. capo_codeguru_security/_operations/aws_code_guru_security/create_scan.py +183 -0
  11. capo_codeguru_security/_operations/aws_code_guru_security/create_upload_url.py +165 -0
  12. capo_codeguru_security/_operations/aws_code_guru_security/get_account_configuration.py +161 -0
  13. capo_codeguru_security/_operations/aws_code_guru_security/get_findings.py +191 -0
  14. capo_codeguru_security/_operations/aws_code_guru_security/get_metrics_summary.py +176 -0
  15. capo_codeguru_security/_operations/aws_code_guru_security/get_scan.py +173 -0
  16. capo_codeguru_security/_operations/aws_code_guru_security/list_findings_metrics.py +185 -0
  17. capo_codeguru_security/_operations/aws_code_guru_security/list_scans.py +167 -0
  18. capo_codeguru_security/_operations/aws_code_guru_security/list_tags_for_resource.py +173 -0
  19. capo_codeguru_security/_operations/aws_code_guru_security/tag_resource.py +173 -0
  20. capo_codeguru_security/_operations/aws_code_guru_security/untag_resource.py +171 -0
  21. capo_codeguru_security/_operations/aws_code_guru_security/update_account_configuration.py +172 -0
  22. capo_codeguru_security/_pagination.py +21 -0
  23. capo_codeguru_security/_protocol/__init__.py +1 -0
  24. capo_codeguru_security/_protocol/errors.py +93 -0
  25. capo_codeguru_security/_protocol/eventstream.py +238 -0
  26. capo_codeguru_security/_protocol/serialize.py +47 -0
  27. capo_codeguru_security/_protocol/xml.py +33 -0
  28. capo_codeguru_security/_rule_engine/__init__.py +0 -0
  29. capo_codeguru_security/_rule_engine/_aws_partition.py +160 -0
  30. capo_codeguru_security/_rule_engine/_endpoint_rule_set.py +137 -0
  31. capo_codeguru_security/_rule_engine/_endpoint_runtime.py +389 -0
  32. capo_codeguru_security/_services/_aws_config.py +160 -0
  33. capo_codeguru_security/_services/_pipeline.py +198 -0
  34. capo_codeguru_security/_services/async_code_guru_security.py +934 -0
  35. capo_codeguru_security/_services/code_guru_security.py +921 -0
  36. capo_codeguru_security/errors/__init__.py +33 -0
  37. capo_codeguru_security/errors/_base.py +94 -0
  38. capo_codeguru_security/errors/access_denied_exception.py +67 -0
  39. capo_codeguru_security/errors/conflict_exception.py +67 -0
  40. capo_codeguru_security/errors/internal_server_exception.py +53 -0
  41. capo_codeguru_security/errors/resource_not_found_exception.py +69 -0
  42. capo_codeguru_security/errors/throttling_exception.py +65 -0
  43. capo_codeguru_security/errors/validation_exception.py +98 -0
  44. capo_codeguru_security/py.typed +0 -0
  45. capo_codeguru_security/types/_prelude/blob.py +12 -0
  46. capo_codeguru_security/types/_prelude/timestamp.py +17 -0
  47. capo_codeguru_security/types/account_findings_metric.py +118 -0
  48. capo_codeguru_security/types/analysis_type.py +17 -0
  49. capo_codeguru_security/types/batch_get_findings_error.py +61 -0
  50. capo_codeguru_security/types/batch_get_findings_errors.py +35 -0
  51. capo_codeguru_security/types/batch_get_findings_request.py +47 -0
  52. capo_codeguru_security/types/batch_get_findings_response.py +61 -0
  53. capo_codeguru_security/types/categories_with_most_findings.py +37 -0
  54. capo_codeguru_security/types/category_with_finding_num.py +29 -0
  55. capo_codeguru_security/types/client_token.py +5 -0
  56. capo_codeguru_security/types/code_line.py +29 -0
  57. capo_codeguru_security/types/code_snippet.py +29 -0
  58. capo_codeguru_security/types/create_scan_request.py +101 -0
  59. capo_codeguru_security/types/create_scan_response.py +78 -0
  60. capo_codeguru_security/types/create_upload_url_request.py +31 -0
  61. capo_codeguru_security/types/create_upload_url_response.py +59 -0
  62. capo_codeguru_security/types/detector_tags.py +14 -0
  63. capo_codeguru_security/types/encryption_config.py +28 -0
  64. capo_codeguru_security/types/error_code.py +20 -0
  65. capo_codeguru_security/types/error_message.py +5 -0
  66. capo_codeguru_security/types/file_path.py +62 -0
  67. capo_codeguru_security/types/finding.py +207 -0
  68. capo_codeguru_security/types/finding_identifier.py +33 -0
  69. capo_codeguru_security/types/finding_identifiers.py +33 -0
  70. capo_codeguru_security/types/finding_metrics_value_per_severity.py +87 -0
  71. capo_codeguru_security/types/findings.py +29 -0
  72. capo_codeguru_security/types/findings_metric_list.py +35 -0
  73. capo_codeguru_security/types/get_account_configuration_request.py +18 -0
  74. capo_codeguru_security/types/get_account_configuration_response.py +45 -0
  75. capo_codeguru_security/types/get_findings_request.py +32 -0
  76. capo_codeguru_security/types/get_findings_response.py +43 -0
  77. capo_codeguru_security/types/get_metrics_summary_request.py +24 -0
  78. capo_codeguru_security/types/get_metrics_summary_response.py +42 -0
  79. capo_codeguru_security/types/get_scan_request.py +27 -0
  80. capo_codeguru_security/types/get_scan_response.py +132 -0
  81. capo_codeguru_security/types/header_key.py +5 -0
  82. capo_codeguru_security/types/header_value.py +5 -0
  83. capo_codeguru_security/types/kms_key_arn.py +5 -0
  84. capo_codeguru_security/types/list_findings_metrics_request.py +32 -0
  85. capo_codeguru_security/types/list_findings_metrics_response.py +49 -0
  86. capo_codeguru_security/types/list_scans_request.py +26 -0
  87. capo_codeguru_security/types/list_scans_response.py +43 -0
  88. capo_codeguru_security/types/list_tags_for_resource_request.py +24 -0
  89. capo_codeguru_security/types/list_tags_for_resource_response.py +34 -0
  90. capo_codeguru_security/types/metrics_summary.py +121 -0
  91. capo_codeguru_security/types/next_token.py +5 -0
  92. capo_codeguru_security/types/recommendation.py +29 -0
  93. capo_codeguru_security/types/reference_urls.py +14 -0
  94. capo_codeguru_security/types/related_vulnerabilities.py +14 -0
  95. capo_codeguru_security/types/remediation.py +63 -0
  96. capo_codeguru_security/types/request_header_map.py +29 -0
  97. capo_codeguru_security/types/resource.py +29 -0
  98. capo_codeguru_security/types/resource_id.py +32 -0
  99. capo_codeguru_security/types/s3_url.py +5 -0
  100. capo_codeguru_security/types/scan_name.py +5 -0
  101. capo_codeguru_security/types/scan_name_arn.py +5 -0
  102. capo_codeguru_security/types/scan_name_with_finding_num.py +29 -0
  103. capo_codeguru_security/types/scan_state.py +18 -0
  104. capo_codeguru_security/types/scan_summaries.py +29 -0
  105. capo_codeguru_security/types/scan_summary.py +99 -0
  106. capo_codeguru_security/types/scan_type.py +17 -0
  107. capo_codeguru_security/types/scans_with_most_open_critical_findings.py +37 -0
  108. capo_codeguru_security/types/scans_with_most_open_findings.py +37 -0
  109. capo_codeguru_security/types/severity.py +20 -0
  110. capo_codeguru_security/types/status.py +18 -0
  111. capo_codeguru_security/types/suggested_fix.py +29 -0
  112. capo_codeguru_security/types/suggested_fixes.py +31 -0
  113. capo_codeguru_security/types/tag_key.py +5 -0
  114. capo_codeguru_security/types/tag_key_list.py +17 -0
  115. capo_codeguru_security/types/tag_map.py +29 -0
  116. capo_codeguru_security/types/tag_resource_request.py +40 -0
  117. capo_codeguru_security/types/tag_resource_response.py +18 -0
  118. capo_codeguru_security/types/tag_value.py +5 -0
  119. capo_codeguru_security/types/untag_resource_request.py +27 -0
  120. capo_codeguru_security/types/untag_resource_response.py +18 -0
  121. capo_codeguru_security/types/update_account_configuration_request.py +45 -0
  122. capo_codeguru_security/types/update_account_configuration_response.py +45 -0
  123. capo_codeguru_security/types/uuid.py +5 -0
  124. capo_codeguru_security/types/validation_exception_field.py +33 -0
  125. capo_codeguru_security/types/validation_exception_field_list.py +37 -0
  126. capo_codeguru_security/types/validation_exception_reason.py +20 -0
  127. capo_codeguru_security/types/vulnerability.py +90 -0
  128. capo_codeguru_security-0.1.0.dist-info/METADATA +101 -0
  129. capo_codeguru_security-0.1.0.dist-info/RECORD +132 -0
  130. capo_codeguru_security-0.1.0.dist-info/WHEEL +5 -0
  131. capo_codeguru_security-0.1.0.dist-info/licenses/LICENSE +21 -0
  132. capo_codeguru_security-0.1.0.dist-info/top_level.txt +1 -0
@@ -0,0 +1,185 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#ListFindingsMetrics``."""
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_codeguru_security._auth._signers
12
+ import capo_codeguru_security._auth._sigv4
13
+ import capo_codeguru_security._protocol.eventstream
14
+ import capo_codeguru_security.errors.access_denied_exception
15
+ import capo_codeguru_security.errors.internal_server_exception
16
+ import capo_codeguru_security.errors.throttling_exception
17
+ import capo_codeguru_security.errors.validation_exception
18
+ import capo_codeguru_security.types.findings_metric_list
19
+ import capo_codeguru_security.types.list_findings_metrics_request
20
+ import capo_codeguru_security.types.list_findings_metrics_response
21
+ from capo_codeguru_security._protocol.errors import parse_error_metadata_json
22
+ from capo_codeguru_security._rule_engine._endpoint_rule_set import (
23
+ EndpointParams,
24
+ resolve,
25
+ )
26
+ from capo_codeguru_security._services._pipeline import (
27
+ AsyncOperationOptions,
28
+ OperationOptions,
29
+ )
30
+ from capo_codeguru_security.errors import UnknownServiceError
31
+
32
+
33
+ def handle_error(response: zapros.Response) -> Never:
34
+ data = json.loads(response.read())
35
+ code, message = parse_error_metadata_json(response, data)
36
+ match code:
37
+ case "AccessDeniedException":
38
+ raise capo_codeguru_security.errors.access_denied_exception.AccessDeniedException.from_json(
39
+ data, message
40
+ )
41
+ case "InternalServerException":
42
+ raise capo_codeguru_security.errors.internal_server_exception.InternalServerException.from_json(
43
+ data, message
44
+ )
45
+ case "ThrottlingException":
46
+ raise capo_codeguru_security.errors.throttling_exception.ThrottlingException.from_json(
47
+ data, message
48
+ )
49
+ case "ValidationException":
50
+ raise capo_codeguru_security.errors.validation_exception.ValidationException.from_json(
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_codeguru_security.types.list_findings_metrics_response.ListFindingsMetricsResponse:
60
+ out: capo_codeguru_security.types.list_findings_metrics_response.ListFindingsMetricsResponse = capo_codeguru_security.types.list_findings_metrics_response.deserialize_json(
61
+ json.loads(response.read())
62
+ )
63
+ return out
64
+
65
+
66
+ async def async_handle_response(
67
+ response: zapros.Response,
68
+ ) -> capo_codeguru_security.types.list_findings_metrics_response.ListFindingsMetricsResponse:
69
+ out: capo_codeguru_security.types.list_findings_metrics_response.ListFindingsMetricsResponse = capo_codeguru_security.types.list_findings_metrics_response.deserialize_json(
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_codeguru_security._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_codeguru_security._auth._sigv4.build_sigv4_auth_scheme(
94
+ "codeguru-security", options.region, endpoint_scheme
95
+ )
96
+ if sigv4_config is not None:
97
+ return capo_codeguru_security._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_codeguru_security.types.list_findings_metrics_request.ListFindingsMetricsRequest,
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
+ import capo_codeguru_security._protocol.serialize
116
+
117
+ url = endpoint.url.rstrip("/") + "/metrics/findings"
118
+ params: list[tuple[str, str]] = []
119
+ if "next_token" in input_:
120
+ params.append(("nextToken", input_["next_token"]))
121
+ if "max_results" in input_:
122
+ params.append(("maxResults", str(input_["max_results"])))
123
+ if "start_date" in input_:
124
+ params.append(
125
+ (
126
+ "startDate",
127
+ capo_codeguru_security._protocol.serialize.fmt_date_time(
128
+ input_["start_date"]
129
+ ),
130
+ )
131
+ )
132
+ if "end_date" in input_:
133
+ params.append(
134
+ (
135
+ "endDate",
136
+ capo_codeguru_security._protocol.serialize.fmt_date_time(
137
+ input_["end_date"]
138
+ ),
139
+ )
140
+ )
141
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
142
+ body: bytes | None = b""
143
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
144
+ normalized_url = zapros.URL(url)
145
+ for k, v in params:
146
+ normalized_url.search_params.append(k, v)
147
+ return zapros.Request(
148
+ normalized_url, "GET", headers=headers, body=body, context={"signer": signer}
149
+ )
150
+
151
+
152
+ def list_findings_metrics(
153
+ options: OperationOptions,
154
+ input_: capo_codeguru_security.types.list_findings_metrics_request.ListFindingsMetricsRequest,
155
+ ) -> tuple[
156
+ capo_codeguru_security.types.list_findings_metrics_response.ListFindingsMetricsResponse,
157
+ zapros.Response,
158
+ ]:
159
+ response = options.client.handler.handle(build_request(options, input_))
160
+ try:
161
+ if response.status >= 300:
162
+ response.read()
163
+ handle_error(response)
164
+ return handle_response(response), response
165
+ except BaseException:
166
+ response.close()
167
+ raise
168
+
169
+
170
+ async def async_list_findings_metrics(
171
+ options: AsyncOperationOptions,
172
+ input_: capo_codeguru_security.types.list_findings_metrics_request.ListFindingsMetricsRequest,
173
+ ) -> tuple[
174
+ capo_codeguru_security.types.list_findings_metrics_response.ListFindingsMetricsResponse,
175
+ zapros.Response,
176
+ ]:
177
+ response = await options.client.handler.ahandle(build_request(options, input_))
178
+ try:
179
+ if response.status >= 300:
180
+ await response.aread()
181
+ handle_error(response)
182
+ return await async_handle_response(response), response
183
+ except BaseException:
184
+ await response.aclose()
185
+ raise
@@ -0,0 +1,167 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#ListScans``."""
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_codeguru_security._auth._signers
12
+ import capo_codeguru_security._auth._sigv4
13
+ import capo_codeguru_security._protocol.eventstream
14
+ import capo_codeguru_security.errors.access_denied_exception
15
+ import capo_codeguru_security.errors.internal_server_exception
16
+ import capo_codeguru_security.errors.throttling_exception
17
+ import capo_codeguru_security.errors.validation_exception
18
+ import capo_codeguru_security.types.list_scans_request
19
+ import capo_codeguru_security.types.list_scans_response
20
+ import capo_codeguru_security.types.scan_summaries
21
+ from capo_codeguru_security._protocol.errors import parse_error_metadata_json
22
+ from capo_codeguru_security._rule_engine._endpoint_rule_set import (
23
+ EndpointParams,
24
+ resolve,
25
+ )
26
+ from capo_codeguru_security._services._pipeline import (
27
+ AsyncOperationOptions,
28
+ OperationOptions,
29
+ )
30
+ from capo_codeguru_security.errors import UnknownServiceError
31
+
32
+
33
+ def handle_error(response: zapros.Response) -> Never:
34
+ data = json.loads(response.read())
35
+ code, message = parse_error_metadata_json(response, data)
36
+ match code:
37
+ case "AccessDeniedException":
38
+ raise capo_codeguru_security.errors.access_denied_exception.AccessDeniedException.from_json(
39
+ data, message
40
+ )
41
+ case "InternalServerException":
42
+ raise capo_codeguru_security.errors.internal_server_exception.InternalServerException.from_json(
43
+ data, message
44
+ )
45
+ case "ThrottlingException":
46
+ raise capo_codeguru_security.errors.throttling_exception.ThrottlingException.from_json(
47
+ data, message
48
+ )
49
+ case "ValidationException":
50
+ raise capo_codeguru_security.errors.validation_exception.ValidationException.from_json(
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_codeguru_security.types.list_scans_response.ListScansResponse:
60
+ out: capo_codeguru_security.types.list_scans_response.ListScansResponse = (
61
+ capo_codeguru_security.types.list_scans_response.deserialize_json(
62
+ json.loads(response.read())
63
+ )
64
+ )
65
+ return out
66
+
67
+
68
+ async def async_handle_response(
69
+ response: zapros.Response,
70
+ ) -> capo_codeguru_security.types.list_scans_response.ListScansResponse:
71
+ out: capo_codeguru_security.types.list_scans_response.ListScansResponse = (
72
+ capo_codeguru_security.types.list_scans_response.deserialize_json(
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_codeguru_security._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_codeguru_security._auth._sigv4.build_sigv4_auth_scheme(
98
+ "codeguru-security", options.region, endpoint_scheme
99
+ )
100
+ if sigv4_config is not None:
101
+ return capo_codeguru_security._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_codeguru_security.types.list_scans_request.ListScansRequest,
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("/") + "/scans"
120
+ params: list[tuple[str, str]] = []
121
+ if "next_token" in input_:
122
+ params.append(("nextToken", input_["next_token"]))
123
+ if "max_results" in input_:
124
+ params.append(("maxResults", str(input_["max_results"])))
125
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
126
+ body: bytes | None = b""
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, "GET", headers=headers, body=body, context={"signer": signer}
133
+ )
134
+
135
+
136
+ def list_scans(
137
+ options: OperationOptions,
138
+ input_: capo_codeguru_security.types.list_scans_request.ListScansRequest,
139
+ ) -> tuple[
140
+ capo_codeguru_security.types.list_scans_response.ListScansResponse, 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_scans(
154
+ options: AsyncOperationOptions,
155
+ input_: capo_codeguru_security.types.list_scans_request.ListScansRequest,
156
+ ) -> tuple[
157
+ capo_codeguru_security.types.list_scans_response.ListScansResponse, 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
@@ -0,0 +1,173 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#ListTagsForResource``."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import json
6
+ from typing import Any
7
+ from urllib.parse import quote
8
+
9
+ import zapros
10
+ from typing_extensions import Never
11
+
12
+ import capo_codeguru_security._auth._signers
13
+ import capo_codeguru_security._auth._sigv4
14
+ import capo_codeguru_security._protocol.eventstream
15
+ import capo_codeguru_security.errors.access_denied_exception
16
+ import capo_codeguru_security.errors.conflict_exception
17
+ import capo_codeguru_security.errors.internal_server_exception
18
+ import capo_codeguru_security.errors.resource_not_found_exception
19
+ import capo_codeguru_security.errors.throttling_exception
20
+ import capo_codeguru_security.errors.validation_exception
21
+ import capo_codeguru_security.types.list_tags_for_resource_request
22
+ import capo_codeguru_security.types.list_tags_for_resource_response
23
+ import capo_codeguru_security.types.tag_map
24
+ from capo_codeguru_security._protocol.errors import parse_error_metadata_json
25
+ from capo_codeguru_security._rule_engine._endpoint_rule_set import (
26
+ EndpointParams,
27
+ resolve,
28
+ )
29
+ from capo_codeguru_security._services._pipeline import (
30
+ AsyncOperationOptions,
31
+ OperationOptions,
32
+ )
33
+ from capo_codeguru_security.errors import UnknownServiceError
34
+
35
+
36
+ def handle_error(response: zapros.Response) -> Never:
37
+ data = json.loads(response.read())
38
+ code, message = parse_error_metadata_json(response, data)
39
+ match code:
40
+ case "AccessDeniedException":
41
+ raise capo_codeguru_security.errors.access_denied_exception.AccessDeniedException.from_json(
42
+ data, message
43
+ )
44
+ case "ConflictException":
45
+ raise capo_codeguru_security.errors.conflict_exception.ConflictException.from_json(
46
+ data, message
47
+ )
48
+ case "InternalServerException":
49
+ raise capo_codeguru_security.errors.internal_server_exception.InternalServerException.from_json(
50
+ data, message
51
+ )
52
+ case "ResourceNotFoundException":
53
+ raise capo_codeguru_security.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
54
+ data, message
55
+ )
56
+ case "ThrottlingException":
57
+ raise capo_codeguru_security.errors.throttling_exception.ThrottlingException.from_json(
58
+ data, message
59
+ )
60
+ case "ValidationException":
61
+ raise capo_codeguru_security.errors.validation_exception.ValidationException.from_json(
62
+ data, message
63
+ )
64
+ case _:
65
+ raise UnknownServiceError(code=code, message=message, response=response)
66
+
67
+
68
+ def handle_response(
69
+ response: zapros.Response,
70
+ ) -> capo_codeguru_security.types.list_tags_for_resource_response.ListTagsForResourceResponse:
71
+ out: capo_codeguru_security.types.list_tags_for_resource_response.ListTagsForResourceResponse = capo_codeguru_security.types.list_tags_for_resource_response.deserialize_json(
72
+ json.loads(response.read())
73
+ )
74
+ return out
75
+
76
+
77
+ async def async_handle_response(
78
+ response: zapros.Response,
79
+ ) -> capo_codeguru_security.types.list_tags_for_resource_response.ListTagsForResourceResponse:
80
+ out: capo_codeguru_security.types.list_tags_for_resource_response.ListTagsForResourceResponse = capo_codeguru_security.types.list_tags_for_resource_response.deserialize_json(
81
+ json.loads(await response.aread())
82
+ )
83
+ return out
84
+
85
+
86
+ def get_signer(
87
+ options: AsyncOperationOptions | OperationOptions,
88
+ auth_schemes: list[dict[str, Any]] | None = None,
89
+ ) -> capo_codeguru_security._auth._signers.Signer | None:
90
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
91
+ if (
92
+ options.credentials_provider is not None
93
+ and name_to_schema
94
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
95
+ ):
96
+ raise RuntimeError(
97
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
98
+ )
99
+ if options.credentials_provider is not None:
100
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
101
+ "sigv4-s3express"
102
+ )
103
+ if endpoint_scheme is not None or not name_to_schema:
104
+ sigv4_config = capo_codeguru_security._auth._sigv4.build_sigv4_auth_scheme(
105
+ "codeguru-security", options.region, endpoint_scheme
106
+ )
107
+ if sigv4_config is not None:
108
+ return capo_codeguru_security._auth._signers.SigV4Signer(
109
+ options.credentials_provider, auth_scheme=sigv4_config
110
+ )
111
+ raise RuntimeError("Auth was not resolved")
112
+
113
+
114
+ def build_request(
115
+ options: OperationOptions | AsyncOperationOptions,
116
+ input_: capo_codeguru_security.types.list_tags_for_resource_request.ListTagsForResourceRequest,
117
+ ) -> zapros.Request:
118
+ endpoint = resolve(
119
+ EndpointParams(
120
+ Region=options.region,
121
+ UseDualStack=options.use_dual_stack,
122
+ UseFIPS=options.use_fips,
123
+ Endpoint=options.endpoint,
124
+ )
125
+ ) # noqa: F841
126
+ url = endpoint.url.rstrip("/") + "/tags/{resourceArn}"
127
+ url = url.replace("{resourceArn}", quote(input_["resource_arn"], safe=""))
128
+ params: list[tuple[str, str]] = []
129
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
130
+ body: bytes | None = b""
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, "GET", headers=headers, body=body, context={"signer": signer}
137
+ )
138
+
139
+
140
+ def list_tags_for_resource(
141
+ options: OperationOptions,
142
+ input_: capo_codeguru_security.types.list_tags_for_resource_request.ListTagsForResourceRequest,
143
+ ) -> tuple[
144
+ capo_codeguru_security.types.list_tags_for_resource_response.ListTagsForResourceResponse,
145
+ zapros.Response,
146
+ ]:
147
+ response = options.client.handler.handle(build_request(options, input_))
148
+ try:
149
+ if response.status >= 300:
150
+ response.read()
151
+ handle_error(response)
152
+ return handle_response(response), response
153
+ except BaseException:
154
+ response.close()
155
+ raise
156
+
157
+
158
+ async def async_list_tags_for_resource(
159
+ options: AsyncOperationOptions,
160
+ input_: capo_codeguru_security.types.list_tags_for_resource_request.ListTagsForResourceRequest,
161
+ ) -> tuple[
162
+ capo_codeguru_security.types.list_tags_for_resource_response.ListTagsForResourceResponse,
163
+ zapros.Response,
164
+ ]:
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,173 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#TagResource``."""
2
+
3
+ from __future__ import annotations
4
+
5
+ import json
6
+ from typing import Any
7
+ from urllib.parse import quote
8
+
9
+ import zapros
10
+ from typing_extensions import Never
11
+
12
+ import capo_codeguru_security._auth._signers
13
+ import capo_codeguru_security._auth._sigv4
14
+ import capo_codeguru_security._protocol.eventstream
15
+ import capo_codeguru_security.errors.access_denied_exception
16
+ import capo_codeguru_security.errors.conflict_exception
17
+ import capo_codeguru_security.errors.internal_server_exception
18
+ import capo_codeguru_security.errors.resource_not_found_exception
19
+ import capo_codeguru_security.errors.throttling_exception
20
+ import capo_codeguru_security.errors.validation_exception
21
+ import capo_codeguru_security.types.tag_map
22
+ import capo_codeguru_security.types.tag_resource_request
23
+ import capo_codeguru_security.types.tag_resource_response
24
+ from capo_codeguru_security._protocol.errors import parse_error_metadata_json
25
+ from capo_codeguru_security._rule_engine._endpoint_rule_set import (
26
+ EndpointParams,
27
+ resolve,
28
+ )
29
+ from capo_codeguru_security._services._pipeline import (
30
+ AsyncOperationOptions,
31
+ OperationOptions,
32
+ )
33
+ from capo_codeguru_security.errors import UnknownServiceError
34
+
35
+
36
+ def handle_error(response: zapros.Response) -> Never:
37
+ data = json.loads(response.read())
38
+ code, message = parse_error_metadata_json(response, data)
39
+ match code:
40
+ case "AccessDeniedException":
41
+ raise capo_codeguru_security.errors.access_denied_exception.AccessDeniedException.from_json(
42
+ data, message
43
+ )
44
+ case "ConflictException":
45
+ raise capo_codeguru_security.errors.conflict_exception.ConflictException.from_json(
46
+ data, message
47
+ )
48
+ case "InternalServerException":
49
+ raise capo_codeguru_security.errors.internal_server_exception.InternalServerException.from_json(
50
+ data, message
51
+ )
52
+ case "ResourceNotFoundException":
53
+ raise capo_codeguru_security.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
54
+ data, message
55
+ )
56
+ case "ThrottlingException":
57
+ raise capo_codeguru_security.errors.throttling_exception.ThrottlingException.from_json(
58
+ data, message
59
+ )
60
+ case "ValidationException":
61
+ raise capo_codeguru_security.errors.validation_exception.ValidationException.from_json(
62
+ data, message
63
+ )
64
+ case _:
65
+ raise UnknownServiceError(code=code, message=message, response=response)
66
+
67
+
68
+ def handle_response(
69
+ response: zapros.Response,
70
+ ) -> capo_codeguru_security.types.tag_resource_response.TagResourceResponse:
71
+ out: capo_codeguru_security.types.tag_resource_response.TagResourceResponse = {} # type: ignore[typeddict-item]
72
+ return out
73
+
74
+
75
+ async def async_handle_response(
76
+ response: zapros.Response,
77
+ ) -> capo_codeguru_security.types.tag_resource_response.TagResourceResponse:
78
+ out: capo_codeguru_security.types.tag_resource_response.TagResourceResponse = {} # type: ignore[typeddict-item]
79
+ return out
80
+
81
+
82
+ def get_signer(
83
+ options: AsyncOperationOptions | OperationOptions,
84
+ auth_schemes: list[dict[str, Any]] | None = None,
85
+ ) -> capo_codeguru_security._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_codeguru_security._auth._sigv4.build_sigv4_auth_scheme(
101
+ "codeguru-security", options.region, endpoint_scheme
102
+ )
103
+ if sigv4_config is not None:
104
+ return capo_codeguru_security._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_codeguru_security.types.tag_resource_request.TagResourceRequest,
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("/") + "/tags/{resourceArn}"
123
+ url = url.replace("{resourceArn}", quote(input_["resource_arn"], safe=""))
124
+ params: list[tuple[str, str]] = []
125
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
126
+ body: bytes | None = json.dumps(
127
+ capo_codeguru_security.types.tag_resource_request.serialize_json(input_),
128
+ allow_nan=False,
129
+ ).encode()
130
+ headers["content-type"] = "application/json"
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 tag_resource(
141
+ options: OperationOptions,
142
+ input_: capo_codeguru_security.types.tag_resource_request.TagResourceRequest,
143
+ ) -> tuple[
144
+ capo_codeguru_security.types.tag_resource_response.TagResourceResponse,
145
+ zapros.Response,
146
+ ]:
147
+ response = options.client.handler.handle(build_request(options, input_))
148
+ try:
149
+ if response.status >= 300:
150
+ response.read()
151
+ handle_error(response)
152
+ return handle_response(response), response
153
+ except BaseException:
154
+ response.close()
155
+ raise
156
+
157
+
158
+ async def async_tag_resource(
159
+ options: AsyncOperationOptions,
160
+ input_: capo_codeguru_security.types.tag_resource_request.TagResourceRequest,
161
+ ) -> tuple[
162
+ capo_codeguru_security.types.tag_resource_response.TagResourceResponse,
163
+ zapros.Response,
164
+ ]:
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