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,161 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#GetAccountConfiguration``."""
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.encryption_config
19
+ import capo_codeguru_security.types.get_account_configuration_request
20
+ import capo_codeguru_security.types.get_account_configuration_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.get_account_configuration_response.GetAccountConfigurationResponse:
60
+ out: capo_codeguru_security.types.get_account_configuration_response.GetAccountConfigurationResponse = capo_codeguru_security.types.get_account_configuration_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.get_account_configuration_response.GetAccountConfigurationResponse:
69
+ out: capo_codeguru_security.types.get_account_configuration_response.GetAccountConfigurationResponse = capo_codeguru_security.types.get_account_configuration_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.get_account_configuration_request.GetAccountConfigurationRequest,
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("/") + "/accountConfiguration/get"
116
+ params: list[tuple[str, str]] = []
117
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
118
+ body: bytes | None = b""
119
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
120
+ normalized_url = zapros.URL(url)
121
+ for k, v in params:
122
+ normalized_url.search_params.append(k, v)
123
+ return zapros.Request(
124
+ normalized_url, "GET", headers=headers, body=body, context={"signer": signer}
125
+ )
126
+
127
+
128
+ def get_account_configuration(
129
+ options: OperationOptions,
130
+ input_: capo_codeguru_security.types.get_account_configuration_request.GetAccountConfigurationRequest,
131
+ ) -> tuple[
132
+ capo_codeguru_security.types.get_account_configuration_response.GetAccountConfigurationResponse,
133
+ zapros.Response,
134
+ ]:
135
+ response = options.client.handler.handle(build_request(options, input_))
136
+ try:
137
+ if response.status >= 300:
138
+ response.read()
139
+ handle_error(response)
140
+ return handle_response(response), response
141
+ except BaseException:
142
+ response.close()
143
+ raise
144
+
145
+
146
+ async def async_get_account_configuration(
147
+ options: AsyncOperationOptions,
148
+ input_: capo_codeguru_security.types.get_account_configuration_request.GetAccountConfigurationRequest,
149
+ ) -> tuple[
150
+ capo_codeguru_security.types.get_account_configuration_response.GetAccountConfigurationResponse,
151
+ zapros.Response,
152
+ ]:
153
+ response = await options.client.handler.ahandle(build_request(options, input_))
154
+ try:
155
+ if response.status >= 300:
156
+ await response.aread()
157
+ handle_error(response)
158
+ return await async_handle_response(response), response
159
+ except BaseException:
160
+ await response.aclose()
161
+ raise
@@ -0,0 +1,191 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#GetFindings``."""
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.findings
22
+ import capo_codeguru_security.types.get_findings_request
23
+ import capo_codeguru_security.types.get_findings_response
24
+ import capo_codeguru_security.types.status
25
+ from capo_codeguru_security._protocol.errors import parse_error_metadata_json
26
+ from capo_codeguru_security._rule_engine._endpoint_rule_set import (
27
+ EndpointParams,
28
+ resolve,
29
+ )
30
+ from capo_codeguru_security._services._pipeline import (
31
+ AsyncOperationOptions,
32
+ OperationOptions,
33
+ )
34
+ from capo_codeguru_security.errors import UnknownServiceError
35
+
36
+
37
+ def handle_error(response: zapros.Response) -> Never:
38
+ data = json.loads(response.read())
39
+ code, message = parse_error_metadata_json(response, data)
40
+ match code:
41
+ case "AccessDeniedException":
42
+ raise capo_codeguru_security.errors.access_denied_exception.AccessDeniedException.from_json(
43
+ data, message
44
+ )
45
+ case "ConflictException":
46
+ raise capo_codeguru_security.errors.conflict_exception.ConflictException.from_json(
47
+ data, message
48
+ )
49
+ case "InternalServerException":
50
+ raise capo_codeguru_security.errors.internal_server_exception.InternalServerException.from_json(
51
+ data, message
52
+ )
53
+ case "ResourceNotFoundException":
54
+ raise capo_codeguru_security.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
55
+ data, message
56
+ )
57
+ case "ThrottlingException":
58
+ raise capo_codeguru_security.errors.throttling_exception.ThrottlingException.from_json(
59
+ data, message
60
+ )
61
+ case "ValidationException":
62
+ raise capo_codeguru_security.errors.validation_exception.ValidationException.from_json(
63
+ data, message
64
+ )
65
+ case _:
66
+ raise UnknownServiceError(code=code, message=message, response=response)
67
+
68
+
69
+ def handle_response(
70
+ response: zapros.Response,
71
+ ) -> capo_codeguru_security.types.get_findings_response.GetFindingsResponse:
72
+ out: capo_codeguru_security.types.get_findings_response.GetFindingsResponse = (
73
+ capo_codeguru_security.types.get_findings_response.deserialize_json(
74
+ json.loads(response.read())
75
+ )
76
+ )
77
+ return out
78
+
79
+
80
+ async def async_handle_response(
81
+ response: zapros.Response,
82
+ ) -> capo_codeguru_security.types.get_findings_response.GetFindingsResponse:
83
+ out: capo_codeguru_security.types.get_findings_response.GetFindingsResponse = (
84
+ capo_codeguru_security.types.get_findings_response.deserialize_json(
85
+ json.loads(await response.aread())
86
+ )
87
+ )
88
+ return out
89
+
90
+
91
+ def get_signer(
92
+ options: AsyncOperationOptions | OperationOptions,
93
+ auth_schemes: list[dict[str, Any]] | None = None,
94
+ ) -> capo_codeguru_security._auth._signers.Signer | None:
95
+ name_to_schema = {s["name"]: s for s in (auth_schemes or [])} # noqa: F841
96
+ if (
97
+ options.credentials_provider is not None
98
+ and name_to_schema
99
+ and not name_to_schema.keys() & {"sigv4", "sigv4-s3express"}
100
+ ):
101
+ raise RuntimeError(
102
+ "Endpoint requires an unsupported auth scheme: " + ", ".join(name_to_schema)
103
+ )
104
+ if options.credentials_provider is not None:
105
+ endpoint_scheme = name_to_schema.get("sigv4") or name_to_schema.get(
106
+ "sigv4-s3express"
107
+ )
108
+ if endpoint_scheme is not None or not name_to_schema:
109
+ sigv4_config = capo_codeguru_security._auth._sigv4.build_sigv4_auth_scheme(
110
+ "codeguru-security", options.region, endpoint_scheme
111
+ )
112
+ if sigv4_config is not None:
113
+ return capo_codeguru_security._auth._signers.SigV4Signer(
114
+ options.credentials_provider, auth_scheme=sigv4_config
115
+ )
116
+ raise RuntimeError("Auth was not resolved")
117
+
118
+
119
+ def build_request(
120
+ options: OperationOptions | AsyncOperationOptions,
121
+ input_: capo_codeguru_security.types.get_findings_request.GetFindingsRequest,
122
+ ) -> zapros.Request:
123
+ endpoint = resolve(
124
+ EndpointParams(
125
+ Region=options.region,
126
+ UseDualStack=options.use_dual_stack,
127
+ UseFIPS=options.use_fips,
128
+ Endpoint=options.endpoint,
129
+ )
130
+ ) # noqa: F841
131
+ import capo_codeguru_security.types.status
132
+
133
+ url = endpoint.url.rstrip("/") + "/findings/{scanName}"
134
+ url = url.replace("{scanName}", quote(input_["scan_name"], safe=""))
135
+ params: list[tuple[str, str]] = []
136
+ if "next_token" in input_:
137
+ params.append(("nextToken", input_["next_token"]))
138
+ if "max_results" in input_:
139
+ params.append(("maxResults", str(input_["max_results"])))
140
+ if "status" in input_:
141
+ params.append(
142
+ (
143
+ "status",
144
+ capo_codeguru_security.types.status.serialize_json(input_["status"]),
145
+ )
146
+ )
147
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
148
+ body: bytes | None = b""
149
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
150
+ normalized_url = zapros.URL(url)
151
+ for k, v in params:
152
+ normalized_url.search_params.append(k, v)
153
+ return zapros.Request(
154
+ normalized_url, "GET", headers=headers, body=body, context={"signer": signer}
155
+ )
156
+
157
+
158
+ def get_findings(
159
+ options: OperationOptions,
160
+ input_: capo_codeguru_security.types.get_findings_request.GetFindingsRequest,
161
+ ) -> tuple[
162
+ capo_codeguru_security.types.get_findings_response.GetFindingsResponse,
163
+ zapros.Response,
164
+ ]:
165
+ response = options.client.handler.handle(build_request(options, input_))
166
+ try:
167
+ if response.status >= 300:
168
+ response.read()
169
+ handle_error(response)
170
+ return handle_response(response), response
171
+ except BaseException:
172
+ response.close()
173
+ raise
174
+
175
+
176
+ async def async_get_findings(
177
+ options: AsyncOperationOptions,
178
+ input_: capo_codeguru_security.types.get_findings_request.GetFindingsRequest,
179
+ ) -> tuple[
180
+ capo_codeguru_security.types.get_findings_response.GetFindingsResponse,
181
+ zapros.Response,
182
+ ]:
183
+ response = await options.client.handler.ahandle(build_request(options, input_))
184
+ try:
185
+ if response.status >= 300:
186
+ await response.aread()
187
+ handle_error(response)
188
+ return await async_handle_response(response), response
189
+ except BaseException:
190
+ await response.aclose()
191
+ raise
@@ -0,0 +1,176 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#GetMetricsSummary``."""
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.get_metrics_summary_request
19
+ import capo_codeguru_security.types.get_metrics_summary_response
20
+ import capo_codeguru_security.types.metrics_summary
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
+ ) -> (
60
+ capo_codeguru_security.types.get_metrics_summary_response.GetMetricsSummaryResponse
61
+ ):
62
+ out: capo_codeguru_security.types.get_metrics_summary_response.GetMetricsSummaryResponse = capo_codeguru_security.types.get_metrics_summary_response.deserialize_json(
63
+ json.loads(response.read())
64
+ )
65
+ return out
66
+
67
+
68
+ async def async_handle_response(
69
+ response: zapros.Response,
70
+ ) -> (
71
+ capo_codeguru_security.types.get_metrics_summary_response.GetMetricsSummaryResponse
72
+ ):
73
+ out: capo_codeguru_security.types.get_metrics_summary_response.GetMetricsSummaryResponse = capo_codeguru_security.types.get_metrics_summary_response.deserialize_json(
74
+ json.loads(await response.aread())
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.get_metrics_summary_request.GetMetricsSummaryRequest,
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
+ import capo_codeguru_security._protocol.serialize
120
+
121
+ url = endpoint.url.rstrip("/") + "/metrics/summary"
122
+ params: list[tuple[str, str]] = []
123
+ if "date" in input_:
124
+ params.append(
125
+ (
126
+ "date",
127
+ capo_codeguru_security._protocol.serialize.fmt_date_time(
128
+ input_["date"]
129
+ ),
130
+ )
131
+ )
132
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
133
+ body: bytes | None = b""
134
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
135
+ normalized_url = zapros.URL(url)
136
+ for k, v in params:
137
+ normalized_url.search_params.append(k, v)
138
+ return zapros.Request(
139
+ normalized_url, "GET", headers=headers, body=body, context={"signer": signer}
140
+ )
141
+
142
+
143
+ def get_metrics_summary(
144
+ options: OperationOptions,
145
+ input_: capo_codeguru_security.types.get_metrics_summary_request.GetMetricsSummaryRequest,
146
+ ) -> tuple[
147
+ capo_codeguru_security.types.get_metrics_summary_response.GetMetricsSummaryResponse,
148
+ zapros.Response,
149
+ ]:
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_metrics_summary(
162
+ options: AsyncOperationOptions,
163
+ input_: capo_codeguru_security.types.get_metrics_summary_request.GetMetricsSummaryRequest,
164
+ ) -> tuple[
165
+ capo_codeguru_security.types.get_metrics_summary_response.GetMetricsSummaryResponse,
166
+ zapros.Response,
167
+ ]:
168
+ response = await options.client.handler.ahandle(build_request(options, input_))
169
+ try:
170
+ if response.status >= 300:
171
+ await response.aread()
172
+ handle_error(response)
173
+ return await async_handle_response(response), response
174
+ except BaseException:
175
+ await response.aclose()
176
+ raise
@@ -0,0 +1,173 @@
1
+ """Generated from Smithy shape ``com.amazonaws.codegurusecurity#GetScan``."""
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.internal_server_exception
17
+ import capo_codeguru_security.errors.resource_not_found_exception
18
+ import capo_codeguru_security.errors.throttling_exception
19
+ import capo_codeguru_security.errors.validation_exception
20
+ import capo_codeguru_security.types.analysis_type
21
+ import capo_codeguru_security.types.get_scan_request
22
+ import capo_codeguru_security.types.get_scan_response
23
+ import capo_codeguru_security.types.scan_state
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 "InternalServerException":
45
+ raise capo_codeguru_security.errors.internal_server_exception.InternalServerException.from_json(
46
+ data, message
47
+ )
48
+ case "ResourceNotFoundException":
49
+ raise capo_codeguru_security.errors.resource_not_found_exception.ResourceNotFoundException.from_json(
50
+ data, message
51
+ )
52
+ case "ThrottlingException":
53
+ raise capo_codeguru_security.errors.throttling_exception.ThrottlingException.from_json(
54
+ data, message
55
+ )
56
+ case "ValidationException":
57
+ raise capo_codeguru_security.errors.validation_exception.ValidationException.from_json(
58
+ data, message
59
+ )
60
+ case _:
61
+ raise UnknownServiceError(code=code, message=message, response=response)
62
+
63
+
64
+ def handle_response(
65
+ response: zapros.Response,
66
+ ) -> capo_codeguru_security.types.get_scan_response.GetScanResponse:
67
+ out: capo_codeguru_security.types.get_scan_response.GetScanResponse = (
68
+ capo_codeguru_security.types.get_scan_response.deserialize_json(
69
+ json.loads(response.read())
70
+ )
71
+ )
72
+ return out
73
+
74
+
75
+ async def async_handle_response(
76
+ response: zapros.Response,
77
+ ) -> capo_codeguru_security.types.get_scan_response.GetScanResponse:
78
+ out: capo_codeguru_security.types.get_scan_response.GetScanResponse = (
79
+ capo_codeguru_security.types.get_scan_response.deserialize_json(
80
+ json.loads(await response.aread())
81
+ )
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.get_scan_request.GetScanRequest,
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("/") + "/scans/{scanName}"
127
+ url = url.replace("{scanName}", quote(input_["scan_name"], safe=""))
128
+ params: list[tuple[str, str]] = []
129
+ if "run_id" in input_:
130
+ params.append(("runId", input_["run_id"]))
131
+ headers: dict[str, str] = {k: ", ".join(v) for k, v in endpoint.headers.items()}
132
+ body: bytes | None = b""
133
+ signer = get_signer(options, auth_schemes=endpoint.properties.get("authSchemes"))
134
+ normalized_url = zapros.URL(url)
135
+ for k, v in params:
136
+ normalized_url.search_params.append(k, v)
137
+ return zapros.Request(
138
+ normalized_url, "GET", headers=headers, body=body, context={"signer": signer}
139
+ )
140
+
141
+
142
+ def get_scan(
143
+ options: OperationOptions,
144
+ input_: capo_codeguru_security.types.get_scan_request.GetScanRequest,
145
+ ) -> tuple[
146
+ capo_codeguru_security.types.get_scan_response.GetScanResponse, zapros.Response
147
+ ]:
148
+ response = options.client.handler.handle(build_request(options, input_))
149
+ try:
150
+ if response.status >= 300:
151
+ response.read()
152
+ handle_error(response)
153
+ return handle_response(response), response
154
+ except BaseException:
155
+ response.close()
156
+ raise
157
+
158
+
159
+ async def async_get_scan(
160
+ options: AsyncOperationOptions,
161
+ input_: capo_codeguru_security.types.get_scan_request.GetScanRequest,
162
+ ) -> tuple[
163
+ capo_codeguru_security.types.get_scan_response.GetScanResponse, 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