Adding async support for AWSSigV4 (#254)

* Adding async support for AWSSigV4 Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Fix names for connection classes Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Update tests to async name space Signed-off-by: Harsha Vamsi Kalluri <[email protected]> * Add import exceptions to python < 3.6 Signed-off-by: Harsha Vamsi Kalluri <[email protected]> Signed-off-by: Harsha Vamsi Kalluri <[email protected]>
2023-01-02 17:25:48 -05:00
parent 4c4091b12c
commit cfb2cf7b60
14 changed files with 541 additions and 7 deletions
@@ -62,7 +62,7 @@ from .exceptions import (
    SSLError,
    TransportError,
 )
-from .helpers import AWSV4SignerAuth
+from .helpers import AWSV4SignerAsyncAuth, AWSV4SignerAuth
 from .serializer import JSONSerializer
 from .transport import Transport

@@ -79,6 +79,7 @@ __all__ = [
    "JSONSerializer",
    "Connection",
    "RequestsHttpConnection",
+    "AsyncHttpConnection",
    "Urllib3HttpConnection",
    "ImproperlyConfigured",
    "OpenSearchException",
@@ -95,6 +96,7 @@ __all__ = [
    "OpenSearchWarning",
    "OpenSearchDeprecationWarning",
    "AWSV4SignerAuth",
+    "AWSV4SignerAsyncAuth",
 ]

 try:
@@ -105,12 +107,14 @@ try:
    from ._async.client import AsyncOpenSearch
    from ._async.http_aiohttp import AIOHttpConnection, AsyncConnection
    from ._async.transport import AsyncTransport
+    from .connection import AsyncHttpConnection

    __all__ += [
        "AIOHttpConnection",
        "AsyncConnection",
        "AsyncTransport",
        "AsyncOpenSearch",
+        "AsyncHttpConnection",
    ]
 except (ImportError, SyntaxError):
    pass