|
31 | 31 | SchemaValidationResourceWithStreamingResponse, |
32 | 32 | AsyncSchemaValidationResourceWithStreamingResponse, |
33 | 33 | ) |
34 | | -from ....types.api_gateway import operation_get_params, operation_list_params, operation_create_params |
| 34 | +from ....types.api_gateway import operation_get_params, operation_list_params |
35 | 35 | from ....types.api_gateway.operation_get_response import OperationGetResponse |
36 | 36 | from ....types.api_gateway.operation_list_response import OperationListResponse |
37 | 37 | from ....types.api_gateway.operation_create_response import OperationCreateResponse |
38 | 38 | from ....types.api_gateway.operation_delete_response import OperationDeleteResponse |
| 39 | +from ....types.api_gateway.api_shield_operation_param import APIShieldOperationParam |
39 | 40 | from ....types.api_gateway.operation_bulk_delete_response import OperationBulkDeleteResponse |
40 | 41 |
|
41 | 42 | __all__ = ["OperationsResource", "AsyncOperationsResource"] |
@@ -69,7 +70,7 @@ def create( |
69 | 70 | self, |
70 | 71 | *, |
71 | 72 | zone_id: str, |
72 | | - body: Iterable[operation_create_params.Body], |
| 73 | + body: Iterable[APIShieldOperationParam], |
73 | 74 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
74 | 75 | # The extra values given here take precedence over values defined on the client or passed to this method. |
75 | 76 | extra_headers: Headers | None = None, |
@@ -100,7 +101,7 @@ def create( |
100 | 101 | raise ValueError(f"Expected a non-empty value for `zone_id` but received {zone_id!r}") |
101 | 102 | return self._post( |
102 | 103 | f"/zones/{zone_id}/api_gateway/operations", |
103 | | - body=maybe_transform(body, Iterable[operation_create_params.Body]), |
| 104 | + body=maybe_transform(body, Iterable[APIShieldOperationParam]), |
104 | 105 | options=make_request_options( |
105 | 106 | extra_headers=extra_headers, |
106 | 107 | extra_query=extra_query, |
@@ -344,7 +345,7 @@ async def create( |
344 | 345 | self, |
345 | 346 | *, |
346 | 347 | zone_id: str, |
347 | | - body: Iterable[operation_create_params.Body], |
| 348 | + body: Iterable[APIShieldOperationParam], |
348 | 349 | # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs. |
349 | 350 | # The extra values given here take precedence over values defined on the client or passed to this method. |
350 | 351 | extra_headers: Headers | None = None, |
@@ -375,7 +376,7 @@ async def create( |
375 | 376 | raise ValueError(f"Expected a non-empty value for `zone_id` but received {zone_id!r}") |
376 | 377 | return await self._post( |
377 | 378 | f"/zones/{zone_id}/api_gateway/operations", |
378 | | - body=await async_maybe_transform(body, Iterable[operation_create_params.Body]), |
| 379 | + body=await async_maybe_transform(body, Iterable[APIShieldOperationParam]), |
379 | 380 | options=make_request_options( |
380 | 381 | extra_headers=extra_headers, |
381 | 382 | extra_query=extra_query, |
|
0 commit comments