|
20 | 20 | from .types import ( |
21 | 21 | Retryer, |
22 | 22 | CredentialsProvider, |
23 | | - HttpClient, |
| 23 | + HttpClient, AsyncHttpClient, |
24 | 24 | HttpRequest, |
25 | 25 | HttpResponse, |
26 | 26 | SigningContext, |
@@ -100,7 +100,7 @@ def __init__( |
100 | 100 | retryer: Optional[Retryer] = None, |
101 | 101 | signer: Optional[Signer] = None, |
102 | 102 | credentials_provider: Optional[CredentialsProvider] = None, |
103 | | - http_client: Optional[Union[HttpClient]] = None, |
| 103 | + http_client: Optional[Union[HttpClient, AsyncHttpClient]] = None, |
104 | 104 | address_style: Optional[AddressStyle] = None, |
105 | 105 | readwrite_timeout: Optional[Union[int, float]] = None, |
106 | 106 | response_handlers: Optional[List] = None, |
@@ -211,7 +211,7 @@ def verify_operation(self, op_input: OperationInput, options: _Options) -> None: |
211 | 211 | def apply_operation(self, options: _Options, op_input: OperationInput) -> None: |
212 | 212 | """apply operation""" |
213 | 213 | self._apply_operation_options(options) # pylint: disable=no-member |
214 | | - _apply_operation_metadata(op_input, options) |
| 214 | + self._apply_operation_metadata(op_input, options) |
215 | 215 |
|
216 | 216 |
|
217 | 217 | def build_request_context(self, op_input: OperationInput, options: _Options, inner: _InnerOptions |
@@ -377,6 +377,15 @@ def service_error_response_handler(response: HttpResponse) -> None: |
377 | 377 |
|
378 | 378 | options.response_handlers = handlers |
379 | 379 |
|
| 380 | + def _apply_operation_metadata(self, op_input: OperationInput, options: _Options) -> None: |
| 381 | + handlers = op_input.op_metadata.get('opm-response-handler', None) |
| 382 | + if handlers is not None: |
| 383 | + options.response_handlers.extend(handlers) |
| 384 | + |
| 385 | + stream = op_input.op_metadata.get('response-stream', None) |
| 386 | + if stream is not None: |
| 387 | + options.response_stream = stream |
| 388 | + |
380 | 389 | def _sent_request(self, op_input: OperationInput, options: _Options) -> OperationOutput: |
381 | 390 | context = self.build_request_context(op_input, options, self._inner) |
382 | 391 | response = self._sent_http_request(context, options) |
@@ -592,15 +601,6 @@ def _resolve_cloud_box(config: Config, options: _Options) -> None: |
592 | 601 | options.product = defaults.CLOUD_BOX_PRODUCT |
593 | 602 |
|
594 | 603 |
|
595 | | -def _apply_operation_metadata(op_input: OperationInput, options: _Options) -> None: |
596 | | - handlers = op_input.op_metadata.get('opm-response-handler', None) |
597 | | - if handlers is not None: |
598 | | - options.response_handlers.extend(handlers) |
599 | | - |
600 | | - stream = op_input.op_metadata.get('response-stream', None) |
601 | | - if stream is not None: |
602 | | - options.response_stream = stream |
603 | | - |
604 | 604 |
|
605 | 605 | def _build_url(op_input: OperationInput, options: _Options) -> str: |
606 | 606 | host = "" |
|
0 commit comments