Bruno MurinoandGitHub
7815c6abe8
* made custom headers be available to async aws signer Signed-off-by: Bruno Murino <[email protected]> * updated changelog Signed-off-by: Bruno Murino <[email protected]> * added tests for using host header for AWS request signature on both sync and async clients Signed-off-by: Bruno Murino <[email protected]> * added documentation guide about aws auth when accessing via tunnel Signed-off-by: Bruno Murino <[email protected]> * small refactor of AWS Signer classes on sync and async clients; improved testing on them as well Signed-off-by: Bruno Murino <[email protected]> * changelog Signed-off-by: Bruno Murino <[email protected]> * fixed test Signed-off-by: Bruno Murino <[email protected]> * lint fix Signed-off-by: Bruno Murino <[email protected]> --------- Signed-off-by: Bruno Murino <[email protected]>
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# The OpenSearch Contributors require contributions made to
|
|
# this file be licensed under the Apache-2.0 license or a
|
|
# compatible open source license.
|
|
#
|
|
# Modifications Copyright OpenSearch Contributors. See
|
|
# GitHub history for details.
|
|
|
|
from typing import Any, Dict, Optional, Union
|
|
|
|
from opensearchpy.helpers.signer import AWSV4Signer
|
|
|
|
|
|
class AWSV4SignerAsyncAuth:
|
|
"""
|
|
AWS V4 Request Signer for Async Requests.
|
|
"""
|
|
|
|
def __init__(self, credentials: Any, region: str, service: str = "es") -> None:
|
|
self.signer = AWSV4Signer(credentials, region, service)
|
|
|
|
def __call__(
|
|
self,
|
|
method: str,
|
|
url: str,
|
|
body: Optional[Union[str, bytes]] = None,
|
|
headers: Optional[Dict[str, str]] = None,
|
|
) -> Dict[str, str]:
|
|
return self._sign_request(method=method, url=url, body=body, headers=headers)
|
|
|
|
def _sign_request(
|
|
self,
|
|
method: str,
|
|
url: str,
|
|
body: Optional[Union[str, bytes]],
|
|
headers: Optional[Dict[str, str]],
|
|
) -> Dict[str, str]:
|
|
"""
|
|
This method helps in signing the request by injecting the required headers.
|
|
:param prepared_request: unsigned headers
|
|
:return: signed headers
|
|
"""
|
|
|
|
updated_headers = self.signer.sign(
|
|
method=method,
|
|
url=url,
|
|
body=body,
|
|
headers=headers,
|
|
)
|
|
return updated_headers
|