Add support for async I/O

dev-requirements.txt

@@ -14,3 +14,4 @@ pandas
 pyyaml<5.3
 
 black; python_version>="3.6"
+git+https://github.com/python-trio/unasync

elasticsearch/__init__.py

@@ -5,6 +5,7 @@
 __version__ = VERSION
 __versionstr__ = ".".join(map(str, VERSION))
 
+import sys
 import logging
 import warnings
 
@@ -60,3 +61,23 @@
     "AuthorizationException",
     "ElasticsearchDeprecationWarning",
 ]
+
+try:
+    # Async is only supported on Python 3.6+
+    if sys.version_info < (3, 6):
+        raise ImportError()
+    import asyncio
+
+    from ._async import (
+        AsyncElasticsearch as AsyncElasticsearch,
+        AsyncTransport as AsyncTransport,
+        AIOHttpConnection as AIOHttpConnection,
+    )
+
+    __all__ += [
+        "AsyncElasticsearch",
+        "AsyncTransport",
+        "AIOHttpConnection",
+    ]
+except ImportError as e:
+    pass

elasticsearch/_async/__init__.py

@@ -0,0 +1,9 @@
+from .client import Elasticsearch as AsyncElasticsearch
+from .transport import AsyncTransport
+from .http_aiohttp import AIOHttpConnection
+
+__all__ = [
+    "AsyncElasticsearch",
+    "AsyncTransport",
+    "AIOHttpConnection",
+]